anyscale 0.24.86__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- anyscale/__init__.py +181 -0
- anyscale/_private/anyscale_client/README.md +16 -0
- anyscale/_private/anyscale_client/__init__.py +8 -0
- anyscale/_private/anyscale_client/anyscale_client.py +1847 -0
- anyscale/_private/anyscale_client/common.py +593 -0
- anyscale/_private/anyscale_client/fake_anyscale_client.py +1080 -0
- anyscale/_private/docgen/README.md +15 -0
- anyscale/_private/docgen/__main__.py +700 -0
- anyscale/_private/docgen/api.md +1106 -0
- anyscale/_private/docgen/generator.py +559 -0
- anyscale/_private/docgen/generator_legacy.py +104 -0
- anyscale/_private/docgen/models.md +2261 -0
- anyscale/_private/models/__init__.py +2 -0
- anyscale/_private/models/image_uri.py +116 -0
- anyscale/_private/models/model_base.py +251 -0
- anyscale/_private/sdk/__init__.py +102 -0
- anyscale/_private/sdk/base_sdk.py +35 -0
- anyscale/_private/sdk/timer.py +46 -0
- anyscale/_private/utils/__init__.py +0 -0
- anyscale/_private/utils/progress_util.py +85 -0
- anyscale/_private/workload/__init__.py +2 -0
- anyscale/_private/workload/workload_config.py +195 -0
- anyscale/_private/workload/workload_sdk.py +324 -0
- anyscale/aggregated_instance_usage/__init__.py +36 -0
- anyscale/aggregated_instance_usage/_private/aggregated_instance_usage_sdk.py +30 -0
- anyscale/aggregated_instance_usage/commands.py +42 -0
- anyscale/aggregated_instance_usage/models.py +85 -0
- anyscale/anyscale-cloud-setup-gcp-oa.yaml +88 -0
- anyscale/anyscale-cloud-setup-gcp.yaml +113 -0
- anyscale/anyscale-cloud-setup-oa.yaml +121 -0
- anyscale/anyscale-cloud-setup.yaml +327 -0
- anyscale/anyscale_pydantic/HISTORY.md +1254 -0
- anyscale/anyscale_pydantic/LICENSE +21 -0
- anyscale/anyscale_pydantic/PKG-INFO +1351 -0
- anyscale/anyscale_pydantic/README.md +7 -0
- anyscale/anyscale_pydantic/__init__.py +131 -0
- anyscale/anyscale_pydantic/_hypothesis_plugin.py +391 -0
- anyscale/anyscale_pydantic/annotated_types.py +72 -0
- anyscale/anyscale_pydantic/class_validators.py +361 -0
- anyscale/anyscale_pydantic/color.py +494 -0
- anyscale/anyscale_pydantic/config.py +191 -0
- anyscale/anyscale_pydantic/dataclasses.py +478 -0
- anyscale/anyscale_pydantic/datetime_parse.py +248 -0
- anyscale/anyscale_pydantic/decorator.py +264 -0
- anyscale/anyscale_pydantic/env_settings.py +350 -0
- anyscale/anyscale_pydantic/error_wrappers.py +162 -0
- anyscale/anyscale_pydantic/errors.py +646 -0
- anyscale/anyscale_pydantic/fields.py +1256 -0
- anyscale/anyscale_pydantic/generics.py +400 -0
- anyscale/anyscale_pydantic/json.py +112 -0
- anyscale/anyscale_pydantic/main.py +1109 -0
- anyscale/anyscale_pydantic/mypy.py +943 -0
- anyscale/anyscale_pydantic/networks.py +739 -0
- anyscale/anyscale_pydantic/parse.py +66 -0
- anyscale/anyscale_pydantic/py.typed +0 -0
- anyscale/anyscale_pydantic/schema.py +1164 -0
- anyscale/anyscale_pydantic/tools.py +92 -0
- anyscale/anyscale_pydantic/types.py +1206 -0
- anyscale/anyscale_pydantic/typing.py +603 -0
- anyscale/anyscale_pydantic/utils.py +803 -0
- anyscale/anyscale_pydantic/validators.py +765 -0
- anyscale/anyscale_pydantic/version.py +38 -0
- anyscale/anyscale_schema.json +9 -0
- anyscale/api.py +215 -0
- anyscale/api_utils/README.md +2 -0
- anyscale/api_utils/__init__.py +0 -0
- anyscale/api_utils/common_utils.py +81 -0
- anyscale/api_utils/exceptions/__init__.py +0 -0
- anyscale/api_utils/exceptions/job_errors.py +2 -0
- anyscale/api_utils/job_logs_util.py +116 -0
- anyscale/api_utils/job_util.py +22 -0
- anyscale/api_utils/logs_util.py +61 -0
- anyscale/authenticate.py +298 -0
- anyscale/aws_iam_policies.py +465 -0
- anyscale/background/__init__.py +0 -0
- anyscale/background/job_runner.py +64 -0
- anyscale/cli_logger.py +378 -0
- anyscale/client/.gitignore +66 -0
- anyscale/client/.openapi-generator/VERSION +1 -0
- anyscale/client/.openapi-generator-ignore +23 -0
- anyscale/client/README.md +1070 -0
- anyscale/client/git_push.sh +58 -0
- anyscale/client/openapi_client/__init__.py +667 -0
- anyscale/client/openapi_client/api/__init__.py +6 -0
- anyscale/client/openapi_client/api/default_api.py +40922 -0
- anyscale/client/openapi_client/api_client.py +647 -0
- anyscale/client/openapi_client/configuration.py +373 -0
- anyscale/client/openapi_client/exceptions.py +120 -0
- anyscale/client/openapi_client/models/__init__.py +652 -0
- anyscale/client/openapi_client/models/access_config.py +122 -0
- anyscale/client/openapi_client/models/aggregated_instance_usage_with_cost_model.py +733 -0
- anyscale/client/openapi_client/models/aggregatedinstanceusagewithcostmodel_list_response.py +147 -0
- anyscale/client/openapi_client/models/aica_endpoint.py +527 -0
- anyscale/client/openapi_client/models/aica_endpoint_event.py +433 -0
- anyscale/client/openapi_client/models/aica_endpoint_event_level.py +103 -0
- anyscale/client/openapi_client/models/aica_endpoint_event_type.py +120 -0
- anyscale/client/openapi_client/models/aica_endpoint_scope.py +102 -0
- anyscale/client/openapi_client/models/aica_model.py +398 -0
- anyscale/client/openapi_client/models/aica_model_accelerator_map.py +123 -0
- anyscale/client/openapi_client/models/aica_model_configuration.py +209 -0
- anyscale/client/openapi_client/models/aica_observability_urls.py +178 -0
- anyscale/client/openapi_client/models/aicaendpoint_list_response.py +147 -0
- anyscale/client/openapi_client/models/aicaendpoint_response.py +121 -0
- anyscale/client/openapi_client/models/aicaendpointevent_list_response.py +147 -0
- anyscale/client/openapi_client/models/aicamodel_list_response.py +147 -0
- anyscale/client/openapi_client/models/aicamodel_response.py +121 -0
- anyscale/client/openapi_client/models/aioa_cloud_waitlist_record.py +254 -0
- anyscale/client/openapi_client/models/aioacloudwaitlistrecord_response.py +121 -0
- anyscale/client/openapi_client/models/alert_type.py +103 -0
- anyscale/client/openapi_client/models/anyscale_aws_account.py +121 -0
- anyscale/client/openapi_client/models/anyscale_service_account.py +256 -0
- anyscale/client/openapi_client/models/anyscale_version_response.py +121 -0
- anyscale/client/openapi_client/models/anyscaleawsaccount_response.py +121 -0
- anyscale/client/openapi_client/models/anyscaled_credential_response.py +121 -0
- anyscale/client/openapi_client/models/anyscaledcredentialresponse_response.py +121 -0
- anyscale/client/openapi_client/models/anyscaleserviceaccount_response.py +121 -0
- anyscale/client/openapi_client/models/anyscaleversionresponse_response.py +121 -0
- anyscale/client/openapi_client/models/api_key_parameters.py +147 -0
- anyscale/client/openapi_client/models/app_config.py +436 -0
- anyscale/client/openapi_client/models/app_config_config_schema.py +235 -0
- anyscale/client/openapi_client/models/appconfig_list_response.py +147 -0
- anyscale/client/openapi_client/models/appconfig_response.py +121 -0
- anyscale/client/openapi_client/models/application_type.py +99 -0
- anyscale/client/openapi_client/models/applied_snapshot.py +175 -0
- anyscale/client/openapi_client/models/apply_production_service_v2_model.py +490 -0
- anyscale/client/openapi_client/models/archive_status.py +101 -0
- anyscale/client/openapi_client/models/archived_logs_info.py +164 -0
- anyscale/client/openapi_client/models/archivedlogsinfo_response.py +121 -0
- anyscale/client/openapi_client/models/attach_machine_pool_to_cloud_request.py +152 -0
- anyscale/client/openapi_client/models/attachmachinepooltocloudresponse_response.py +121 -0
- anyscale/client/openapi_client/models/aviary_model_config_v2.py +358 -0
- anyscale/client/openapi_client/models/aws_credentials.py +181 -0
- anyscale/client/openapi_client/models/aws_memory_db_cluster_config.py +148 -0
- anyscale/client/openapi_client/models/aws_region_and_zones.py +123 -0
- anyscale/client/openapi_client/models/aws_region_info.py +152 -0
- anyscale/client/openapi_client/models/awsregionandzones_response.py +121 -0
- anyscale/client/openapi_client/models/bank_account_information.py +239 -0
- anyscale/client/openapi_client/models/base_job_status.py +105 -0
- anyscale/client/openapi_client/models/baseimagesenum.py +2130 -0
- anyscale/client/openapi_client/models/batch_response_batched_result_organization_invitation_base.py +121 -0
- anyscale/client/openapi_client/models/batched_result_organization_invitation_base.py +173 -0
- anyscale/client/openapi_client/models/billing_information.py +181 -0
- anyscale/client/openapi_client/models/billing_version_code.py +100 -0
- anyscale/client/openapi_client/models/body_aws_marketplace_registration_api_v2_organization_billing_aws_marketplace_registration_post.py +121 -0
- anyscale/client/openapi_client/models/buffer_registration.py +285 -0
- anyscale/client/openapi_client/models/build.py +607 -0
- anyscale/client/openapi_client/models/build_log_response.py +123 -0
- anyscale/client/openapi_client/models/build_registration.py +285 -0
- anyscale/client/openapi_client/models/build_response.py +121 -0
- anyscale/client/openapi_client/models/build_status.py +104 -0
- anyscale/client/openapi_client/models/buildlogresponse_response.py +121 -0
- anyscale/client/openapi_client/models/card.py +181 -0
- anyscale/client/openapi_client/models/card_id.py +108 -0
- anyscale/client/openapi_client/models/card_list_response.py +147 -0
- anyscale/client/openapi_client/models/change_password_params.py +148 -0
- anyscale/client/openapi_client/models/cleanup_leaked_grafana_dashboard_response.py +208 -0
- anyscale/client/openapi_client/models/cleanupleakedgrafanadashboardresponse_response.py +121 -0
- anyscale/client/openapi_client/models/clone_experimental_workspace.py +151 -0
- anyscale/client/openapi_client/models/cloud.py +802 -0
- anyscale/client/openapi_client/models/cloud_analytics_event.py +351 -0
- anyscale/client/openapi_client/models/cloud_analytics_event_cloud_provider_error.py +152 -0
- anyscale/client/openapi_client/models/cloud_analytics_event_cloud_resource.py +117 -0
- anyscale/client/openapi_client/models/cloud_analytics_event_command_name.py +103 -0
- anyscale/client/openapi_client/models/cloud_analytics_event_error.py +150 -0
- anyscale/client/openapi_client/models/cloud_analytics_event_name.py +109 -0
- anyscale/client/openapi_client/models/cloud_collaborator.py +175 -0
- anyscale/client/openapi_client/models/cloud_collaborator_value.py +177 -0
- anyscale/client/openapi_client/models/cloud_collaborators_query.py +122 -0
- anyscale/client/openapi_client/models/cloud_config.py +206 -0
- anyscale/client/openapi_client/models/cloud_data_bucket_access_mode.py +100 -0
- anyscale/client/openapi_client/models/cloud_data_bucket_file_type.py +102 -0
- anyscale/client/openapi_client/models/cloud_data_bucket_presigned_upload_info.py +268 -0
- anyscale/client/openapi_client/models/cloud_data_bucket_presigned_upload_request.py +152 -0
- anyscale/client/openapi_client/models/cloud_data_bucket_presigned_upload_scheme.py +100 -0
- anyscale/client/openapi_client/models/cloud_data_bucket_presigned_url_request.py +209 -0
- anyscale/client/openapi_client/models/cloud_data_bucket_presigned_url_response.py +296 -0
- anyscale/client/openapi_client/models/cloud_data_bucket_presigned_url_scheme.py +100 -0
- anyscale/client/openapi_client/models/cloud_data_bucket_request_scope.py +100 -0
- anyscale/client/openapi_client/models/cloud_hosting_type.py +100 -0
- anyscale/client/openapi_client/models/cloud_list_response.py +147 -0
- anyscale/client/openapi_client/models/cloud_name_options.py +121 -0
- anyscale/client/openapi_client/models/cloud_overview_dashboard.py +175 -0
- anyscale/client/openapi_client/models/cloud_project_collaborator.py +175 -0
- anyscale/client/openapi_client/models/cloud_project_collaborator_value.py +121 -0
- anyscale/client/openapi_client/models/cloud_provider.py +102 -0
- anyscale/client/openapi_client/models/cloud_providers.py +103 -0
- anyscale/client/openapi_client/models/cloud_region_and_zones.py +123 -0
- anyscale/client/openapi_client/models/cloud_region_info.py +152 -0
- anyscale/client/openapi_client/models/cloud_resource.py +740 -0
- anyscale/client/openapi_client/models/cloud_resource_gcp.py +691 -0
- anyscale/client/openapi_client/models/cloud_response.py +121 -0
- anyscale/client/openapi_client/models/cloud_state.py +104 -0
- anyscale/client/openapi_client/models/cloud_status.py +100 -0
- anyscale/client/openapi_client/models/cloud_type.py +100 -0
- anyscale/client/openapi_client/models/cloud_types.py +100 -0
- anyscale/client/openapi_client/models/cloud_version.py +100 -0
- anyscale/client/openapi_client/models/cloud_waitlist_status.py +102 -0
- anyscale/client/openapi_client/models/cloud_with_cloud_resource.py +830 -0
- anyscale/client/openapi_client/models/cloud_with_cloud_resource_gcp.py +830 -0
- anyscale/client/openapi_client/models/cloudcollaborator_list_response.py +147 -0
- anyscale/client/openapi_client/models/clouddatabucketpresigneduploadinfo_response.py +121 -0
- anyscale/client/openapi_client/models/clouddatabucketpresignedurlresponse_response.py +121 -0
- anyscale/client/openapi_client/models/cloudoverviewdashboard_response.py +121 -0
- anyscale/client/openapi_client/models/cloudregionandzones_response.py +121 -0
- anyscale/client/openapi_client/models/cloudresource_response.py +121 -0
- anyscale/client/openapi_client/models/cloudresourcegcp_response.py +121 -0
- anyscale/client/openapi_client/models/cloudwithcloudresource_response.py +121 -0
- anyscale/client/openapi_client/models/cloudwithcloudresourcegcp_response.py +121 -0
- anyscale/client/openapi_client/models/cluster_auth_response.py +148 -0
- anyscale/client/openapi_client/models/cluster_config.py +178 -0
- anyscale/client/openapi_client/models/cluster_config_with_session_idle_timeout.py +204 -0
- anyscale/client/openapi_client/models/cluster_environments_query.py +290 -0
- anyscale/client/openapi_client/models/cluster_event.py +174 -0
- anyscale/client/openapi_client/models/cluster_events_output.py +175 -0
- anyscale/client/openapi_client/models/cluster_features.py +152 -0
- anyscale/client/openapi_client/models/cluster_management_stack_versions.py +100 -0
- anyscale/client/openapi_client/models/cluster_startup.py +208 -0
- anyscale/client/openapi_client/models/cluster_status.py +104 -0
- anyscale/client/openapi_client/models/cluster_status_details.py +100 -0
- anyscale/client/openapi_client/models/clusterauthresponse_response.py +121 -0
- anyscale/client/openapi_client/models/clusterconfig_response.py +121 -0
- anyscale/client/openapi_client/models/clusterconfigwithsessionidletimeout_response.py +121 -0
- anyscale/client/openapi_client/models/clustereventsoutput_response.py +121 -0
- anyscale/client/openapi_client/models/clusterfeatures_response.py +121 -0
- anyscale/client/openapi_client/models/company_size.py +103 -0
- anyscale/client/openapi_client/models/compute_node_type.py +292 -0
- anyscale/client/openapi_client/models/compute_stack.py +100 -0
- anyscale/client/openapi_client/models/compute_template.py +415 -0
- anyscale/client/openapi_client/models/compute_template_config.py +461 -0
- anyscale/client/openapi_client/models/compute_template_query.py +316 -0
- anyscale/client/openapi_client/models/computetemplate_response.py +121 -0
- anyscale/client/openapi_client/models/computetemplateconfig_response.py +121 -0
- anyscale/client/openapi_client/models/create_aica_endpoint.py +210 -0
- anyscale/client/openapi_client/models/create_aioa_cloud_waitlist.py +173 -0
- anyscale/client/openapi_client/models/create_analytics_event.py +122 -0
- anyscale/client/openapi_client/models/create_app_config.py +235 -0
- anyscale/client/openapi_client/models/create_app_config_configuration_schema.py +235 -0
- anyscale/client/openapi_client/models/create_billing_version.py +181 -0
- anyscale/client/openapi_client/models/create_bug_report_response.py +152 -0
- anyscale/client/openapi_client/models/create_build.py +263 -0
- anyscale/client/openapi_client/models/create_byod_app_config.py +180 -0
- anyscale/client/openapi_client/models/create_byod_app_config_configuration_schema.py +206 -0
- anyscale/client/openapi_client/models/create_byod_build.py +152 -0
- anyscale/client/openapi_client/models/create_cloud_collaborator.py +148 -0
- anyscale/client/openapi_client/models/create_cloud_resource.py +682 -0
- anyscale/client/openapi_client/models/create_cloud_resource_gcp.py +633 -0
- anyscale/client/openapi_client/models/create_cloud_with_cloud_resource.py +546 -0
- anyscale/client/openapi_client/models/create_cluster_compute_config.py +463 -0
- anyscale/client/openapi_client/models/create_compute_template.py +229 -0
- anyscale/client/openapi_client/models/create_compute_template_config.py +464 -0
- anyscale/client/openapi_client/models/create_dataset.py +200 -0
- anyscale/client/openapi_client/models/create_experimental_workspace.py +435 -0
- anyscale/client/openapi_client/models/create_experimental_workspace_from_job.py +123 -0
- anyscale/client/openapi_client/models/create_fine_tuning_hyperparameters.py +156 -0
- anyscale/client/openapi_client/models/create_fine_tuning_job_product_request.py +353 -0
- anyscale/client/openapi_client/models/create_instance_usage_budget.py +253 -0
- anyscale/client/openapi_client/models/create_internal_production_job.py +262 -0
- anyscale/client/openapi_client/models/create_job_queue_config.py +206 -0
- anyscale/client/openapi_client/models/create_job_queue_requests.py +323 -0
- anyscale/client/openapi_client/models/create_machine_pool_request.py +151 -0
- anyscale/client/openapi_client/models/create_machine_pool_response.py +123 -0
- anyscale/client/openapi_client/models/create_machine_request.py +151 -0
- anyscale/client/openapi_client/models/create_machine_response.py +123 -0
- anyscale/client/openapi_client/models/create_metronome_webhook_notification.py +175 -0
- anyscale/client/openapi_client/models/create_notification_channel_record.py +146 -0
- anyscale/client/openapi_client/models/create_organization_configuration.py +199 -0
- anyscale/client/openapi_client/models/create_organization_invitation.py +121 -0
- anyscale/client/openapi_client/models/create_otp_return_api_model.py +148 -0
- anyscale/client/openapi_client/models/create_production_job_config.py +347 -0
- anyscale/client/openapi_client/models/create_resource_quota.py +293 -0
- anyscale/client/openapi_client/models/create_schedule.py +263 -0
- anyscale/client/openapi_client/models/create_session_from_snapshot_options.py +565 -0
- anyscale/client/openapi_client/models/create_session_in_db.py +434 -0
- anyscale/client/openapi_client/models/create_session_response.py +174 -0
- anyscale/client/openapi_client/models/create_user.py +439 -0
- anyscale/client/openapi_client/models/create_user_project_collaborator.py +148 -0
- anyscale/client/openapi_client/models/create_user_project_collaborator_value.py +121 -0
- anyscale/client/openapi_client/models/create_workspace_from_template.py +263 -0
- anyscale/client/openapi_client/models/createbugreportresponse_response.py +121 -0
- anyscale/client/openapi_client/models/createcomputetemplateconfig_response.py +121 -0
- anyscale/client/openapi_client/models/createmachinepoolresponse_response.py +121 -0
- anyscale/client/openapi_client/models/createmachineresponse_response.py +121 -0
- anyscale/client/openapi_client/models/createotpreturnapimodel_response.py +121 -0
- anyscale/client/openapi_client/models/createsessionresponse_response.py +121 -0
- anyscale/client/openapi_client/models/credit_card_information.py +268 -0
- anyscale/client/openapi_client/models/customer_alert_status.py +101 -0
- anyscale/client/openapi_client/models/customer_billing_type.py +101 -0
- anyscale/client/openapi_client/models/dataplane_services.py +102 -0
- anyscale/client/openapi_client/models/dataset.py +416 -0
- anyscale/client/openapi_client/models/dataset_list_response.py +150 -0
- anyscale/client/openapi_client/models/dataset_response.py +121 -0
- anyscale/client/openapi_client/models/dataset_upload.py +148 -0
- anyscale/client/openapi_client/models/datasetupload_response.py +121 -0
- anyscale/client/openapi_client/models/decorated_application_template.py +493 -0
- anyscale/client/openapi_client/models/decorated_build.py +664 -0
- anyscale/client/openapi_client/models/decorated_compute_template.py +446 -0
- anyscale/client/openapi_client/models/decorated_compute_template_config.py +490 -0
- anyscale/client/openapi_client/models/decorated_interactive_session.py +793 -0
- anyscale/client/openapi_client/models/decorated_job.py +793 -0
- anyscale/client/openapi_client/models/decorated_job_queue.py +639 -0
- anyscale/client/openapi_client/models/decorated_job_submission.py +575 -0
- anyscale/client/openapi_client/models/decorated_list_service_api_model.py +670 -0
- anyscale/client/openapi_client/models/decorated_production_job.py +805 -0
- anyscale/client/openapi_client/models/decorated_production_job_state_transition.py +319 -0
- anyscale/client/openapi_client/models/decorated_production_service_v2_api_model.py +641 -0
- anyscale/client/openapi_client/models/decorated_production_service_v2_version_api_model.py +437 -0
- anyscale/client/openapi_client/models/decorated_runtime_env.py +488 -0
- anyscale/client/openapi_client/models/decorated_schedule.py +552 -0
- anyscale/client/openapi_client/models/decorated_serve_deployment.py +711 -0
- anyscale/client/openapi_client/models/decorated_service_event_api_model.py +513 -0
- anyscale/client/openapi_client/models/decorated_session.py +1789 -0
- anyscale/client/openapi_client/models/decorated_support_request.py +283 -0
- anyscale/client/openapi_client/models/decorated_unified_job.py +466 -0
- anyscale/client/openapi_client/models/decoratedapplicationtemplate_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedapplicationtemplate_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedbuild_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedbuild_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedcomputetemplate_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedcomputetemplate_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedinteractivesession_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedinteractivesession_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedjob_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedjob_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedjobqueue_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedjobqueue_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedjobsubmission_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedjobsubmission_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedlistserviceapimodel_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedproductionjob_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedproductionjob_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedproductionjobstatetransition_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedproductionservicev2_apimodel_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedproductionservicev2_versionapimodel_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedruntimeenv_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedruntimeenv_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedschedule_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedschedule_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedservedeployment_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedservedeployment_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedserviceeventapimodel_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedsession_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedsession_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedsupportrequest_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedsupportrequest_response.py +121 -0
- anyscale/client/openapi_client/models/decoratedunifiedjob_list_response.py +147 -0
- anyscale/client/openapi_client/models/decoratedunifiedjob_response.py +121 -0
- anyscale/client/openapi_client/models/delete_machine_pool_request.py +123 -0
- anyscale/client/openapi_client/models/delete_machine_request.py +206 -0
- anyscale/client/openapi_client/models/deleted_platform_fine_tuned_model.py +148 -0
- anyscale/client/openapi_client/models/deletedplatformfinetunedmodel_response.py +121 -0
- anyscale/client/openapi_client/models/deletemachinepoolresponse_response.py +121 -0
- anyscale/client/openapi_client/models/detach_machine_pool_from_cloud_request.py +152 -0
- anyscale/client/openapi_client/models/detachmachinepoolfromcloudresponse_response.py +121 -0
- anyscale/client/openapi_client/models/dismissal_type.py +100 -0
- anyscale/client/openapi_client/models/editable_cloud_resource.py +206 -0
- anyscale/client/openapi_client/models/editable_cloud_resource_gcp.py +178 -0
- anyscale/client/openapi_client/models/error.py +174 -0
- anyscale/client/openapi_client/models/event_level.py +104 -0
- anyscale/client/openapi_client/models/execute_command_response.py +175 -0
- anyscale/client/openapi_client/models/execute_interactive_command_options.py +147 -0
- anyscale/client/openapi_client/models/execute_shell_command_options.py +121 -0
- anyscale/client/openapi_client/models/executecommandresponse_response.py +121 -0
- anyscale/client/openapi_client/models/experimental_workspace.py +748 -0
- anyscale/client/openapi_client/models/experimental_workspaces_sort_field.py +100 -0
- anyscale/client/openapi_client/models/experimentalworkspace_list_response.py +147 -0
- anyscale/client/openapi_client/models/experimentalworkspace_response.py +121 -0
- anyscale/client/openapi_client/models/external_service_status.py +147 -0
- anyscale/client/openapi_client/models/external_service_status_response.py +250 -0
- anyscale/client/openapi_client/models/external_terminal_command.py +280 -0
- anyscale/client/openapi_client/models/externalservicestatusresponse_response.py +121 -0
- anyscale/client/openapi_client/models/feature_compatibility.py +152 -0
- anyscale/client/openapi_client/models/feature_flag_response.py +121 -0
- anyscale/client/openapi_client/models/featureflagresponse_response.py +121 -0
- anyscale/client/openapi_client/models/fine_tune_type.py +100 -0
- anyscale/client/openapi_client/models/fine_tuned_model.py +412 -0
- anyscale/client/openapi_client/models/fine_tuning_job_status.py +103 -0
- anyscale/client/openapi_client/models/finetunedmodel_list_response.py +147 -0
- anyscale/client/openapi_client/models/finetunedmodel_response.py +121 -0
- anyscale/client/openapi_client/models/finish_ft_job_request.py +204 -0
- anyscale/client/openapi_client/models/finish_ft_job_request_v2.py +183 -0
- anyscale/client/openapi_client/models/gcp_file_store_config.py +175 -0
- anyscale/client/openapi_client/models/gcp_memorystore_instance_config.py +148 -0
- anyscale/client/openapi_client/models/grafana_dashboard.py +201 -0
- anyscale/client/openapi_client/models/grpc_protocol_config.py +178 -0
- anyscale/client/openapi_client/models/ha_job_error_types.py +103 -0
- anyscale/client/openapi_client/models/ha_job_event_level.py +101 -0
- anyscale/client/openapi_client/models/ha_job_event_origin.py +100 -0
- anyscale/client/openapi_client/models/ha_job_goal_states.py +102 -0
- anyscale/client/openapi_client/models/ha_job_states.py +109 -0
- anyscale/client/openapi_client/models/ha_job_type.py +100 -0
- anyscale/client/openapi_client/models/ha_jobs_sort_field.py +105 -0
- anyscale/client/openapi_client/models/head_ip.py +121 -0
- anyscale/client/openapi_client/models/headip_response.py +121 -0
- anyscale/client/openapi_client/models/http_protocol_config.py +150 -0
- anyscale/client/openapi_client/models/http_validation_error.py +120 -0
- anyscale/client/openapi_client/models/idle_termination_status.py +104 -0
- anyscale/client/openapi_client/models/import_aica_model.py +241 -0
- anyscale/client/openapi_client/models/instance_usage_budget.py +572 -0
- anyscale/client/openapi_client/models/instance_usage_budget_evaluation_period.py +100 -0
- anyscale/client/openapi_client/models/instanceusagebudget_list_response.py +147 -0
- anyscale/client/openapi_client/models/instanceusagebudget_response.py +121 -0
- anyscale/client/openapi_client/models/integration_details.py +120 -0
- anyscale/client/openapi_client/models/interactive_session_logs.py +152 -0
- anyscale/client/openapi_client/models/interactivesessionlogs_response.py +121 -0
- anyscale/client/openapi_client/models/internal_production_job.py +663 -0
- anyscale/client/openapi_client/models/internalproductionjob_response.py +121 -0
- anyscale/client/openapi_client/models/invoice.py +413 -0
- anyscale/client/openapi_client/models/invoice_list_response.py +147 -0
- anyscale/client/openapi_client/models/invoice_status.py +102 -0
- anyscale/client/openapi_client/models/invoices_query.py +150 -0
- anyscale/client/openapi_client/models/job_access.py +102 -0
- anyscale/client/openapi_client/models/job_queue.py +467 -0
- anyscale/client/openapi_client/models/job_queue_config.py +122 -0
- anyscale/client/openapi_client/models/job_queue_execution_mode.py +101 -0
- anyscale/client/openapi_client/models/job_queue_spec.py +263 -0
- anyscale/client/openapi_client/models/job_queue_state.py +100 -0
- anyscale/client/openapi_client/models/job_queues_query.py +262 -0
- anyscale/client/openapi_client/models/job_run_type.py +101 -0
- anyscale/client/openapi_client/models/job_state_log_level_types.py +100 -0
- anyscale/client/openapi_client/models/job_status.py +105 -0
- anyscale/client/openapi_client/models/job_submissions_sort_field.py +101 -0
- anyscale/client/openapi_client/models/jobqueue_response.py +121 -0
- anyscale/client/openapi_client/models/jobs_logs.py +152 -0
- anyscale/client/openapi_client/models/jobs_logs_query_info.py +181 -0
- anyscale/client/openapi_client/models/jobs_sort_field.py +104 -0
- anyscale/client/openapi_client/models/jobslogs_response.py +121 -0
- anyscale/client/openapi_client/models/jobslogsqueryinfo_response.py +121 -0
- anyscale/client/openapi_client/models/json_patch_operation.py +200 -0
- anyscale/client/openapi_client/models/kubernetes_manager_registration_request.py +123 -0
- anyscale/client/openapi_client/models/kubernetes_manager_registration_response.py +123 -0
- anyscale/client/openapi_client/models/kubernetesmanagerregistrationresponse_response.py +121 -0
- anyscale/client/openapi_client/models/lb_resource.py +123 -0
- anyscale/client/openapi_client/models/lbresource_response.py +121 -0
- anyscale/client/openapi_client/models/list_machine_pools_response.py +123 -0
- anyscale/client/openapi_client/models/list_machines_response.py +121 -0
- anyscale/client/openapi_client/models/list_resource_quotas_query.py +234 -0
- anyscale/client/openapi_client/models/list_response_metadata.py +146 -0
- anyscale/client/openapi_client/models/listmachinepoolsresponse_response.py +121 -0
- anyscale/client/openapi_client/models/listmachinesresponse_response.py +121 -0
- anyscale/client/openapi_client/models/log_detail.py +187 -0
- anyscale/client/openapi_client/models/log_details.py +151 -0
- anyscale/client/openapi_client/models/log_download_config.py +206 -0
- anyscale/client/openapi_client/models/log_download_request.py +150 -0
- anyscale/client/openapi_client/models/log_download_result.py +207 -0
- anyscale/client/openapi_client/models/log_file_chunk.py +439 -0
- anyscale/client/openapi_client/models/log_filter.py +430 -0
- anyscale/client/openapi_client/models/log_item.py +181 -0
- anyscale/client/openapi_client/models/log_item_batch.py +151 -0
- anyscale/client/openapi_client/models/log_level_types.py +100 -0
- anyscale/client/openapi_client/models/log_stream.py +151 -0
- anyscale/client/openapi_client/models/logdetails_response.py +121 -0
- anyscale/client/openapi_client/models/logdownloadresult_response.py +121 -0
- anyscale/client/openapi_client/models/login_user_params.py +205 -0
- anyscale/client/openapi_client/models/logitembatch_response.py +121 -0
- anyscale/client/openapi_client/models/logs_output.py +202 -0
- anyscale/client/openapi_client/models/logsoutput_response.py +121 -0
- anyscale/client/openapi_client/models/logstream_response.py +121 -0
- anyscale/client/openapi_client/models/long_running_workload.py +256 -0
- anyscale/client/openapi_client/models/longrunningworkload_list_response.py +147 -0
- anyscale/client/openapi_client/models/machine_allocation_state.py +100 -0
- anyscale/client/openapi_client/models/machine_connection_state.py +100 -0
- anyscale/client/openapi_client/models/machine_info.py +466 -0
- anyscale/client/openapi_client/models/machine_pool.py +266 -0
- anyscale/client/openapi_client/models/metronome_customer_info_model.py +148 -0
- anyscale/client/openapi_client/models/metronome_dashboard_type.py +101 -0
- anyscale/client/openapi_client/models/metronomecustomerinfomodel_list_response.py +147 -0
- anyscale/client/openapi_client/models/metronomecustomerinfomodel_response.py +121 -0
- anyscale/client/openapi_client/models/mini_build.py +267 -0
- anyscale/client/openapi_client/models/mini_cloud.py +321 -0
- anyscale/client/openapi_client/models/mini_cluster.py +148 -0
- anyscale/client/openapi_client/models/mini_compute_template.py +228 -0
- anyscale/client/openapi_client/models/mini_compute_template_config.py +121 -0
- anyscale/client/openapi_client/models/mini_job_run.py +599 -0
- anyscale/client/openapi_client/models/mini_namespace.py +148 -0
- anyscale/client/openapi_client/models/mini_organization.py +148 -0
- anyscale/client/openapi_client/models/mini_production_job.py +202 -0
- anyscale/client/openapi_client/models/mini_project.py +205 -0
- anyscale/client/openapi_client/models/mini_runtime_environment.py +147 -0
- anyscale/client/openapi_client/models/mini_schedule.py +180 -0
- anyscale/client/openapi_client/models/mini_user.py +266 -0
- anyscale/client/openapi_client/models/minibuild_list_response.py +147 -0
- anyscale/client/openapi_client/models/minicomputetemplate_list_response.py +147 -0
- anyscale/client/openapi_client/models/miniproject_list_response.py +147 -0
- anyscale/client/openapi_client/models/monitor_logs_extension.py +100 -0
- anyscale/client/openapi_client/models/named_entity.py +148 -0
- anyscale/client/openapi_client/models/nfs_mount_target.py +151 -0
- anyscale/client/openapi_client/models/node_registration_aws.py +152 -0
- anyscale/client/openapi_client/models/node_registration_gcp.py +123 -0
- anyscale/client/openapi_client/models/node_registration_k8_s.py +178 -0
- anyscale/client/openapi_client/models/node_registration_provisioned.py +150 -0
- anyscale/client/openapi_client/models/node_registration_v2.py +279 -0
- anyscale/client/openapi_client/models/node_type.py +100 -0
- anyscale/client/openapi_client/models/notification_channel_email_config.py +121 -0
- anyscale/client/openapi_client/models/notification_channel_webhook_config.py +121 -0
- anyscale/client/openapi_client/models/onboarding_user_cards_query.py +122 -0
- anyscale/client/openapi_client/models/organization.py +490 -0
- anyscale/client/openapi_client/models/organization_availability.py +148 -0
- anyscale/client/openapi_client/models/organization_collaborator.py +259 -0
- anyscale/client/openapi_client/models/organization_configuration.py +280 -0
- anyscale/client/openapi_client/models/organization_configuration_response.py +227 -0
- anyscale/client/openapi_client/models/organization_invitation.py +255 -0
- anyscale/client/openapi_client/models/organization_invitation_base.py +121 -0
- anyscale/client/openapi_client/models/organization_marketing_questions.py +198 -0
- anyscale/client/openapi_client/models/organization_permission_level.py +100 -0
- anyscale/client/openapi_client/models/organization_project_collaborator.py +175 -0
- anyscale/client/openapi_client/models/organization_project_collaborator_value.py +148 -0
- anyscale/client/openapi_client/models/organization_public_identifier.py +121 -0
- anyscale/client/openapi_client/models/organization_response.py +121 -0
- anyscale/client/openapi_client/models/organization_summary.py +229 -0
- anyscale/client/openapi_client/models/organization_usage_alert.py +210 -0
- anyscale/client/openapi_client/models/organization_usage_alert_severity.py +100 -0
- anyscale/client/openapi_client/models/organizationavailability_response.py +121 -0
- anyscale/client/openapi_client/models/organizationcollaborator_list_response.py +147 -0
- anyscale/client/openapi_client/models/organizationconfiguration_list_response.py +147 -0
- anyscale/client/openapi_client/models/organizationconfigurationresponse_response.py +121 -0
- anyscale/client/openapi_client/models/organizationinvitation_list_response.py +147 -0
- anyscale/client/openapi_client/models/organizationinvitation_response.py +121 -0
- anyscale/client/openapi_client/models/organizationinvitationbase_response.py +121 -0
- anyscale/client/openapi_client/models/organizationprojectcollaborator_list_response.py +147 -0
- anyscale/client/openapi_client/models/organizationpublicidentifier_response.py +121 -0
- anyscale/client/openapi_client/models/organizationusagealert_list_response.py +147 -0
- anyscale/client/openapi_client/models/page_query.py +153 -0
- anyscale/client/openapi_client/models/pause_schedule.py +123 -0
- anyscale/client/openapi_client/models/permission_level.py +101 -0
- anyscale/client/openapi_client/models/platform_fine_tuning_job.py +577 -0
- anyscale/client/openapi_client/models/platformfinetuningjob_list_response.py +147 -0
- anyscale/client/openapi_client/models/platformfinetuningjob_response.py +121 -0
- anyscale/client/openapi_client/models/product_autoscaler_flag.py +122 -0
- anyscale/client/openapi_client/models/product_type.py +100 -0
- anyscale/client/openapi_client/models/productautoscalerflag_response.py +121 -0
- anyscale/client/openapi_client/models/production_job.py +437 -0
- anyscale/client/openapi_client/models/production_job_config.py +348 -0
- anyscale/client/openapi_client/models/production_job_event.py +378 -0
- anyscale/client/openapi_client/models/production_job_event_scope_filter.py +101 -0
- anyscale/client/openapi_client/models/production_job_state_transition.py +293 -0
- anyscale/client/openapi_client/models/productionjob_response.py +121 -0
- anyscale/client/openapi_client/models/productionjobevent_list_response.py +147 -0
- anyscale/client/openapi_client/models/project.py +554 -0
- anyscale/client/openapi_client/models/project_base.py +121 -0
- anyscale/client/openapi_client/models/project_collaborator.py +175 -0
- anyscale/client/openapi_client/models/project_collaborator_value.py +175 -0
- anyscale/client/openapi_client/models/project_collaborators_put_message.py +121 -0
- anyscale/client/openapi_client/models/project_create_message.py +148 -0
- anyscale/client/openapi_client/models/project_default_session_name.py +121 -0
- anyscale/client/openapi_client/models/project_delete_message.py +121 -0
- anyscale/client/openapi_client/models/project_list_response.py +147 -0
- anyscale/client/openapi_client/models/project_patch_message.py +121 -0
- anyscale/client/openapi_client/models/project_response.py +121 -0
- anyscale/client/openapi_client/models/projectbase_response.py +121 -0
- anyscale/client/openapi_client/models/projectcollaborator_list_response.py +147 -0
- anyscale/client/openapi_client/models/projectdefaultsessionname_response.py +121 -0
- anyscale/client/openapi_client/models/projects_sort_field.py +101 -0
- anyscale/client/openapi_client/models/projects_violating_tree_hierarchy_response.py +121 -0
- anyscale/client/openapi_client/models/projectsviolatingtreehierarchyresponse_response.py +121 -0
- anyscale/client/openapi_client/models/protocols.py +150 -0
- anyscale/client/openapi_client/models/provider_metadata.py +205 -0
- anyscale/client/openapi_client/models/providermetadata_response.py +121 -0
- anyscale/client/openapi_client/models/python_modules.py +150 -0
- anyscale/client/openapi_client/models/quota.py +198 -0
- anyscale/client/openapi_client/models/ray_gcs_external_storage_config.py +178 -0
- anyscale/client/openapi_client/models/ray_runtime_env_config.py +262 -0
- anyscale/client/openapi_client/models/read_billing_version.py +210 -0
- anyscale/client/openapi_client/models/readbillingversion_list_response.py +147 -0
- anyscale/client/openapi_client/models/replica_details.py +152 -0
- anyscale/client/openapi_client/models/replica_state.py +104 -0
- anyscale/client/openapi_client/models/request_email_magic_link_response.py +147 -0
- anyscale/client/openapi_client/models/request_otp_return_api_model.py +148 -0
- anyscale/client/openapi_client/models/request_password_reset_params.py +147 -0
- anyscale/client/openapi_client/models/requestemailmagiclinkresponse_response.py +121 -0
- anyscale/client/openapi_client/models/requestotpreturnapimodel_response.py +121 -0
- anyscale/client/openapi_client/models/reset_password_params.py +148 -0
- anyscale/client/openapi_client/models/resource_quota.py +465 -0
- anyscale/client/openapi_client/models/resource_quota_status.py +123 -0
- anyscale/client/openapi_client/models/resourcequota_list_response.py +147 -0
- anyscale/client/openapi_client/models/resourcequota_response.py +121 -0
- anyscale/client/openapi_client/models/resources.py +234 -0
- anyscale/client/openapi_client/models/resubmit_ft_job_request.py +121 -0
- anyscale/client/openapi_client/models/rollback_service_model.py +122 -0
- anyscale/client/openapi_client/models/rollout_strategy.py +100 -0
- anyscale/client/openapi_client/models/s3_download_location.py +148 -0
- anyscale/client/openapi_client/models/schedule_config.py +151 -0
- anyscale/client/openapi_client/models/serve_deployment_grafana_dashboard_status.py +101 -0
- anyscale/client/openapi_client/models/serve_deployment_logs.py +152 -0
- anyscale/client/openapi_client/models/serve_deployment_state.py +104 -0
- anyscale/client/openapi_client/models/servedeploymentlogs_response.py +121 -0
- anyscale/client/openapi_client/models/server_session_token.py +121 -0
- anyscale/client/openapi_client/models/serversessiontoken_response.py +121 -0
- anyscale/client/openapi_client/models/service_config.py +178 -0
- anyscale/client/openapi_client/models/service_event_current_state.py +108 -0
- anyscale/client/openapi_client/models/service_event_level.py +102 -0
- anyscale/client/openapi_client/models/service_event_origin.py +103 -0
- anyscale/client/openapi_client/models/service_event_scope.py +103 -0
- anyscale/client/openapi_client/models/service_event_scope_filter.py +104 -0
- anyscale/client/openapi_client/models/service_event_type.py +125 -0
- anyscale/client/openapi_client/models/service_event_verbose_message_model.py +180 -0
- anyscale/client/openapi_client/models/service_goal_states.py +100 -0
- anyscale/client/openapi_client/models/service_observability_urls.py +206 -0
- anyscale/client/openapi_client/models/service_sort_field.py +101 -0
- anyscale/client/openapi_client/models/service_type.py +100 -0
- anyscale/client/openapi_client/models/service_usage.py +353 -0
- anyscale/client/openapi_client/models/service_version_state.py +106 -0
- anyscale/client/openapi_client/models/serviceeventverbosemessagemodel_list_response.py +147 -0
- anyscale/client/openapi_client/models/session.py +834 -0
- anyscale/client/openapi_client/models/session_access.py +102 -0
- anyscale/client/openapi_client/models/session_autosync_sessions_update_message.py +121 -0
- anyscale/client/openapi_client/models/session_command.py +413 -0
- anyscale/client/openapi_client/models/session_command_finish_options.py +226 -0
- anyscale/client/openapi_client/models/session_command_id.py +121 -0
- anyscale/client/openapi_client/models/session_command_types.py +100 -0
- anyscale/client/openapi_client/models/session_create_message.py +148 -0
- anyscale/client/openapi_client/models/session_delete_message.py +121 -0
- anyscale/client/openapi_client/models/session_describe.py +175 -0
- anyscale/client/openapi_client/models/session_details.py +148 -0
- anyscale/client/openapi_client/models/session_event.py +267 -0
- anyscale/client/openapi_client/models/session_event_cause.py +150 -0
- anyscale/client/openapi_client/models/session_event_types.py +111 -0
- anyscale/client/openapi_client/models/session_execute_message.py +121 -0
- anyscale/client/openapi_client/models/session_finish_command_message.py +175 -0
- anyscale/client/openapi_client/models/session_history_item.py +146 -0
- anyscale/client/openapi_client/models/session_kill_command_message.py +121 -0
- anyscale/client/openapi_client/models/session_list_response.py +147 -0
- anyscale/client/openapi_client/models/session_patch_message.py +121 -0
- anyscale/client/openapi_client/models/session_response.py +121 -0
- anyscale/client/openapi_client/models/session_ssh_key.py +148 -0
- anyscale/client/openapi_client/models/session_starting_up_data.py +146 -0
- anyscale/client/openapi_client/models/session_state.py +111 -0
- anyscale/client/openapi_client/models/session_state_change_message.py +121 -0
- anyscale/client/openapi_client/models/session_state_data.py +146 -0
- anyscale/client/openapi_client/models/session_stopping_data.py +146 -0
- anyscale/client/openapi_client/models/sessioncommand_list_response.py +147 -0
- anyscale/client/openapi_client/models/sessioncommandid_response.py +121 -0
- anyscale/client/openapi_client/models/sessiondescribe_response.py +121 -0
- anyscale/client/openapi_client/models/sessiondetails_response.py +121 -0
- anyscale/client/openapi_client/models/sessionevent_list_response.py +147 -0
- anyscale/client/openapi_client/models/sessionhistoryitem_list_response.py +147 -0
- anyscale/client/openapi_client/models/sessions_sort_field.py +104 -0
- anyscale/client/openapi_client/models/sessionsshkey_response.py +121 -0
- anyscale/client/openapi_client/models/setup_initialize_session_options.py +225 -0
- anyscale/client/openapi_client/models/show_otp_source_return_api_model.py +121 -0
- anyscale/client/openapi_client/models/showotpsourcereturnapimodel_response.py +121 -0
- anyscale/client/openapi_client/models/snapshot_create_message.py +148 -0
- anyscale/client/openapi_client/models/snapshot_delete_message.py +148 -0
- anyscale/client/openapi_client/models/snapshot_patch_message.py +148 -0
- anyscale/client/openapi_client/models/socket_message_schemas.py +499 -0
- anyscale/client/openapi_client/models/socket_message_types.py +113 -0
- anyscale/client/openapi_client/models/socketmessageschemas_response.py +121 -0
- anyscale/client/openapi_client/models/socketmessagetypes_response.py +121 -0
- anyscale/client/openapi_client/models/sort_order.py +100 -0
- anyscale/client/openapi_client/models/sso_login_info.py +151 -0
- anyscale/client/openapi_client/models/ssologininfo_response.py +121 -0
- anyscale/client/openapi_client/models/start_session_options.py +146 -0
- anyscale/client/openapi_client/models/stop_session_options.py +227 -0
- anyscale/client/openapi_client/models/stream_publish_request.py +239 -0
- anyscale/client/openapi_client/models/subnet_id_with_availability_zone_aws.py +148 -0
- anyscale/client/openapi_client/models/support_requests_query.py +178 -0
- anyscale/client/openapi_client/models/supportedbaseimagesenum.py +1570 -0
- anyscale/client/openapi_client/models/templatized_compute_configs.py +202 -0
- anyscale/client/openapi_client/models/templatized_decorated_application_templates.py +202 -0
- anyscale/client/openapi_client/models/templatizedcomputeconfigs_response.py +121 -0
- anyscale/client/openapi_client/models/templatizeddecoratedapplicationtemplates_response.py +121 -0
- anyscale/client/openapi_client/models/text_query.py +178 -0
- anyscale/client/openapi_client/models/timestamped_logs_output.py +148 -0
- anyscale/client/openapi_client/models/timestampedlogsoutput_response.py +121 -0
- anyscale/client/openapi_client/models/tool.py +100 -0
- anyscale/client/openapi_client/models/tracing_config.py +178 -0
- anyscale/client/openapi_client/models/try_login_email_response.py +208 -0
- anyscale/client/openapi_client/models/tryloginemailresponse_response.py +121 -0
- anyscale/client/openapi_client/models/unified_job_sort_field.py +103 -0
- anyscale/client/openapi_client/models/unified_job_status.py +114 -0
- anyscale/client/openapi_client/models/unified_job_type.py +100 -0
- anyscale/client/openapi_client/models/update_cloud_with_cloud_resource.py +178 -0
- anyscale/client/openapi_client/models/update_cloud_with_cloud_resource_gcp.py +178 -0
- anyscale/client/openapi_client/models/update_cluster_dns.py +152 -0
- anyscale/client/openapi_client/models/update_compute_template.py +146 -0
- anyscale/client/openapi_client/models/update_compute_template_config.py +464 -0
- anyscale/client/openapi_client/models/update_endpoint.py +152 -0
- anyscale/client/openapi_client/models/update_machine_pool_request.py +151 -0
- anyscale/client/openapi_client/models/update_model_deployment.py +152 -0
- anyscale/client/openapi_client/models/update_organization_collaborator.py +121 -0
- anyscale/client/openapi_client/models/update_project_collaborator.py +121 -0
- anyscale/client/openapi_client/models/update_resource_quota.py +122 -0
- anyscale/client/openapi_client/models/updatemachinepoolresponse_response.py +121 -0
- anyscale/client/openapi_client/models/upload_session_command_logs_locations.py +148 -0
- anyscale/client/openapi_client/models/uploadsessioncommandlogslocations_response.py +121 -0
- anyscale/client/openapi_client/models/user_info.py +569 -0
- anyscale/client/openapi_client/models/user_resend_email_options.py +147 -0
- anyscale/client/openapi_client/models/user_service_access_types.py +100 -0
- anyscale/client/openapi_client/models/userinfo_response.py +121 -0
- anyscale/client/openapi_client/models/utm_fields.py +224 -0
- anyscale/client/openapi_client/models/ux_instance.py +468 -0
- anyscale/client/openapi_client/models/validate_otp_params_api_model.py +121 -0
- anyscale/client/openapi_client/models/validation_error.py +175 -0
- anyscale/client/openapi_client/models/verify_response.py +147 -0
- anyscale/client/openapi_client/models/verifyresponse_response.py +121 -0
- anyscale/client/openapi_client/models/visibility.py +100 -0
- anyscale/client/openapi_client/models/waitlist_status_response.py +121 -0
- anyscale/client/openapi_client/models/waitlist_status_type.py +100 -0
- anyscale/client/openapi_client/models/waitliststatusresponse_response.py +121 -0
- anyscale/client/openapi_client/models/wand_b_run_details.py +147 -0
- anyscale/client/openapi_client/models/web_terminal.py +121 -0
- anyscale/client/openapi_client/models/webterminal_list_response.py +147 -0
- anyscale/client/openapi_client/models/webterminal_response.py +121 -0
- anyscale/client/openapi_client/models/worker_node_type.py +404 -0
- anyscale/client/openapi_client/models/workload_type.py +102 -0
- anyscale/client/openapi_client/models/workspace_dataplane_artifact.py +181 -0
- anyscale/client/openapi_client/models/workspace_dataplane_artifacts.py +123 -0
- anyscale/client/openapi_client/models/workspace_dataplane_proxied_artifacts.py +178 -0
- anyscale/client/openapi_client/models/workspace_event.py +325 -0
- anyscale/client/openapi_client/models/workspace_event_source.py +100 -0
- anyscale/client/openapi_client/models/workspace_event_source_filter.py +101 -0
- anyscale/client/openapi_client/models/workspace_readme.py +123 -0
- anyscale/client/openapi_client/models/workspace_snapshot_states.py +108 -0
- anyscale/client/openapi_client/models/workspace_template.py +353 -0
- anyscale/client/openapi_client/models/workspace_template_cluster_environment_metadata.py +178 -0
- anyscale/client/openapi_client/models/workspacedataplaneartifacts_response.py +121 -0
- anyscale/client/openapi_client/models/workspacedataplaneproxiedartifacts_response.py +121 -0
- anyscale/client/openapi_client/models/workspaceevent_list_response.py +147 -0
- anyscale/client/openapi_client/models/workspacereadme_response.py +121 -0
- anyscale/client/openapi_client/models/workspacetemplate_list_response.py +147 -0
- anyscale/client/openapi_client/models/workspacetemplateclusterenvironmentmetadata_response.py +121 -0
- anyscale/client/openapi_client/models/write_cloud.py +546 -0
- anyscale/client/openapi_client/models/write_cluster_config.py +123 -0
- anyscale/client/openapi_client/models/write_project.py +226 -0
- anyscale/client/openapi_client/models/write_session.py +147 -0
- anyscale/client/openapi_client/models/write_support_request.py +121 -0
- anyscale/client/openapi_client/rest.py +296 -0
- anyscale/client/requirements.txt +6 -0
- anyscale/client/setup.cfg +2 -0
- anyscale/client/setup.py +40 -0
- anyscale/client/test-requirements.txt +3 -0
- anyscale/client/tox.ini +9 -0
- anyscale/cloud.py +216 -0
- anyscale/cloud_resource.py +1032 -0
- anyscale/cluster.py +138 -0
- anyscale/cluster_compute.py +167 -0
- anyscale/cluster_env.py +173 -0
- anyscale/commands/__init__.py +0 -0
- anyscale/commands/aggregated_instance_usage_commands.py +86 -0
- anyscale/commands/anyscale_api/__init__.py +0 -0
- anyscale/commands/anyscale_api/api_commands.py +23 -0
- anyscale/commands/anyscale_api/session_commands_commands.py +80 -0
- anyscale/commands/anyscale_api/session_operations_commands.py +28 -0
- anyscale/commands/anyscale_api/sessions_commands.py +152 -0
- anyscale/commands/auth_commands.py +41 -0
- anyscale/commands/cloud_commands.py +1011 -0
- anyscale/commands/cloud_commands_util.py +10 -0
- anyscale/commands/cluster_commands.py +476 -0
- anyscale/commands/cluster_env_commands.py +139 -0
- anyscale/commands/command_examples.py +495 -0
- anyscale/commands/compute_config_commands.py +252 -0
- anyscale/commands/config_commands.py +213 -0
- anyscale/commands/exec_commands.py +14 -0
- anyscale/commands/experimental_integrations_commands.py +70 -0
- anyscale/commands/image_commands.py +125 -0
- anyscale/commands/job_commands.py +745 -0
- anyscale/commands/list_commands.py +85 -0
- anyscale/commands/llm/dataset_commands.py +269 -0
- anyscale/commands/llm/group.py +15 -0
- anyscale/commands/llm/models_commands.py +123 -0
- anyscale/commands/login_commands.py +79 -0
- anyscale/commands/logs_commands.py +312 -0
- anyscale/commands/machine_commands.py +116 -0
- anyscale/commands/machine_pool_commands.py +163 -0
- anyscale/commands/migrate_commands.py +84 -0
- anyscale/commands/project_commands.py +203 -0
- anyscale/commands/resource_quota_commands.py +214 -0
- anyscale/commands/schedule_commands.py +436 -0
- anyscale/commands/service_account_commands.py +72 -0
- anyscale/commands/service_commands.py +738 -0
- anyscale/commands/session_commands_hidden.py +179 -0
- anyscale/commands/util.py +152 -0
- anyscale/commands/workspace_commands.py +511 -0
- anyscale/commands/workspace_commands_v2.py +874 -0
- anyscale/component_activity_util.py +83 -0
- anyscale/compute_config/__init__.py +84 -0
- anyscale/compute_config/_private/compute_config_sdk.py +433 -0
- anyscale/compute_config/commands.py +122 -0
- anyscale/compute_config/models.py +630 -0
- anyscale/conf.py +23 -0
- anyscale/connect.py +1323 -0
- anyscale/connect_utils/__init__.py +0 -0
- anyscale/connect_utils/prepare_cluster.py +962 -0
- anyscale/connect_utils/project.py +298 -0
- anyscale/connect_utils/start_interactive_session.py +437 -0
- anyscale/controllers/__init__.py +0 -0
- anyscale/controllers/auth_controller.py +134 -0
- anyscale/controllers/base_controller.py +52 -0
- anyscale/controllers/cloud_controller.py +3609 -0
- anyscale/controllers/cloud_functional_verification_controller.py +858 -0
- anyscale/controllers/cluster_controller.py +720 -0
- anyscale/controllers/cluster_env_controller.py +219 -0
- anyscale/controllers/compute_config_controller.py +351 -0
- anyscale/controllers/config_controller.py +422 -0
- anyscale/controllers/experimental_integrations_controller.py +80 -0
- anyscale/controllers/job_controller.py +647 -0
- anyscale/controllers/jobs_bg_controller.py +0 -0
- anyscale/controllers/list_controller.py +290 -0
- anyscale/controllers/llm/__init__.py +0 -0
- anyscale/controllers/llm/models_controller.py +144 -0
- anyscale/controllers/logs_controller.py +449 -0
- anyscale/controllers/machine_controller.py +37 -0
- anyscale/controllers/machine_pool_controller.py +86 -0
- anyscale/controllers/project_controller.py +281 -0
- anyscale/controllers/resource_quota_controller.py +183 -0
- anyscale/controllers/schedule_controller.py +333 -0
- anyscale/controllers/service_account_controller.py +168 -0
- anyscale/controllers/service_controller.py +453 -0
- anyscale/controllers/workspace_controller.py +253 -0
- anyscale/feature_flags.py +4 -0
- anyscale/fingerprint.py +62 -0
- anyscale/formatters/__init__.py +0 -0
- anyscale/formatters/clouds_formatter.py +65 -0
- anyscale/formatters/common_formatter.py +22 -0
- anyscale/gcp_verification.py +792 -0
- anyscale/image/__init__.py +73 -0
- anyscale/image/_private/image_sdk.py +202 -0
- anyscale/image/commands.py +117 -0
- anyscale/image/models.py +57 -0
- anyscale/integrations.py +329 -0
- anyscale/job/__init__.py +166 -0
- anyscale/job/_private/job_sdk.py +497 -0
- anyscale/job/commands.py +267 -0
- anyscale/job/models.py +500 -0
- anyscale/links.py +4 -0
- anyscale/llm/__init__.py +2 -0
- anyscale/llm/dataset/__init__.py +2 -0
- anyscale/llm/dataset/_private/__init__.py +0 -0
- anyscale/llm/dataset/_private/docs.py +63 -0
- anyscale/llm/dataset/_private/models.py +71 -0
- anyscale/llm/dataset/_private/sdk.py +147 -0
- anyscale/llm/model/__init__.py +2 -0
- anyscale/llm/model/_private/models_sdk.py +62 -0
- anyscale/llm/model/commands.py +93 -0
- anyscale/llm/model/models.py +171 -0
- anyscale/llm/model/sdk.py +62 -0
- anyscale/llm/sdk.py +27 -0
- anyscale/memorydb_supported_zones.json +22 -0
- anyscale/models/job_model.py +457 -0
- anyscale/models/service_model.py +125 -0
- anyscale/project.py +501 -0
- anyscale/schedule/__init__.py +91 -0
- anyscale/schedule/_private/schedule_sdk.py +165 -0
- anyscale/schedule/commands.py +149 -0
- anyscale/schedule/models.py +145 -0
- anyscale/scripts.py +164 -0
- anyscale/sdk/anyscale_client/__init__.py +235 -0
- anyscale/sdk/anyscale_client/api/__init__.py +6 -0
- anyscale/sdk/anyscale_client/api/default_api.py +11625 -0
- anyscale/sdk/anyscale_client/api_client.py +647 -0
- anyscale/sdk/anyscale_client/configuration.py +373 -0
- anyscale/sdk/anyscale_client/exceptions.py +120 -0
- anyscale/sdk/anyscale_client/models/__init__.py +220 -0
- anyscale/sdk/anyscale_client/models/access_config.py +122 -0
- anyscale/sdk/anyscale_client/models/app_config.py +436 -0
- anyscale/sdk/anyscale_client/models/app_config_config_schema.py +235 -0
- anyscale/sdk/anyscale_client/models/appconfig_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/appconfig_response.py +121 -0
- anyscale/sdk/anyscale_client/models/apply_production_service_v2_model.py +490 -0
- anyscale/sdk/anyscale_client/models/apply_service_model.py +490 -0
- anyscale/sdk/anyscale_client/models/archive_status.py +101 -0
- anyscale/sdk/anyscale_client/models/base_job_status.py +105 -0
- anyscale/sdk/anyscale_client/models/baseimagesenum.py +2130 -0
- anyscale/sdk/anyscale_client/models/build.py +607 -0
- anyscale/sdk/anyscale_client/models/build_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/build_log_response.py +123 -0
- anyscale/sdk/anyscale_client/models/build_response.py +121 -0
- anyscale/sdk/anyscale_client/models/build_status.py +104 -0
- anyscale/sdk/anyscale_client/models/buildlogresponse_response.py +121 -0
- anyscale/sdk/anyscale_client/models/cloud.py +802 -0
- anyscale/sdk/anyscale_client/models/cloud_config.py +206 -0
- anyscale/sdk/anyscale_client/models/cloud_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/cloud_providers.py +103 -0
- anyscale/sdk/anyscale_client/models/cloud_response.py +121 -0
- anyscale/sdk/anyscale_client/models/cloud_state.py +104 -0
- anyscale/sdk/anyscale_client/models/cloud_status.py +100 -0
- anyscale/sdk/anyscale_client/models/cloud_type.py +100 -0
- anyscale/sdk/anyscale_client/models/cloud_types.py +100 -0
- anyscale/sdk/anyscale_client/models/cloud_version.py +100 -0
- anyscale/sdk/anyscale_client/models/clouds_query.py +150 -0
- anyscale/sdk/anyscale_client/models/cluster.py +721 -0
- anyscale/sdk/anyscale_client/models/cluster_compute.py +415 -0
- anyscale/sdk/anyscale_client/models/cluster_compute_config.py +461 -0
- anyscale/sdk/anyscale_client/models/cluster_computes_query.py +293 -0
- anyscale/sdk/anyscale_client/models/cluster_environment.py +380 -0
- anyscale/sdk/anyscale_client/models/cluster_environment_build.py +578 -0
- anyscale/sdk/anyscale_client/models/cluster_environment_build_log_response.py +123 -0
- anyscale/sdk/anyscale_client/models/cluster_environment_build_operation.py +237 -0
- anyscale/sdk/anyscale_client/models/cluster_environment_build_status.py +104 -0
- anyscale/sdk/anyscale_client/models/cluster_environments_query.py +290 -0
- anyscale/sdk/anyscale_client/models/cluster_head_node_info.py +152 -0
- anyscale/sdk/anyscale_client/models/cluster_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/cluster_management_stack_versions.py +100 -0
- anyscale/sdk/anyscale_client/models/cluster_operation.py +266 -0
- anyscale/sdk/anyscale_client/models/cluster_operation_type.py +101 -0
- anyscale/sdk/anyscale_client/models/cluster_response.py +121 -0
- anyscale/sdk/anyscale_client/models/cluster_services_urls.py +430 -0
- anyscale/sdk/anyscale_client/models/cluster_state.py +108 -0
- anyscale/sdk/anyscale_client/models/cluster_status.py +104 -0
- anyscale/sdk/anyscale_client/models/cluster_status_details.py +100 -0
- anyscale/sdk/anyscale_client/models/clustercompute_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/clustercompute_response.py +121 -0
- anyscale/sdk/anyscale_client/models/clusterenvironment_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/clusterenvironment_response.py +121 -0
- anyscale/sdk/anyscale_client/models/clusterenvironmentbuild_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/clusterenvironmentbuild_response.py +121 -0
- anyscale/sdk/anyscale_client/models/clusterenvironmentbuildlogresponse_response.py +121 -0
- anyscale/sdk/anyscale_client/models/clusterenvironmentbuildoperation_response.py +121 -0
- anyscale/sdk/anyscale_client/models/clusteroperation_response.py +121 -0
- anyscale/sdk/anyscale_client/models/clusters_query.py +234 -0
- anyscale/sdk/anyscale_client/models/compute_node_type.py +292 -0
- anyscale/sdk/anyscale_client/models/compute_stack.py +100 -0
- anyscale/sdk/anyscale_client/models/compute_template.py +415 -0
- anyscale/sdk/anyscale_client/models/compute_template_config.py +461 -0
- anyscale/sdk/anyscale_client/models/compute_template_query.py +316 -0
- anyscale/sdk/anyscale_client/models/computetemplate_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/computetemplate_response.py +121 -0
- anyscale/sdk/anyscale_client/models/computetemplateconfig_response.py +121 -0
- anyscale/sdk/anyscale_client/models/create_app_config.py +235 -0
- anyscale/sdk/anyscale_client/models/create_app_config_configuration_schema.py +235 -0
- anyscale/sdk/anyscale_client/models/create_build.py +263 -0
- anyscale/sdk/anyscale_client/models/create_byod_app_config_configuration_schema.py +206 -0
- anyscale/sdk/anyscale_client/models/create_byod_cluster_environment.py +180 -0
- anyscale/sdk/anyscale_client/models/create_byod_cluster_environment_build.py +152 -0
- anyscale/sdk/anyscale_client/models/create_byod_cluster_environment_configuration_schema.py +208 -0
- anyscale/sdk/anyscale_client/models/create_cloud.py +518 -0
- anyscale/sdk/anyscale_client/models/create_cluster.py +376 -0
- anyscale/sdk/anyscale_client/models/create_cluster_compute.py +229 -0
- anyscale/sdk/anyscale_client/models/create_cluster_compute_config.py +463 -0
- anyscale/sdk/anyscale_client/models/create_cluster_environment.py +235 -0
- anyscale/sdk/anyscale_client/models/create_cluster_environment_build.py +263 -0
- anyscale/sdk/anyscale_client/models/create_cluster_environment_configuration_schema.py +235 -0
- anyscale/sdk/anyscale_client/models/create_compute_template.py +229 -0
- anyscale/sdk/anyscale_client/models/create_compute_template_config.py +464 -0
- anyscale/sdk/anyscale_client/models/create_job_queue_config.py +206 -0
- anyscale/sdk/anyscale_client/models/create_production_job.py +234 -0
- anyscale/sdk/anyscale_client/models/create_production_job_config.py +347 -0
- anyscale/sdk/anyscale_client/models/create_project.py +207 -0
- anyscale/sdk/anyscale_client/models/create_schedule.py +263 -0
- anyscale/sdk/anyscale_client/models/create_session.py +432 -0
- anyscale/sdk/anyscale_client/models/create_session_command.py +152 -0
- anyscale/sdk/anyscale_client/models/create_sso_config.py +150 -0
- anyscale/sdk/anyscale_client/models/grpc_protocol_config.py +178 -0
- anyscale/sdk/anyscale_client/models/ha_job_goal_states.py +102 -0
- anyscale/sdk/anyscale_client/models/ha_job_states.py +109 -0
- anyscale/sdk/anyscale_client/models/http_protocol_config.py +150 -0
- anyscale/sdk/anyscale_client/models/http_validation_error.py +120 -0
- anyscale/sdk/anyscale_client/models/idle_termination_status.py +104 -0
- anyscale/sdk/anyscale_client/models/job.py +466 -0
- anyscale/sdk/anyscale_client/models/job_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/job_queue_config.py +122 -0
- anyscale/sdk/anyscale_client/models/job_queue_execution_mode.py +101 -0
- anyscale/sdk/anyscale_client/models/job_queue_spec.py +263 -0
- anyscale/sdk/anyscale_client/models/job_run_type.py +101 -0
- anyscale/sdk/anyscale_client/models/job_status.py +105 -0
- anyscale/sdk/anyscale_client/models/jobs_query.py +458 -0
- anyscale/sdk/anyscale_client/models/jobs_sort_field.py +104 -0
- anyscale/sdk/anyscale_client/models/list_response_metadata.py +146 -0
- anyscale/sdk/anyscale_client/models/list_service_model.py +347 -0
- anyscale/sdk/anyscale_client/models/listservicemodel_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/log_download_result.py +207 -0
- anyscale/sdk/anyscale_client/models/log_file_chunk.py +439 -0
- anyscale/sdk/anyscale_client/models/log_level_types.py +100 -0
- anyscale/sdk/anyscale_client/models/log_stream.py +151 -0
- anyscale/sdk/anyscale_client/models/logdownloadresult_response.py +121 -0
- anyscale/sdk/anyscale_client/models/logstream_response.py +121 -0
- anyscale/sdk/anyscale_client/models/node_type.py +100 -0
- anyscale/sdk/anyscale_client/models/object_storage_config.py +122 -0
- anyscale/sdk/anyscale_client/models/object_storage_config_s3.py +256 -0
- anyscale/sdk/anyscale_client/models/objectstorageconfig_response.py +121 -0
- anyscale/sdk/anyscale_client/models/operation_error.py +123 -0
- anyscale/sdk/anyscale_client/models/operation_progress.py +123 -0
- anyscale/sdk/anyscale_client/models/operation_result.py +150 -0
- anyscale/sdk/anyscale_client/models/organization.py +209 -0
- anyscale/sdk/anyscale_client/models/organization_response.py +121 -0
- anyscale/sdk/anyscale_client/models/page_query.py +153 -0
- anyscale/sdk/anyscale_client/models/pause_schedule.py +123 -0
- anyscale/sdk/anyscale_client/models/production_job.py +437 -0
- anyscale/sdk/anyscale_client/models/production_job_config.py +348 -0
- anyscale/sdk/anyscale_client/models/production_job_state_transition.py +293 -0
- anyscale/sdk/anyscale_client/models/production_service_v2_model.py +612 -0
- anyscale/sdk/anyscale_client/models/production_service_v2_version_model.py +437 -0
- anyscale/sdk/anyscale_client/models/productionjob_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/productionjob_response.py +121 -0
- anyscale/sdk/anyscale_client/models/productionservicev2_model_response.py +121 -0
- anyscale/sdk/anyscale_client/models/project.py +467 -0
- anyscale/sdk/anyscale_client/models/project_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/project_response.py +121 -0
- anyscale/sdk/anyscale_client/models/projects_query.py +234 -0
- anyscale/sdk/anyscale_client/models/protocols.py +150 -0
- anyscale/sdk/anyscale_client/models/python_modules.py +150 -0
- anyscale/sdk/anyscale_client/models/python_version.py +105 -0
- anyscale/sdk/anyscale_client/models/ray_gcs_external_storage_config.py +178 -0
- anyscale/sdk/anyscale_client/models/ray_runtime_env_config.py +262 -0
- anyscale/sdk/anyscale_client/models/resources.py +234 -0
- anyscale/sdk/anyscale_client/models/rollback_service_model.py +122 -0
- anyscale/sdk/anyscale_client/models/rollout_strategy.py +100 -0
- anyscale/sdk/anyscale_client/models/runtime_environment.py +406 -0
- anyscale/sdk/anyscale_client/models/runtimeenvironment_response.py +121 -0
- anyscale/sdk/anyscale_client/models/schedule_api_model.py +467 -0
- anyscale/sdk/anyscale_client/models/schedule_config.py +151 -0
- anyscale/sdk/anyscale_client/models/scheduleapimodel_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/scheduleapimodel_response.py +121 -0
- anyscale/sdk/anyscale_client/models/service_config.py +178 -0
- anyscale/sdk/anyscale_client/models/service_event_current_state.py +108 -0
- anyscale/sdk/anyscale_client/models/service_goal_states.py +100 -0
- anyscale/sdk/anyscale_client/models/service_model.py +612 -0
- anyscale/sdk/anyscale_client/models/service_observability_urls.py +206 -0
- anyscale/sdk/anyscale_client/models/service_sort_field.py +101 -0
- anyscale/sdk/anyscale_client/models/service_type.py +100 -0
- anyscale/sdk/anyscale_client/models/service_version_state.py +106 -0
- anyscale/sdk/anyscale_client/models/servicemodel_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/servicemodel_response.py +121 -0
- anyscale/sdk/anyscale_client/models/session.py +1535 -0
- anyscale/sdk/anyscale_client/models/session_command.py +350 -0
- anyscale/sdk/anyscale_client/models/session_command_types.py +100 -0
- anyscale/sdk/anyscale_client/models/session_event.py +267 -0
- anyscale/sdk/anyscale_client/models/session_event_cause.py +150 -0
- anyscale/sdk/anyscale_client/models/session_event_types.py +111 -0
- anyscale/sdk/anyscale_client/models/session_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/session_operation.py +266 -0
- anyscale/sdk/anyscale_client/models/session_operation_type.py +101 -0
- anyscale/sdk/anyscale_client/models/session_response.py +121 -0
- anyscale/sdk/anyscale_client/models/session_starting_up_data.py +146 -0
- anyscale/sdk/anyscale_client/models/session_state.py +111 -0
- anyscale/sdk/anyscale_client/models/session_state_data.py +146 -0
- anyscale/sdk/anyscale_client/models/session_stopping_data.py +146 -0
- anyscale/sdk/anyscale_client/models/sessioncommand_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/sessioncommand_response.py +121 -0
- anyscale/sdk/anyscale_client/models/sessionevent_list_response.py +147 -0
- anyscale/sdk/anyscale_client/models/sessionoperation_response.py +121 -0
- anyscale/sdk/anyscale_client/models/sessions_query.py +206 -0
- anyscale/sdk/anyscale_client/models/sort_by_clause_jobs_sort_field.py +148 -0
- anyscale/sdk/anyscale_client/models/sort_order.py +100 -0
- anyscale/sdk/anyscale_client/models/sso_config.py +237 -0
- anyscale/sdk/anyscale_client/models/sso_mode.py +101 -0
- anyscale/sdk/anyscale_client/models/ssoconfig_response.py +121 -0
- anyscale/sdk/anyscale_client/models/start_cluster_options.py +178 -0
- anyscale/sdk/anyscale_client/models/start_session_options.py +206 -0
- anyscale/sdk/anyscale_client/models/static_sso_config.py +210 -0
- anyscale/sdk/anyscale_client/models/supportedbaseimagesenum.py +1570 -0
- anyscale/sdk/anyscale_client/models/terminate_cluster_options.py +122 -0
- anyscale/sdk/anyscale_client/models/terminate_session_options.py +206 -0
- anyscale/sdk/anyscale_client/models/text_query.py +178 -0
- anyscale/sdk/anyscale_client/models/tracing_config.py +178 -0
- anyscale/sdk/anyscale_client/models/update_app_config.py +122 -0
- anyscale/sdk/anyscale_client/models/update_cloud.py +150 -0
- anyscale/sdk/anyscale_client/models/update_cluster.py +206 -0
- anyscale/sdk/anyscale_client/models/update_compute_template.py +146 -0
- anyscale/sdk/anyscale_client/models/update_compute_template_config.py +464 -0
- anyscale/sdk/anyscale_client/models/update_organization.py +123 -0
- anyscale/sdk/anyscale_client/models/update_project.py +150 -0
- anyscale/sdk/anyscale_client/models/update_session.py +150 -0
- anyscale/sdk/anyscale_client/models/user_service_access_types.py +100 -0
- anyscale/sdk/anyscale_client/models/ux_instance.py +468 -0
- anyscale/sdk/anyscale_client/models/validation_error.py +175 -0
- anyscale/sdk/anyscale_client/models/worker_node_type.py +404 -0
- anyscale/sdk/anyscale_client/rest.py +296 -0
- anyscale/sdk/anyscale_client/sdk.py +634 -0
- anyscale/service/__init__.py +168 -0
- anyscale/service/_private/service_sdk.py +702 -0
- anyscale/service/commands.py +261 -0
- anyscale/service/models.py +671 -0
- anyscale/shared_anyscale_utils/__init__.py +1 -0
- anyscale/shared_anyscale_utils/aws.py +153 -0
- anyscale/shared_anyscale_utils/bytes_util.py +10 -0
- anyscale/shared_anyscale_utils/conf.py +47 -0
- anyscale/shared_anyscale_utils/default_anyscale_aws.yaml +74 -0
- anyscale/shared_anyscale_utils/default_anyscale_gcp.yaml +80 -0
- anyscale/shared_anyscale_utils/headers.py +38 -0
- anyscale/shared_anyscale_utils/latest_ray_version.py +2 -0
- anyscale/shared_anyscale_utils/project.py +15 -0
- anyscale/shared_anyscale_utils/test_util.py +22 -0
- anyscale/shared_anyscale_utils/tests/__init__.py +1 -0
- anyscale/shared_anyscale_utils/tests/test_asyncio.py +41 -0
- anyscale/shared_anyscale_utils/tests/test_ray_semver.py +63 -0
- anyscale/shared_anyscale_utils/util.py +50 -0
- anyscale/shared_anyscale_utils/utils/__init__.py +2 -0
- anyscale/shared_anyscale_utils/utils/asyncio.py +120 -0
- anyscale/shared_anyscale_utils/utils/byod.py +40 -0
- anyscale/shared_anyscale_utils/utils/collections.py +33 -0
- anyscale/shared_anyscale_utils/utils/id_gen.py +147 -0
- anyscale/shared_anyscale_utils/utils/protected_string.py +89 -0
- anyscale/shared_anyscale_utils/utils/ray_semver.py +81 -0
- anyscale/snapshot.py +46 -0
- anyscale/tables.py +82 -0
- anyscale/util.py +1155 -0
- anyscale/utils/__init__.py +0 -0
- anyscale/utils/cli_version_check_util.py +63 -0
- anyscale/utils/cloud_update_utils.py +862 -0
- anyscale/utils/cloud_utils.py +317 -0
- anyscale/utils/cluster_debug.py +191 -0
- anyscale/utils/connect_helpers.py +155 -0
- anyscale/utils/deprecation_util.py +32 -0
- anyscale/utils/entity_arg_utils.py +43 -0
- anyscale/utils/env_utils.py +17 -0
- anyscale/utils/gcp_managed_setup_utils.py +888 -0
- anyscale/utils/gcp_utils.py +312 -0
- anyscale/utils/imports/__init__.py +0 -0
- anyscale/utils/imports/all.py +13 -0
- anyscale/utils/imports/azure.py +14 -0
- anyscale/utils/imports/gcp.py +59 -0
- anyscale/utils/logs_utils.py +141 -0
- anyscale/utils/name_utils.py +33 -0
- anyscale/utils/network_verification.py +153 -0
- anyscale/utils/ray_utils.py +128 -0
- anyscale/utils/ray_version_checker.py +48 -0
- anyscale/utils/ray_version_utils.py +53 -0
- anyscale/utils/runtime_env.py +487 -0
- anyscale/utils/s3.py +92 -0
- anyscale/utils/user_utils.py +17 -0
- anyscale/utils/workload_types.py +7 -0
- anyscale/utils/workspace_notification.py +39 -0
- anyscale/utils/workspace_utils.py +65 -0
- anyscale/version.py +1 -0
- anyscale/webterminal/__init__.py +0 -0
- anyscale/webterminal/bash-preexec.sh +370 -0
- anyscale/webterminal/command_persister.py +164 -0
- anyscale/webterminal/utils.py +176 -0
- anyscale/webterminal/webterminal.py +311 -0
- anyscale/workspace/__init__.py +270 -0
- anyscale/workspace/_private/workspace_sdk.py +737 -0
- anyscale/workspace/commands.py +472 -0
- anyscale/workspace/models.py +296 -0
- anyscale/workspace_utils.py +35 -0
- anyscale-0.24.86.dist-info/LICENSE +68 -0
- anyscale-0.24.86.dist-info/METADATA +82 -0
- anyscale-0.24.86.dist-info/NOTICE +6 -0
- anyscale-0.24.86.dist-info/RECORD +1131 -0
- anyscale-0.24.86.dist-info/WHEEL +5 -0
- anyscale-0.24.86.dist-info/entry_points.txt +2 -0
- anyscale-0.24.86.dist-info/top_level.txt +1 -0
|
@@ -0,0 +1,858 @@
|
|
|
1
|
+
import concurrent.futures
|
|
2
|
+
from contextlib import contextmanager
|
|
3
|
+
from datetime import datetime
|
|
4
|
+
from enum import Enum
|
|
5
|
+
import random
|
|
6
|
+
import time
|
|
7
|
+
from typing import Any, Callable, Dict, List, Optional, Set
|
|
8
|
+
|
|
9
|
+
from click import ClickException
|
|
10
|
+
from rich.console import Group
|
|
11
|
+
from rich.live import Live
|
|
12
|
+
from rich.markup import escape
|
|
13
|
+
from rich.panel import Panel
|
|
14
|
+
from rich.progress import (
|
|
15
|
+
BarColumn,
|
|
16
|
+
Progress,
|
|
17
|
+
TaskID,
|
|
18
|
+
TextColumn,
|
|
19
|
+
TimeElapsedColumn,
|
|
20
|
+
)
|
|
21
|
+
from rich.table import Table
|
|
22
|
+
|
|
23
|
+
from anyscale.cli_logger import LogsLogger
|
|
24
|
+
from anyscale.client.openapi_client.models import (
|
|
25
|
+
ApplyProductionServiceV2Model,
|
|
26
|
+
CloudAnalyticsEventName,
|
|
27
|
+
CloudProviders,
|
|
28
|
+
CreateExperimentalWorkspace,
|
|
29
|
+
ServiceEventCurrentState,
|
|
30
|
+
SessionState,
|
|
31
|
+
)
|
|
32
|
+
from anyscale.controllers.base_controller import BaseController
|
|
33
|
+
from anyscale.project import get_default_project
|
|
34
|
+
from anyscale.sdk.anyscale_client.models import (
|
|
35
|
+
ComputeNodeType,
|
|
36
|
+
ComputeTemplateQuery,
|
|
37
|
+
CreateClusterCompute,
|
|
38
|
+
CreateClusterComputeConfig,
|
|
39
|
+
ServiceConfig,
|
|
40
|
+
)
|
|
41
|
+
from anyscale.util import confirm, get_endpoint
|
|
42
|
+
from anyscale.utils.cloud_utils import CloudEventProducer
|
|
43
|
+
|
|
44
|
+
|
|
45
|
+
POLL_INTERVAL_SECONDS = 10
|
|
46
|
+
WORKSPACE_VERIFICATION_TIMEOUT_MINUTES = 10
|
|
47
|
+
SERVICE_VERIFICATION_TIMEOUT_MINUTES = 30 # for a single rollout
|
|
48
|
+
|
|
49
|
+
# default values for cluster compute config
|
|
50
|
+
MAXIMUM_UPTIME_MINUTES = 15
|
|
51
|
+
IDLE_TERMINATION_MINUTES = 5
|
|
52
|
+
HEAD_NODE_TYPE_AWS = "m5.xlarge" # on demand price ~$0.20 per hour
|
|
53
|
+
HEAD_NODE_TYPE_GCP = "n1-standard-2" # on demand price ~$0.11 per hour
|
|
54
|
+
CREATE_COMPUTE_CONFIG_TIMEOUT_SECONDS = 600 # 10 minutes
|
|
55
|
+
|
|
56
|
+
# Workspace verification will fail fast if any of the following logs are found
|
|
57
|
+
WORKSPACE_FAIL_FAST_MATCHING_LOGS = ["Failed to execute"]
|
|
58
|
+
|
|
59
|
+
|
|
60
|
+
class CloudFunctionalVerificationType(str, Enum):
|
|
61
|
+
WORKSPACE = "WORKSPACE"
|
|
62
|
+
SERVICE = "SERVICE"
|
|
63
|
+
|
|
64
|
+
|
|
65
|
+
class CloudFunctionalVerificationTask:
|
|
66
|
+
def __init__(self, task_id: TaskID, description: str):
|
|
67
|
+
self.task_id = task_id
|
|
68
|
+
self.description = description
|
|
69
|
+
self.succeeded = False
|
|
70
|
+
self.completed = False
|
|
71
|
+
|
|
72
|
+
def update(self, succeeded: bool, description: str, completed: bool = False):
|
|
73
|
+
self.description = description
|
|
74
|
+
self.succeeded = succeeded
|
|
75
|
+
self.completed = completed
|
|
76
|
+
|
|
77
|
+
|
|
78
|
+
class CloudFunctionalVerificationController(BaseController):
|
|
79
|
+
def __init__(
|
|
80
|
+
self,
|
|
81
|
+
cloud_event_producer: CloudEventProducer,
|
|
82
|
+
log: Optional[LogsLogger] = None,
|
|
83
|
+
initialize_auth_api_client: bool = True,
|
|
84
|
+
):
|
|
85
|
+
if log is None:
|
|
86
|
+
log = LogsLogger()
|
|
87
|
+
|
|
88
|
+
super().__init__(initialize_auth_api_client=initialize_auth_api_client)
|
|
89
|
+
self.cloud_event_producer = cloud_event_producer
|
|
90
|
+
self.log = log
|
|
91
|
+
|
|
92
|
+
self.event_log_num: Dict[CloudFunctionalVerificationType, int] = {
|
|
93
|
+
CloudFunctionalVerificationType.WORKSPACE: 0,
|
|
94
|
+
CloudFunctionalVerificationType.SERVICE: 0,
|
|
95
|
+
}
|
|
96
|
+
|
|
97
|
+
# Used for rich live console
|
|
98
|
+
self.step_progress: Dict[CloudFunctionalVerificationType, Progress] = {}
|
|
99
|
+
self.overall_progress: Dict[CloudFunctionalVerificationType, Progress] = {}
|
|
100
|
+
self.task_ids: Dict[CloudFunctionalVerificationType, TaskID] = {}
|
|
101
|
+
self.event_log_tables: Dict[CloudFunctionalVerificationType, Table] = {}
|
|
102
|
+
|
|
103
|
+
@staticmethod
|
|
104
|
+
def get_head_node_type(cloud_provider: CloudProviders) -> str:
|
|
105
|
+
"""
|
|
106
|
+
Get the default head node type for the given cloud provider.
|
|
107
|
+
"""
|
|
108
|
+
if cloud_provider == CloudProviders.AWS:
|
|
109
|
+
return HEAD_NODE_TYPE_AWS
|
|
110
|
+
elif cloud_provider == CloudProviders.GCP:
|
|
111
|
+
return HEAD_NODE_TYPE_GCP
|
|
112
|
+
raise ClickException(f"Unsupported cloud provider: {cloud_provider}")
|
|
113
|
+
|
|
114
|
+
def get_or_create_cluster_compute(
|
|
115
|
+
self, cloud_id: str, cloud_provider: CloudProviders
|
|
116
|
+
) -> str:
|
|
117
|
+
"""
|
|
118
|
+
Get or create a cluster compute for cloud functional verification
|
|
119
|
+
"""
|
|
120
|
+
cluster_compute_name = f"functional_verification_{cloud_id}"
|
|
121
|
+
cluster_compute_version = 1
|
|
122
|
+
|
|
123
|
+
cluster_computes = self.api_client.search_compute_templates_api_v2_compute_templates_search_post(
|
|
124
|
+
ComputeTemplateQuery(
|
|
125
|
+
orgwide=True,
|
|
126
|
+
name={"equals": cluster_compute_name},
|
|
127
|
+
include_anonymous=True,
|
|
128
|
+
version=cluster_compute_version,
|
|
129
|
+
)
|
|
130
|
+
).results
|
|
131
|
+
if len(cluster_computes) > 0:
|
|
132
|
+
return cluster_computes[0].id
|
|
133
|
+
|
|
134
|
+
head_node_instance_type = self.get_head_node_type(cloud_provider)
|
|
135
|
+
# no cluster compute found, create one
|
|
136
|
+
cluster_compute_config = CreateClusterComputeConfig(
|
|
137
|
+
cloud_id=cloud_id,
|
|
138
|
+
max_workers=0,
|
|
139
|
+
allowed_azs=["any"],
|
|
140
|
+
head_node_type=ComputeNodeType(
|
|
141
|
+
name="head_node_type", instance_type=head_node_instance_type,
|
|
142
|
+
),
|
|
143
|
+
maximum_uptime_minutes=MAXIMUM_UPTIME_MINUTES,
|
|
144
|
+
idle_termination_minutes=IDLE_TERMINATION_MINUTES,
|
|
145
|
+
worker_node_types=[],
|
|
146
|
+
)
|
|
147
|
+
if cloud_provider == CloudProviders.AWS:
|
|
148
|
+
cluster_compute_config.aws_advanced_configurations_json = {
|
|
149
|
+
"TagSpecifications": [
|
|
150
|
+
{
|
|
151
|
+
"ResourceType": "instance",
|
|
152
|
+
"Tags": [
|
|
153
|
+
{"Key": "cloud_functional_verification", "Value": cloud_id,}
|
|
154
|
+
],
|
|
155
|
+
}
|
|
156
|
+
]
|
|
157
|
+
}
|
|
158
|
+
elif cloud_provider == CloudProviders.GCP:
|
|
159
|
+
cluster_compute_config.gcp_advanced_configurations_json = {
|
|
160
|
+
"instance_properties": {
|
|
161
|
+
"labels": {"cloud_functional_verification": cloud_id},
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
|
|
165
|
+
# Add retries with exponential backoff here to handle the case where
|
|
166
|
+
# the cloud admin zone is not ready yet.
|
|
167
|
+
start_time = time.time()
|
|
168
|
+
end_time = start_time + CREATE_COMPUTE_CONFIG_TIMEOUT_SECONDS
|
|
169
|
+
|
|
170
|
+
delay = 1
|
|
171
|
+
max_delay = 64
|
|
172
|
+
while time.time() < end_time:
|
|
173
|
+
try:
|
|
174
|
+
cluster_compute = self.anyscale_api_client.create_cluster_compute(
|
|
175
|
+
CreateClusterCompute(
|
|
176
|
+
name=cluster_compute_name,
|
|
177
|
+
config=cluster_compute_config,
|
|
178
|
+
anonymous=True,
|
|
179
|
+
)
|
|
180
|
+
).result
|
|
181
|
+
return cluster_compute.id
|
|
182
|
+
except ClickException:
|
|
183
|
+
# Retry if the cloud admin zone is not ready yet
|
|
184
|
+
pass
|
|
185
|
+
delay = min(delay, max_delay)
|
|
186
|
+
# Add jitter to avoid synchronized retries
|
|
187
|
+
jitter = random.uniform(0, delay / 2)
|
|
188
|
+
time.sleep(delay + jitter)
|
|
189
|
+
delay *= 2 # exponential backoff
|
|
190
|
+
raise ClickException(
|
|
191
|
+
"Timed out waiting for compute config to be created. Please try again."
|
|
192
|
+
)
|
|
193
|
+
|
|
194
|
+
def _prepare_verification(self, cloud_id: str, cloud_provider: CloudProviders):
|
|
195
|
+
"""
|
|
196
|
+
Generate the required parameters for cloud functional verification.
|
|
197
|
+
"""
|
|
198
|
+
|
|
199
|
+
cluster_env_build_id = self.get_default_cluster_env_build_id()
|
|
200
|
+
|
|
201
|
+
project_id = get_default_project(
|
|
202
|
+
self.api_client, self.anyscale_api_client, parent_cloud_id=cloud_id
|
|
203
|
+
).id
|
|
204
|
+
|
|
205
|
+
cluster_compute_id = self.get_or_create_cluster_compute(
|
|
206
|
+
cloud_id, cloud_provider
|
|
207
|
+
)
|
|
208
|
+
|
|
209
|
+
return cluster_compute_id, cluster_env_build_id, project_id
|
|
210
|
+
|
|
211
|
+
def get_default_cluster_env_build_id(self):
|
|
212
|
+
try:
|
|
213
|
+
cluster_env_list = self.api_client.list_application_templates_api_v2_application_templates_get(
|
|
214
|
+
defaults_first=True
|
|
215
|
+
).results
|
|
216
|
+
if len(cluster_env_list) == 0:
|
|
217
|
+
raise ClickException("No cluster environments found")
|
|
218
|
+
except Exception as e: # noqa: BLE001
|
|
219
|
+
raise ClickException(f"Failed to list cluster environments: {e}")
|
|
220
|
+
|
|
221
|
+
for cluster_env in cluster_env_list:
|
|
222
|
+
if cluster_env.is_default:
|
|
223
|
+
return cluster_env.latest_build.id
|
|
224
|
+
return cluster_env_list[0].latest_build.id
|
|
225
|
+
|
|
226
|
+
@contextmanager
|
|
227
|
+
def _create_task(self, function: CloudFunctionalVerificationType, description: str):
|
|
228
|
+
"""
|
|
229
|
+
Create a task on the console for cloud functional verification
|
|
230
|
+
"""
|
|
231
|
+
task_id = self.step_progress[function].add_task(description)
|
|
232
|
+
task = CloudFunctionalVerificationTask(task_id, description)
|
|
233
|
+
try:
|
|
234
|
+
yield task
|
|
235
|
+
finally:
|
|
236
|
+
self._update_console(
|
|
237
|
+
task.succeeded, function, task_id, task.description, task.completed
|
|
238
|
+
)
|
|
239
|
+
|
|
240
|
+
def create_workspace(self, cloud_id: str, cloud_provider: CloudProviders):
|
|
241
|
+
"""
|
|
242
|
+
Create a workspace for cloud functional verification
|
|
243
|
+
"""
|
|
244
|
+
(
|
|
245
|
+
cluster_compute_id,
|
|
246
|
+
cluster_env_build_id,
|
|
247
|
+
project_id,
|
|
248
|
+
) = self._prepare_verification(cloud_id, cloud_provider)
|
|
249
|
+
|
|
250
|
+
create_workspace_arg = CreateExperimentalWorkspace(
|
|
251
|
+
name=f"fxnvrf_{cloud_id}_{datetime.utcnow().strftime('%Y%m%d%H%M%S')}",
|
|
252
|
+
description=f"workspace for cloud {cloud_id} functional verification",
|
|
253
|
+
project_id=project_id,
|
|
254
|
+
cloud_id=cloud_id,
|
|
255
|
+
compute_config_id=cluster_compute_id,
|
|
256
|
+
cluster_environment_build_id=cluster_env_build_id,
|
|
257
|
+
idle_timeout_minutes=IDLE_TERMINATION_MINUTES,
|
|
258
|
+
)
|
|
259
|
+
|
|
260
|
+
workspace = self.api_client.create_workspace_api_v2_experimental_workspaces_post(
|
|
261
|
+
create_workspace_arg
|
|
262
|
+
).result
|
|
263
|
+
|
|
264
|
+
return workspace
|
|
265
|
+
|
|
266
|
+
def verify_workspace(self, cloud_id: str, cloud_provider: CloudProviders) -> bool:
|
|
267
|
+
"""
|
|
268
|
+
Verifies that the workspace is setup correctly on the given cloud.
|
|
269
|
+
"""
|
|
270
|
+
# Create workspace
|
|
271
|
+
with self._create_task(
|
|
272
|
+
CloudFunctionalVerificationType.WORKSPACE, "Creating workspace..."
|
|
273
|
+
) as create_workspace_task:
|
|
274
|
+
try:
|
|
275
|
+
workspace = self.create_workspace(cloud_id, cloud_provider)
|
|
276
|
+
except ClickException as e:
|
|
277
|
+
self.cloud_event_producer.produce(
|
|
278
|
+
CloudAnalyticsEventName.WORKSPACE_FUNCTIONAL_VERIFIED,
|
|
279
|
+
succeeded=False,
|
|
280
|
+
internal_error=str(e),
|
|
281
|
+
)
|
|
282
|
+
create_workspace_task.update(
|
|
283
|
+
False, f"[bold red]Failed to create workspace: {e}"
|
|
284
|
+
)
|
|
285
|
+
return False
|
|
286
|
+
url = get_endpoint(f"/workspaces/{workspace.id}/{workspace.cluster_id}")
|
|
287
|
+
create_workspace_task.update(
|
|
288
|
+
True, f"[bold green]Workspace created at {url}"
|
|
289
|
+
)
|
|
290
|
+
|
|
291
|
+
# Wait until workspace is active
|
|
292
|
+
def get_workspace_status(workspace_id):
|
|
293
|
+
return self.api_client.get_workspace_api_v2_experimental_workspaces_workspace_id_get(
|
|
294
|
+
workspace_id
|
|
295
|
+
).result.state
|
|
296
|
+
|
|
297
|
+
allowed_status_set = {
|
|
298
|
+
SessionState.RUNNING,
|
|
299
|
+
SessionState.STARTINGUP,
|
|
300
|
+
SessionState.AWAITINGSTARTUP,
|
|
301
|
+
SessionState.UPDATING,
|
|
302
|
+
}
|
|
303
|
+
|
|
304
|
+
with self._create_task(
|
|
305
|
+
CloudFunctionalVerificationType.WORKSPACE,
|
|
306
|
+
"Waiting for workspace to become active...",
|
|
307
|
+
) as wait_task:
|
|
308
|
+
try:
|
|
309
|
+
self.poll_until_active(
|
|
310
|
+
CloudFunctionalVerificationType.WORKSPACE,
|
|
311
|
+
workspace,
|
|
312
|
+
get_workspace_status,
|
|
313
|
+
SessionState.RUNNING,
|
|
314
|
+
allowed_status_set,
|
|
315
|
+
wait_task,
|
|
316
|
+
WORKSPACE_VERIFICATION_TIMEOUT_MINUTES,
|
|
317
|
+
)
|
|
318
|
+
except ClickException as e:
|
|
319
|
+
self.cloud_event_producer.produce(
|
|
320
|
+
CloudAnalyticsEventName.WORKSPACE_FUNCTIONAL_VERIFIED,
|
|
321
|
+
succeeded=False,
|
|
322
|
+
internal_error=str(e),
|
|
323
|
+
)
|
|
324
|
+
wait_task.update(
|
|
325
|
+
False,
|
|
326
|
+
f"[bold red]Error: {e}. Please click on the URL above to check the logs.",
|
|
327
|
+
)
|
|
328
|
+
return False
|
|
329
|
+
wait_task.update(True, "[bold green]Workspace is active.")
|
|
330
|
+
|
|
331
|
+
# terminate workspace
|
|
332
|
+
with self._create_task(
|
|
333
|
+
CloudFunctionalVerificationType.WORKSPACE, "Terminating workspace..."
|
|
334
|
+
) as terminate_workspace_task:
|
|
335
|
+
try:
|
|
336
|
+
# terminate the cluster leads to workspace termination
|
|
337
|
+
self.anyscale_api_client.terminate_cluster(workspace.cluster_id, {})
|
|
338
|
+
allowed_status_set = {
|
|
339
|
+
SessionState.TERMINATED,
|
|
340
|
+
SessionState.TERMINATING,
|
|
341
|
+
SessionState.RUNNING,
|
|
342
|
+
}
|
|
343
|
+
self.poll_until_active(
|
|
344
|
+
CloudFunctionalVerificationType.WORKSPACE,
|
|
345
|
+
workspace,
|
|
346
|
+
get_workspace_status,
|
|
347
|
+
SessionState.TERMINATED,
|
|
348
|
+
allowed_status_set,
|
|
349
|
+
terminate_workspace_task,
|
|
350
|
+
WORKSPACE_VERIFICATION_TIMEOUT_MINUTES,
|
|
351
|
+
)
|
|
352
|
+
# archive workspace
|
|
353
|
+
self.api_client.delete_workspace_api_v2_experimental_workspaces_workspace_id_delete(
|
|
354
|
+
workspace.id
|
|
355
|
+
)
|
|
356
|
+
except ClickException as e:
|
|
357
|
+
terminate_workspace_task.update(
|
|
358
|
+
False,
|
|
359
|
+
f"[bold red]Failed to terminate workspace: {e}",
|
|
360
|
+
completed=True,
|
|
361
|
+
)
|
|
362
|
+
self.cloud_event_producer.produce(
|
|
363
|
+
CloudAnalyticsEventName.WORKSPACE_FUNCTIONAL_VERIFIED,
|
|
364
|
+
succeeded=False,
|
|
365
|
+
internal_error=str(e),
|
|
366
|
+
)
|
|
367
|
+
return False
|
|
368
|
+
terminate_workspace_task.update(
|
|
369
|
+
True, "[bold green]Workspace terminated.", completed=True
|
|
370
|
+
)
|
|
371
|
+
|
|
372
|
+
self.cloud_event_producer.produce(
|
|
373
|
+
CloudAnalyticsEventName.WORKSPACE_FUNCTIONAL_VERIFIED, succeeded=True,
|
|
374
|
+
)
|
|
375
|
+
return True
|
|
376
|
+
|
|
377
|
+
def rollout_service(
|
|
378
|
+
self,
|
|
379
|
+
cloud_id: str,
|
|
380
|
+
cloud_provider: CloudProviders,
|
|
381
|
+
*,
|
|
382
|
+
version: str = "v1",
|
|
383
|
+
service_name: Optional[str] = None,
|
|
384
|
+
canary_percent: Optional[int] = None,
|
|
385
|
+
):
|
|
386
|
+
"""
|
|
387
|
+
Roll out a service for cloud functional verification
|
|
388
|
+
"""
|
|
389
|
+
(
|
|
390
|
+
cluster_compute_id,
|
|
391
|
+
cluster_env_build_id,
|
|
392
|
+
project_id,
|
|
393
|
+
) = self._prepare_verification(cloud_id, cloud_provider)
|
|
394
|
+
|
|
395
|
+
service_name = (
|
|
396
|
+
service_name
|
|
397
|
+
or f"fxnvrf_{cloud_id}_{datetime.utcnow().strftime('%Y%m%d%H%M%S')}"
|
|
398
|
+
)
|
|
399
|
+
|
|
400
|
+
service_config = ApplyProductionServiceV2Model(
|
|
401
|
+
name=service_name,
|
|
402
|
+
description=f"service for cloud {cloud_id} functional verification",
|
|
403
|
+
project_id=project_id,
|
|
404
|
+
ray_serve_config={
|
|
405
|
+
"applications": [
|
|
406
|
+
{
|
|
407
|
+
"import_path": "serve_hello:entrypoint",
|
|
408
|
+
"runtime_env": {
|
|
409
|
+
"working_dir": "https://github.com/anyscale/docs_examples/archive/refs/heads/main.zip",
|
|
410
|
+
"env_vars": {
|
|
411
|
+
"SERVE_RESPONSE_MESSAGE": f"cloud functional verification {version}",
|
|
412
|
+
},
|
|
413
|
+
},
|
|
414
|
+
}
|
|
415
|
+
],
|
|
416
|
+
},
|
|
417
|
+
version=version,
|
|
418
|
+
build_id=cluster_env_build_id,
|
|
419
|
+
compute_config_id=cluster_compute_id,
|
|
420
|
+
config=ServiceConfig(
|
|
421
|
+
max_uptime_timeout_sec=SERVICE_VERIFICATION_TIMEOUT_MINUTES * 2 * 60,
|
|
422
|
+
),
|
|
423
|
+
canary_percent=canary_percent,
|
|
424
|
+
)
|
|
425
|
+
|
|
426
|
+
# Rollout service
|
|
427
|
+
service = self.api_client.apply_service_api_v2_services_v2_apply_put(
|
|
428
|
+
service_config
|
|
429
|
+
).result
|
|
430
|
+
|
|
431
|
+
return service
|
|
432
|
+
|
|
433
|
+
def verify_service( # noqa: PLR0911
|
|
434
|
+
self, cloud_id: str, cloud_provider: CloudProviders
|
|
435
|
+
) -> bool:
|
|
436
|
+
"""
|
|
437
|
+
Verifies that the service can be deployed and upgraded on the given cloud.
|
|
438
|
+
"""
|
|
439
|
+
# Deploy service
|
|
440
|
+
with self._create_task(
|
|
441
|
+
CloudFunctionalVerificationType.SERVICE, "Deploying service..."
|
|
442
|
+
) as deploy_service_task:
|
|
443
|
+
try:
|
|
444
|
+
service = self.rollout_service(cloud_id, cloud_provider, version="v1")
|
|
445
|
+
except ClickException as e:
|
|
446
|
+
self.cloud_event_producer.produce(
|
|
447
|
+
CloudAnalyticsEventName.SERVICE_FUNCTIONAL_VERIFIED,
|
|
448
|
+
succeeded=False,
|
|
449
|
+
internal_error=str(e),
|
|
450
|
+
)
|
|
451
|
+
deploy_service_task.update(
|
|
452
|
+
False, f"[bold red]Failed to deploy service: {e}"
|
|
453
|
+
)
|
|
454
|
+
return False
|
|
455
|
+
url = get_endpoint(f"/services/{service.id}")
|
|
456
|
+
deploy_service_task.update(True, f"[bold green]Service deployed at {url}")
|
|
457
|
+
|
|
458
|
+
# Wait until service is active
|
|
459
|
+
def get_service_status(service_id):
|
|
460
|
+
return self.api_client.get_service_api_v2_services_v2_service_id_get(
|
|
461
|
+
service_id
|
|
462
|
+
).result.current_state
|
|
463
|
+
|
|
464
|
+
# We add all service states into the set
|
|
465
|
+
allowed_status_set = set(ServiceEventCurrentState.allowable_values)
|
|
466
|
+
|
|
467
|
+
with self._create_task(
|
|
468
|
+
CloudFunctionalVerificationType.SERVICE,
|
|
469
|
+
"Waiting for service to become active...",
|
|
470
|
+
) as wait_task:
|
|
471
|
+
try:
|
|
472
|
+
self.poll_until_active(
|
|
473
|
+
CloudFunctionalVerificationType.SERVICE,
|
|
474
|
+
service,
|
|
475
|
+
get_service_status,
|
|
476
|
+
ServiceEventCurrentState.RUNNING,
|
|
477
|
+
allowed_status_set,
|
|
478
|
+
wait_task,
|
|
479
|
+
SERVICE_VERIFICATION_TIMEOUT_MINUTES,
|
|
480
|
+
)
|
|
481
|
+
except ClickException as e:
|
|
482
|
+
self.cloud_event_producer.produce(
|
|
483
|
+
CloudAnalyticsEventName.SERVICE_FUNCTIONAL_VERIFIED,
|
|
484
|
+
succeeded=False,
|
|
485
|
+
internal_error=str(e),
|
|
486
|
+
)
|
|
487
|
+
wait_task.update(
|
|
488
|
+
False,
|
|
489
|
+
f"[bold red]Error: {e}. Please click on the URL above to check the logs.",
|
|
490
|
+
)
|
|
491
|
+
return False
|
|
492
|
+
wait_task.update(True, "[bold green]Service is active.")
|
|
493
|
+
|
|
494
|
+
# Upgrade service
|
|
495
|
+
with self._create_task(
|
|
496
|
+
CloudFunctionalVerificationType.SERVICE, "Upgrading service..."
|
|
497
|
+
) as upgrade_service_task:
|
|
498
|
+
try:
|
|
499
|
+
service = self.rollout_service(
|
|
500
|
+
cloud_id,
|
|
501
|
+
cloud_provider,
|
|
502
|
+
version="v2",
|
|
503
|
+
service_name=service.name,
|
|
504
|
+
canary_percent=100,
|
|
505
|
+
)
|
|
506
|
+
except ClickException as e:
|
|
507
|
+
self.cloud_event_producer.produce(
|
|
508
|
+
CloudAnalyticsEventName.SERVICE_FUNCTIONAL_VERIFIED,
|
|
509
|
+
succeeded=False,
|
|
510
|
+
internal_error=str(e),
|
|
511
|
+
)
|
|
512
|
+
upgrade_service_task.update(
|
|
513
|
+
False, f"[bold red]Failed to upgrade service: {e}"
|
|
514
|
+
)
|
|
515
|
+
return False
|
|
516
|
+
|
|
517
|
+
try:
|
|
518
|
+
self.poll_until_active(
|
|
519
|
+
CloudFunctionalVerificationType.SERVICE,
|
|
520
|
+
service,
|
|
521
|
+
get_service_status,
|
|
522
|
+
ServiceEventCurrentState.RUNNING,
|
|
523
|
+
allowed_status_set,
|
|
524
|
+
upgrade_service_task,
|
|
525
|
+
SERVICE_VERIFICATION_TIMEOUT_MINUTES,
|
|
526
|
+
)
|
|
527
|
+
except ClickException as e:
|
|
528
|
+
self.cloud_event_producer.produce(
|
|
529
|
+
CloudAnalyticsEventName.SERVICE_FUNCTIONAL_VERIFIED,
|
|
530
|
+
succeeded=False,
|
|
531
|
+
internal_error=str(e),
|
|
532
|
+
)
|
|
533
|
+
upgrade_service_task.update(
|
|
534
|
+
False,
|
|
535
|
+
f"[bold red]Error: {e}. Please click on the URL above to check the logs.",
|
|
536
|
+
)
|
|
537
|
+
return False
|
|
538
|
+
upgrade_service_task.update(
|
|
539
|
+
True, "[bold green]Service upgraded successfully."
|
|
540
|
+
)
|
|
541
|
+
|
|
542
|
+
# Terminate service
|
|
543
|
+
with self._create_task(
|
|
544
|
+
CloudFunctionalVerificationType.SERVICE, "Terminating service..."
|
|
545
|
+
) as terminate_service_task:
|
|
546
|
+
try:
|
|
547
|
+
self.api_client.terminate_service_api_v2_services_v2_service_id_terminate_post(
|
|
548
|
+
service.id
|
|
549
|
+
)
|
|
550
|
+
self.poll_until_active(
|
|
551
|
+
CloudFunctionalVerificationType.SERVICE,
|
|
552
|
+
service,
|
|
553
|
+
get_service_status,
|
|
554
|
+
ServiceEventCurrentState.TERMINATED,
|
|
555
|
+
allowed_status_set,
|
|
556
|
+
terminate_service_task,
|
|
557
|
+
SERVICE_VERIFICATION_TIMEOUT_MINUTES,
|
|
558
|
+
)
|
|
559
|
+
except ClickException as e:
|
|
560
|
+
terminate_service_task.update(
|
|
561
|
+
False,
|
|
562
|
+
f"[bold red]Failed to terminate service: {e}",
|
|
563
|
+
completed=True,
|
|
564
|
+
)
|
|
565
|
+
self.cloud_event_producer.produce(
|
|
566
|
+
CloudAnalyticsEventName.SERVICE_FUNCTIONAL_VERIFIED,
|
|
567
|
+
succeeded=False,
|
|
568
|
+
internal_error=str(e),
|
|
569
|
+
)
|
|
570
|
+
return False
|
|
571
|
+
terminate_service_task.update(
|
|
572
|
+
True, "[bold green]Service terminated.", completed=True
|
|
573
|
+
)
|
|
574
|
+
|
|
575
|
+
self.cloud_event_producer.produce(
|
|
576
|
+
CloudAnalyticsEventName.SERVICE_FUNCTIONAL_VERIFIED, succeeded=True,
|
|
577
|
+
)
|
|
578
|
+
return True
|
|
579
|
+
|
|
580
|
+
def poll_until_active( # noqa: PLR0913
|
|
581
|
+
self,
|
|
582
|
+
function_type: CloudFunctionalVerificationType,
|
|
583
|
+
function: Any,
|
|
584
|
+
get_current_status: Callable[[str], Any],
|
|
585
|
+
goal_status: Any,
|
|
586
|
+
allowed_status_set: Set[Any],
|
|
587
|
+
wait_task: CloudFunctionalVerificationTask,
|
|
588
|
+
timeout_minutes: int,
|
|
589
|
+
) -> bool:
|
|
590
|
+
"""
|
|
591
|
+
Polling until it is active.
|
|
592
|
+
"""
|
|
593
|
+
start_time = time.time()
|
|
594
|
+
end_time = start_time + timeout_minutes * 60
|
|
595
|
+
while time.time() < end_time:
|
|
596
|
+
time.sleep(POLL_INTERVAL_SECONDS)
|
|
597
|
+
try:
|
|
598
|
+
current_status = get_current_status(function.id)
|
|
599
|
+
succeeded = self._render_event_log(function_type, function)
|
|
600
|
+
if not succeeded:
|
|
601
|
+
raise ClickException(
|
|
602
|
+
f"{function_type.capitalize()} verification failed! Please check the logs and terminate the {function_type.lower()} manually."
|
|
603
|
+
)
|
|
604
|
+
except ClickException as e:
|
|
605
|
+
raise ClickException(
|
|
606
|
+
f"Failed to get {function_type.lower()} status: {e}"
|
|
607
|
+
) from None
|
|
608
|
+
self._update_task_in_step_progress(
|
|
609
|
+
function_type,
|
|
610
|
+
wait_task.task_id,
|
|
611
|
+
f"{wait_task.description} [{time.strftime('%H:%M:%S', time.localtime())}] Current status: {current_status}",
|
|
612
|
+
)
|
|
613
|
+
if current_status == goal_status:
|
|
614
|
+
return True
|
|
615
|
+
if current_status not in allowed_status_set:
|
|
616
|
+
raise ClickException(
|
|
617
|
+
f"{function_type.capitalize()} is in an unexpected state: {current_status}"
|
|
618
|
+
)
|
|
619
|
+
raise ClickException(
|
|
620
|
+
f"Timed out waiting for {function_type.lower()} to become active"
|
|
621
|
+
)
|
|
622
|
+
|
|
623
|
+
def verify(
|
|
624
|
+
self,
|
|
625
|
+
function: CloudFunctionalVerificationType,
|
|
626
|
+
cloud_id: str,
|
|
627
|
+
cloud_provider: CloudProviders,
|
|
628
|
+
) -> bool:
|
|
629
|
+
"""
|
|
630
|
+
Kick off a single functional verification given the function type.
|
|
631
|
+
"""
|
|
632
|
+
if function == CloudFunctionalVerificationType.WORKSPACE:
|
|
633
|
+
return self.verify_workspace(cloud_id, cloud_provider)
|
|
634
|
+
elif function == CloudFunctionalVerificationType.SERVICE:
|
|
635
|
+
return self.verify_service(cloud_id, cloud_provider)
|
|
636
|
+
return False
|
|
637
|
+
|
|
638
|
+
def _update_console(
|
|
639
|
+
self,
|
|
640
|
+
succeeded: bool,
|
|
641
|
+
function: CloudFunctionalVerificationType,
|
|
642
|
+
task_id: TaskID,
|
|
643
|
+
description: str,
|
|
644
|
+
completed: bool = False,
|
|
645
|
+
):
|
|
646
|
+
"""
|
|
647
|
+
Update the console based on the verification result
|
|
648
|
+
"""
|
|
649
|
+
self._update_overall_progress(succeeded, function, completed)
|
|
650
|
+
self._finish_task_in_step_progress(function, task_id, description)
|
|
651
|
+
|
|
652
|
+
def _update_task_in_step_progress(
|
|
653
|
+
self,
|
|
654
|
+
function: CloudFunctionalVerificationType,
|
|
655
|
+
task_id: TaskID,
|
|
656
|
+
description: str,
|
|
657
|
+
) -> None:
|
|
658
|
+
"""
|
|
659
|
+
Update the task description in step progress
|
|
660
|
+
"""
|
|
661
|
+
self.step_progress[function].update(task_id, description=description)
|
|
662
|
+
|
|
663
|
+
def _finish_task_in_step_progress(
|
|
664
|
+
self,
|
|
665
|
+
function: CloudFunctionalVerificationType,
|
|
666
|
+
task_id: TaskID,
|
|
667
|
+
description: str,
|
|
668
|
+
) -> None:
|
|
669
|
+
"""
|
|
670
|
+
Finish a task in step progress and update the description
|
|
671
|
+
"""
|
|
672
|
+
self.step_progress[function].stop_task(task_id)
|
|
673
|
+
self.step_progress[function].update(task_id, description=description)
|
|
674
|
+
|
|
675
|
+
def _update_overall_progress(
|
|
676
|
+
self,
|
|
677
|
+
verification_result: bool,
|
|
678
|
+
function: CloudFunctionalVerificationType,
|
|
679
|
+
completed: bool = False,
|
|
680
|
+
) -> None:
|
|
681
|
+
"""
|
|
682
|
+
Update overall progress based on the verification result
|
|
683
|
+
"""
|
|
684
|
+
if verification_result:
|
|
685
|
+
self.overall_progress[function].advance(self.task_ids[function], 1)
|
|
686
|
+
if completed:
|
|
687
|
+
self.overall_progress[function].update(
|
|
688
|
+
self.task_ids[function],
|
|
689
|
+
description=f"[bold green]{function.capitalize()} verification succeeded!",
|
|
690
|
+
)
|
|
691
|
+
else:
|
|
692
|
+
self.overall_progress[function].stop_task(self.task_ids[function])
|
|
693
|
+
self.overall_progress[function].update(
|
|
694
|
+
self.task_ids[function],
|
|
695
|
+
description=f"[bold red]{function.capitalize()} verification failed. Please check the logs and terminate the {function.lower()} manually.",
|
|
696
|
+
)
|
|
697
|
+
|
|
698
|
+
def _format_service_event_log(self, event_log: Any) -> str:
|
|
699
|
+
description = event_log.message.replace("\n", ". ") if event_log.message else ""
|
|
700
|
+
if event_log.event_type:
|
|
701
|
+
return f"[bold cyan]{event_log.created_at}[/bold cyan] [bold blue]{event_log.event_type}[/bold blue] [bold green]{description}[/bold green]"
|
|
702
|
+
else:
|
|
703
|
+
return f"[bold cyan]{event_log.created_at}[/bold cyan] [bold green]{description}[/bold green]"
|
|
704
|
+
|
|
705
|
+
def _render_event_log(
|
|
706
|
+
self, function_type: CloudFunctionalVerificationType, function: Any
|
|
707
|
+
) -> bool:
|
|
708
|
+
if function_type == CloudFunctionalVerificationType.WORKSPACE:
|
|
709
|
+
workspace_log = self.api_client.get_startup_logs_api_v2_sessions_session_id_startup_logs_get(
|
|
710
|
+
function.cluster_id, self.event_log_num[function_type], 100000000
|
|
711
|
+
).result
|
|
712
|
+
if workspace_log.num_lines == self.event_log_num[function_type]:
|
|
713
|
+
# no new logs
|
|
714
|
+
return True
|
|
715
|
+
|
|
716
|
+
# Output logs
|
|
717
|
+
self.event_log_tables[function_type].add_row(escape(workspace_log.lines))
|
|
718
|
+
self.event_log_num[function_type] = workspace_log.num_lines
|
|
719
|
+
|
|
720
|
+
# fail fast if any of the following logs are found
|
|
721
|
+
return not any(
|
|
722
|
+
failure_log in workspace_log.lines
|
|
723
|
+
for failure_log in WORKSPACE_FAIL_FAST_MATCHING_LOGS
|
|
724
|
+
)
|
|
725
|
+
elif function_type == CloudFunctionalVerificationType.SERVICE:
|
|
726
|
+
service_events = self.api_client.get_service_events_api_v2_services_v2_service_id_events_get(
|
|
727
|
+
function.id
|
|
728
|
+
).results
|
|
729
|
+
starting_pos = len(service_events) - self.event_log_num[function_type] - 1
|
|
730
|
+
for idx in range(starting_pos, -1, -1):
|
|
731
|
+
self.event_log_tables[function_type].add_row(
|
|
732
|
+
self._format_service_event_log(service_events[idx])
|
|
733
|
+
)
|
|
734
|
+
self.event_log_num[function_type] = len(service_events)
|
|
735
|
+
return True
|
|
736
|
+
else:
|
|
737
|
+
# we should never enter this branch
|
|
738
|
+
return False
|
|
739
|
+
|
|
740
|
+
def get_live_console(
|
|
741
|
+
self, functions_to_verify: List[CloudFunctionalVerificationType]
|
|
742
|
+
) -> Live:
|
|
743
|
+
"""
|
|
744
|
+
Get a live console for cloud functional verification.
|
|
745
|
+
|
|
746
|
+
Each functional verification contains 3 components:
|
|
747
|
+
1. step progress panel: shows the progress bars for each functional verification step
|
|
748
|
+
2. event log table: event logs for each functional verification
|
|
749
|
+
3. overall progress: overall progress bar for the functional verification
|
|
750
|
+
"""
|
|
751
|
+
progress_group = []
|
|
752
|
+
steps = {
|
|
753
|
+
CloudFunctionalVerificationType.WORKSPACE: 3,
|
|
754
|
+
CloudFunctionalVerificationType.SERVICE: 4,
|
|
755
|
+
}
|
|
756
|
+
for function in functions_to_verify:
|
|
757
|
+
progress_table = Table.grid(expand=True)
|
|
758
|
+
step_progress = Progress(
|
|
759
|
+
TimeElapsedColumn(), TextColumn("{task.description}"),
|
|
760
|
+
)
|
|
761
|
+
self.step_progress[function] = step_progress
|
|
762
|
+
event_log_table = Table(box=None)
|
|
763
|
+
self.event_log_tables[function] = event_log_table
|
|
764
|
+
progress_table.add_row(
|
|
765
|
+
Panel(step_progress, title=f"{function.lower()} verification")
|
|
766
|
+
)
|
|
767
|
+
progress_table.add_row(
|
|
768
|
+
Panel(event_log_table, title=f"{function.lower()} event logs")
|
|
769
|
+
)
|
|
770
|
+
|
|
771
|
+
progress_group.append(progress_table)
|
|
772
|
+
|
|
773
|
+
overall_progress = Progress(
|
|
774
|
+
TimeElapsedColumn(), BarColumn(), TextColumn("{task.description}")
|
|
775
|
+
)
|
|
776
|
+
self.overall_progress[function] = overall_progress
|
|
777
|
+
progress_group.append(overall_progress)
|
|
778
|
+
task_id = overall_progress.add_task("", total=steps[function])
|
|
779
|
+
overall_progress.update(
|
|
780
|
+
task_id, description=f"[bold #AAAAAA]Verifying {function.lower()}..."
|
|
781
|
+
)
|
|
782
|
+
self.task_ids[function] = task_id
|
|
783
|
+
return Live(Group(*progress_group))
|
|
784
|
+
|
|
785
|
+
def start_verification(
|
|
786
|
+
self,
|
|
787
|
+
cloud_id: str,
|
|
788
|
+
cloud_provider: CloudProviders,
|
|
789
|
+
functions_to_verify: List[CloudFunctionalVerificationType],
|
|
790
|
+
yes: bool = False,
|
|
791
|
+
) -> bool:
|
|
792
|
+
"""
|
|
793
|
+
Starts cloud functional verification
|
|
794
|
+
"""
|
|
795
|
+
with self.log.spinner("Starting functional verification..."):
|
|
796
|
+
self._prepare_verification(cloud_id, cloud_provider)
|
|
797
|
+
|
|
798
|
+
self.log.info(
|
|
799
|
+
f"Functional verification for {', '.join(functions_to_verify)} is about to begin."
|
|
800
|
+
)
|
|
801
|
+
|
|
802
|
+
confirmation_message = [
|
|
803
|
+
f"It will spin up one {self.get_head_node_type(cloud_provider)} instance for each function.",
|
|
804
|
+
]
|
|
805
|
+
|
|
806
|
+
if CloudFunctionalVerificationType.WORKSPACE in functions_to_verify:
|
|
807
|
+
confirmation_message.append(
|
|
808
|
+
"Workspace verification takes about 5 minutes. "
|
|
809
|
+
"The approximate cost from the cloud provider is about $0.02 (it varies from cloud provider and region)."
|
|
810
|
+
)
|
|
811
|
+
|
|
812
|
+
if CloudFunctionalVerificationType.SERVICE in functions_to_verify:
|
|
813
|
+
service_time_estimation = {
|
|
814
|
+
CloudProviders.AWS: 8,
|
|
815
|
+
CloudProviders.GCP: 20,
|
|
816
|
+
}
|
|
817
|
+
confirmation_message.append(
|
|
818
|
+
f"Service verification takes about {service_time_estimation[cloud_provider]} minutes. "
|
|
819
|
+
"The approximate cost from the cloud provider is about $0.10 (it varies from cloud provider and region)."
|
|
820
|
+
)
|
|
821
|
+
|
|
822
|
+
confirmation_message.append(
|
|
823
|
+
"The instances will be terminated after verification. Do you want to continue?"
|
|
824
|
+
)
|
|
825
|
+
|
|
826
|
+
confirm(
|
|
827
|
+
"\n".join(confirmation_message), yes,
|
|
828
|
+
)
|
|
829
|
+
|
|
830
|
+
verification_results: List[bool] = []
|
|
831
|
+
with self.get_live_console(
|
|
832
|
+
functions_to_verify
|
|
833
|
+
), concurrent.futures.ThreadPoolExecutor(
|
|
834
|
+
max_workers=len(CloudFunctionalVerificationType)
|
|
835
|
+
) as executor:
|
|
836
|
+
futures = {
|
|
837
|
+
executor.submit(
|
|
838
|
+
self.verify, function, cloud_id, cloud_provider
|
|
839
|
+
): function
|
|
840
|
+
for function in functions_to_verify
|
|
841
|
+
}
|
|
842
|
+
|
|
843
|
+
# Wait for all verifications to complete
|
|
844
|
+
for future in concurrent.futures.as_completed(futures):
|
|
845
|
+
function = futures[future]
|
|
846
|
+
try:
|
|
847
|
+
verification_result = future.result()
|
|
848
|
+
verification_results.append(verification_result)
|
|
849
|
+
except Exception as e: # noqa: BLE001
|
|
850
|
+
self._update_console(
|
|
851
|
+
False,
|
|
852
|
+
function,
|
|
853
|
+
self.task_ids[function],
|
|
854
|
+
f"[bold red]Failed to verify {function.lower()}: {e}",
|
|
855
|
+
completed=True,
|
|
856
|
+
)
|
|
857
|
+
verification_results.append(False)
|
|
858
|
+
return all(verification_results)
|