anyscale 0.26.46__py3-none-any.whl → 0.26.48__py3-none-any.whl

anyscale/_private/workload/workload_config.py

@@ -2,7 +2,12 @@ from dataclasses import dataclass, field
 from typing import Dict, List, Optional, Union
 
 from anyscale._private.models import ImageURI, ModelBase
-from anyscale.compute_config.models import ComputeConfig
+from anyscale.compute_config.models import (
+    compute_config_type_from_dict,
+    ComputeConfig,
+    ComputeConfigType,
+    MultiDeploymentComputeConfig,
+)
 
 
 @dataclass(frozen=True)
@@ -40,24 +45,27 @@ class WorkloadConfig(ModelBase):
         if containerfile is not None and not isinstance(containerfile, str):
             raise TypeError("'containerfile' must be a string.")
 
-    compute_config: Union[ComputeConfig, Dict, str, None] = field(
+    compute_config: Union[ComputeConfigType, Dict, str, None] = field(
         default=None,
         metadata={
-            "docstring": "The name of an existing registered compute config or an inlined ComputeConfig object."
+            "docstring": "The name of an existing registered compute config or an inlined ComputeConfig or MultiDeploymentComputeConfig object."
         },
     )
 
     def _validate_compute_config(
-        self, compute_config: Union[ComputeConfig, Dict, str, None]
-    ) -> Union[None, str, ComputeConfig]:
+        self, compute_config: Union[ComputeConfigType, Dict, str, None]
+    ) -> Union[None, str, ComputeConfigType]:
         if compute_config is None or isinstance(compute_config, str):
             return compute_config
 
         if isinstance(compute_config, dict):
-            compute_config = ComputeConfig.from_dict(compute_config)
-        if not isinstance(compute_config, ComputeConfig):
+            compute_config = compute_config_type_from_dict(compute_config)
+
+        if not isinstance(
+            compute_config, (ComputeConfig, MultiDeploymentComputeConfig)
+        ):
             raise TypeError(
-                "'compute_config' must be a string, ComputeConfig, or corresponding dict"
+                "'compute_config' must be a string, ComputeConfig, MultiDeploymentComputeConfig, or corresponding dict"
             )
 
         return compute_config

anyscale/_private/workload/workload_sdk.py

@@ -10,7 +10,11 @@ from anyscale._private.anyscale_client import (
 from anyscale._private.sdk.base_sdk import BaseSDK, Timer
 from anyscale.cli_logger import BlockLogger
 from anyscale.compute_config._private.compute_config_sdk import PrivateComputeConfigSDK
-from anyscale.compute_config.models import ComputeConfig
+from anyscale.compute_config.models import (
+    ComputeConfig,
+    ComputeConfigType,
+    MultiDeploymentComputeConfig,
+)
 from anyscale.image._private.image_sdk import PrivateImageSDK
 from anyscale.utils.runtime_env import is_dir_remote_uri, parse_requirements_file
 
@@ -189,7 +193,9 @@ class WorkloadSDK(BaseSDK):
         return new_runtime_envs
 
     def _resolve_compute_config_id(
-        self, compute_config: Union[str, ComputeConfig], cloud: Optional[str] = None,
+        self,
+        compute_config: Union[str, ComputeConfigType, None],
+        cloud: Optional[str] = None,
     ) -> str:
         """Resolve the passed compute config to its ID.
 
@@ -207,6 +213,15 @@ class WorkloadSDK(BaseSDK):
                 raise ValueError(
                     f"The compute config '{compute_config}' does not exist."
                 )
+        elif compute_config is None:
+            cloud_id = self.client.get_cloud_id(cloud_name=cloud)  # type: ignore
+            compute_config_id = self._client.get_default_compute_config(
+                cloud_id=cloud_id
+            ).id
+            if compute_config_id is None:
+                raise ValueError(
+                    f"The default compute config for cloud '{cloud}' does not exist."
+                )
         else:
             _, compute_config_id = self._compute_config_sdk.create_compute_config(
                 compute_config
@@ -217,7 +232,7 @@ class WorkloadSDK(BaseSDK):
     def resolve_compute_config_and_cloud_id(
         self,
         *,
-        compute_config: Union[ComputeConfig, Dict, str, None],
+        compute_config: Union[ComputeConfigType, Dict, str, None],
         cloud: Union[None, str],
     ) -> Tuple[str, str]:
         """Resolve the passed compute config to its ID and corresponding cloud ID.
@@ -252,7 +267,9 @@ class WorkloadSDK(BaseSDK):
         else:
             # If we are creating a new compute config, ensure cloud is set accordingly
             if (
-                isinstance(compute_config, ComputeConfig)
+                isinstance(
+                    compute_config, (ComputeConfig, MultiDeploymentComputeConfig)
+                )
                 and compute_config.cloud is None
             ):
                 compute_config = replace(compute_config, cloud=cloud)
@@ -315,7 +332,7 @@ class WorkloadSDK(BaseSDK):
 
     def get_user_facing_compute_config(
         self, compute_config_id: str,
-    ) -> Union[str, ComputeConfig]:
+    ) -> Union[str, ComputeConfigType]:
         """Get the compute config in a format to be displayed in a user-facing status.
 
         If the compute config refers to an anonymous compute config, its config

anyscale/client/README.md

@@ -352,6 +352,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**try_claim_cloud_api_v2_aioa_cloud_waitlist_claim_cloud_post**](docs/DefaultApi.md#try_claim_cloud_api_v2_aioa_cloud_waitlist_claim_cloud_post) | **POST** /api/v2/aioa_cloud_waitlist/claim_cloud | Try Claim Cloud
 *DefaultApi* | [**try_login_api_v2_users_try_login_post**](docs/DefaultApi.md#try_login_api_v2_users_try_login_post) | **POST** /api/v2/users/try-login | Try Login
 *DefaultApi* | [**update_cloud_auto_add_user_api_v2_clouds_cloud_id_auto_add_user_put**](docs/DefaultApi.md#update_cloud_auto_add_user_api_v2_clouds_cloud_id_auto_add_user_put) | **PUT** /api/v2/clouds/{cloud_id}/auto_add_user | Update Cloud Auto Add User
+*DefaultApi* | [**update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put**](docs/DefaultApi.md#update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put) | **PUT** /api/v2/clouds/{cloud_id}/update_deployment | Update Cloud Deployment
 *DefaultApi* | [**update_cloud_deployment_config_api_v2_clouds_cloud_id_deployment_cloud_deployment_id_config_put**](docs/DefaultApi.md#update_cloud_deployment_config_api_v2_clouds_cloud_id_deployment_cloud_deployment_id_config_put) | **PUT** /api/v2/clouds/{cloud_id}/deployment/{cloud_deployment_id}/config | Update Cloud Deployment Config
 *DefaultApi* | [**update_cloud_deployments_api_v2_clouds_cloud_id_deployments_put**](docs/DefaultApi.md#update_cloud_deployments_api_v2_clouds_cloud_id_deployments_put) | **PUT** /api/v2/clouds/{cloud_id}/deployments | Update Cloud Deployments
 *DefaultApi* | [**update_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_gcp_router_cloud_id_put**](docs/DefaultApi.md#update_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_gcp_router_cloud_id_put) | **PUT** /api/v2/clouds_with_cloud_resource_gcp_router/{cloud_id} | Update Cloud With Cloud Resource
@@ -366,6 +367,7 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**update_payment_info_api_v2_organization_billing_update_payment_info_post**](docs/DefaultApi.md#update_payment_info_api_v2_organization_billing_update_payment_info_post) | **POST** /api/v2/organization_billing/update_payment_info | Update Payment Info
 *DefaultApi* | [**update_resource_notification_api_v2_resource_notifications_resource_notification_id_put**](docs/DefaultApi.md#update_resource_notification_api_v2_resource_notifications_resource_notification_id_put) | **PUT** /api/v2/resource_notifications/{resource_notification_id} | Update Resource Notification
 *DefaultApi* | [**update_resource_quota_api_v2_resource_quotas_resource_quota_id_patch**](docs/DefaultApi.md#update_resource_quota_api_v2_resource_quotas_resource_quota_id_patch) | **PATCH** /api/v2/resource_quotas/{resource_quota_id} | Update Resource Quota
+*DefaultApi* | [**update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put**](docs/DefaultApi.md#update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put) | **PUT** /api/v2/clouds/{cloud_id}/update_system_cluster_config | Update System Cluster Config
 *DefaultApi* | [**upload_session_command_logs_api_v2_session_commands_session_command_id_upload_logs_post**](docs/DefaultApi.md#upload_session_command_logs_api_v2_session_commands_session_command_id_upload_logs_post) | **POST** /api/v2/session_commands/{session_command_id}/upload_logs | Upload Session Command Logs
 *DefaultApi* | [**upsert_billing_version_api_v2_organization_billing_billing_versions_put**](docs/DefaultApi.md#upsert_billing_version_api_v2_organization_billing_billing_versions_put) | **PUT** /api/v2/organization_billing/billing_versions | Upsert Billing Version
 *DefaultApi* | [**upsert_support_request_for_user_organization_api_v2_support_requests_support_request_put**](docs/DefaultApi.md#upsert_support_request_for_user_organization_api_v2_support_requests_support_request_put) | **PUT** /api/v2/support_requests/support_request | Upsert Support Request For User Organization
@@ -394,7 +396,7 @@ Class | Method | HTTP request | Description
  - [AggregatedusageResponse](docs/AggregatedusageResponse.md)
  - [AioaCloudWaitlistRecord](docs/AioaCloudWaitlistRecord.md)
  - [AioacloudwaitlistrecordResponse](docs/AioacloudwaitlistrecordResponse.md)
- - [AlertType](docs/AlertType.md)
+ - [AlertIssueType](docs/AlertIssueType.md)
  - [AnyscaleAWSAccount](docs/AnyscaleAWSAccount.md)
  - [AnyscaleServiceAccount](docs/AnyscaleServiceAccount.md)
  - [AnyscaleawsaccountResponse](docs/AnyscaleawsaccountResponse.md)
@@ -783,6 +785,7 @@ Class | Method | HTTP request | Description
  - [OperatorBranch](docs/OperatorBranch.md)
  - [OperatorId](docs/OperatorId.md)
  - [OperatorMetrics](docs/OperatorMetrics.md)
+ - [OperatorStatus](docs/OperatorStatus.md)
  - [Organization](docs/Organization.md)
  - [OrganizationCollaborator](docs/OrganizationCollaborator.md)
  - [OrganizationConfiguration](docs/OrganizationConfiguration.md)

anyscale/client/openapi_client/__init__.py

@@ -42,7 +42,7 @@ from openapi_client.models.aggregatedinstanceusagecsv_list_response import Aggre
 from openapi_client.models.aggregatedusage_response import AggregatedusageResponse
 from openapi_client.models.aioa_cloud_waitlist_record import AioaCloudWaitlistRecord
 from openapi_client.models.aioacloudwaitlistrecord_response import AioacloudwaitlistrecordResponse
-from openapi_client.models.alert_type import AlertType
+from openapi_client.models.alert_issue_type import AlertIssueType
 from openapi_client.models.anyscale_aws_account import AnyscaleAWSAccount
 from openapi_client.models.anyscale_service_account import AnyscaleServiceAccount
 from openapi_client.models.anyscaleawsaccount_response import AnyscaleawsaccountResponse
@@ -431,6 +431,7 @@ from openapi_client.models.operation_result import OperationResult
 from openapi_client.models.operator_branch import OperatorBranch
 from openapi_client.models.operator_id import OperatorId
 from openapi_client.models.operator_metrics import OperatorMetrics
+from openapi_client.models.operator_status import OperatorStatus
 from openapi_client.models.organization import Organization
 from openapi_client.models.organization_collaborator import OrganizationCollaborator
 from openapi_client.models.organization_configuration import OrganizationConfiguration

anyscale/client/openapi_client/api/default_api.py

@@ -16822,8 +16822,8 @@ class DefaultApi(object):
         :param str session_name: (required)
         :param str dataset_id: (required)
         :param str operator_id: (required)
-        :param int dataset_start_time: (required)
-        :param int dataset_end_time: (required)
+        :param float dataset_start_time: (required)
+        :param float dataset_end_time: (required)
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -16851,8 +16851,8 @@ class DefaultApi(object):
         :param str session_name: (required)
         :param str dataset_id: (required)
         :param str operator_id: (required)
-        :param int dataset_start_time: (required)
-        :param int dataset_end_time: (required)
+        :param float dataset_start_time: (required)
+        :param float dataset_end_time: (required)
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -28061,6 +28061,7 @@ class DefaultApi(object):
         :param int end_time: The end time for the query as a nanosecond Unix epoch. If not set, the query will use the time now.
         :param str positive_search_filter: Positive search terms to filter in the logs. Defaults to no filter. This can be a regex and the search is case insensitive.
         :param str negative_search_filter: Negative search terms to filter out the logs. Defaults to no filter. This can be a regex and the search is case insensitive.
+        :param str search_query: Advanced search query with syntax supporting NOT, AND, OR operators. Cannot be used together with positive_search_filter or negative_search_filter.
         :param str source: The source file (worker, raylet, dashboard, controller, proxy, replica, AUTOSCALER..) to filter the logs. Default to no filter.
         :param str filter_tags: The tags to filter the logs. Default to no filter. The format of this field is expected to be in json. The filer will apply to match each key of the object and do exact match on the value.
         :param bool reverse_sort: Whether the result should be sorted by descending timestamp. Also determines the query direction. When false, it will start to query from the start_time and return results until it reaches the page_size or the end_time. When true, it will start to query from the end_time and return results until it reaches the page_size or the start_time. Default to false.
@@ -28097,6 +28098,7 @@ class DefaultApi(object):
         :param int end_time: The end time for the query as a nanosecond Unix epoch. If not set, the query will use the time now.
         :param str positive_search_filter: Positive search terms to filter in the logs. Defaults to no filter. This can be a regex and the search is case insensitive.
         :param str negative_search_filter: Negative search terms to filter out the logs. Defaults to no filter. This can be a regex and the search is case insensitive.
+        :param str search_query: Advanced search query with syntax supporting NOT, AND, OR operators. Cannot be used together with positive_search_filter or negative_search_filter.
         :param str source: The source file (worker, raylet, dashboard, controller, proxy, replica, AUTOSCALER..) to filter the logs. Default to no filter.
         :param str filter_tags: The tags to filter the logs. Default to no filter. The format of this field is expected to be in json. The filer will apply to match each key of the object and do exact match on the value.
         :param bool reverse_sort: Whether the result should be sorted by descending timestamp. Also determines the query direction. When false, it will start to query from the start_time and return results until it reaches the page_size or the end_time. When true, it will start to query from the end_time and return results until it reaches the page_size or the start_time. Default to false.
@@ -28126,6 +28128,7 @@ class DefaultApi(object):
             'end_time',
             'positive_search_filter',
             'negative_search_filter',
+            'search_query',
             'source',
             'filter_tags',
             'reverse_sort'
@@ -28173,6 +28176,8 @@ class DefaultApi(object):
             query_params.append(('positive_search_filter', local_var_params['positive_search_filter']))  # noqa: E501
         if 'negative_search_filter' in local_var_params and local_var_params['negative_search_filter'] is not None:  # noqa: E501
             query_params.append(('negative_search_filter', local_var_params['negative_search_filter']))  # noqa: E501
+        if 'search_query' in local_var_params and local_var_params['search_query'] is not None:  # noqa: E501
+            query_params.append(('search_query', local_var_params['search_query']))  # noqa: E501
         if 'source' in local_var_params and local_var_params['source'] is not None:  # noqa: E501
             query_params.append(('source', local_var_params['source']))  # noqa: E501
         if 'filter_tags' in local_var_params and local_var_params['filter_tags'] is not None:  # noqa: E501
@@ -33423,6 +33428,131 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put(self, cloud_id, cloud_deployment, **kwargs):  # noqa: E501
+        """Update Cloud Deployment  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put(cloud_id, cloud_deployment, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str cloud_id: (required)
+        :param CloudDeployment cloud_deployment: (required)
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: object
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put_with_http_info(cloud_id, cloud_deployment, **kwargs)  # noqa: E501
+
+    def update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put_with_http_info(self, cloud_id, cloud_deployment, **kwargs):  # noqa: E501
+        """Update Cloud Deployment  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put_with_http_info(cloud_id, cloud_deployment, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str cloud_id: (required)
+        :param CloudDeployment cloud_deployment: (required)
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(object, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'cloud_id',
+            'cloud_deployment'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'cloud_id' is set
+        if self.api_client.client_side_validation and ('cloud_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['cloud_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `cloud_id` when calling `update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put`")  # noqa: E501
+        # verify the required parameter 'cloud_deployment' is set
+        if self.api_client.client_side_validation and ('cloud_deployment' not in local_var_params or  # noqa: E501
+                                                        local_var_params['cloud_deployment'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `cloud_deployment` when calling `update_cloud_deployment_api_v2_clouds_cloud_id_update_deployment_put`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'cloud_id' in local_var_params:
+            path_params['cloud_id'] = local_var_params['cloud_id']  # noqa: E501
+
+        query_params = []
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        if 'cloud_deployment' in local_var_params:
+            body_params = local_var_params['cloud_deployment']
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # HTTP header `Content-Type`
+        header_params['Content-Type'] = self.api_client.select_header_content_type(  # noqa: E501
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/clouds/{cloud_id}/update_deployment', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='object',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def update_cloud_deployment_config_api_v2_clouds_cloud_id_deployment_cloud_deployment_id_config_put(self, cloud_id, cloud_deployment_id, cloud_deployment_config, **kwargs):  # noqa: E501
         """Update Cloud Deployment Config  # noqa: E501
 
@@ -35140,6 +35270,125 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
+    def update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put(self, cloud_id, **kwargs):  # noqa: E501
+        """Update System Cluster Config  # noqa: E501
+
+        Update system cluster configuration for a cloud.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put(cloud_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str cloud_id: (required)
+        :param bool is_enabled:
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: object
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+        kwargs['_return_http_data_only'] = True
+        return self.update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put_with_http_info(cloud_id, **kwargs)  # noqa: E501
+
+    def update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put_with_http_info(self, cloud_id, **kwargs):  # noqa: E501
+        """Update System Cluster Config  # noqa: E501
+
+        Update system cluster configuration for a cloud.  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+        >>> thread = api.update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put_with_http_info(cloud_id, async_req=True)
+        >>> result = thread.get()
+
+        :param async_req bool: execute request asynchronously
+        :param str cloud_id: (required)
+        :param bool is_enabled:
+        :param _return_http_data_only: response data without head status code
+                                       and headers
+        :param _preload_content: if False, the urllib3.HTTPResponse object will
+                                 be returned without reading/decoding response
+                                 data. Default is True.
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :return: tuple(object, status_code(int), headers(HTTPHeaderDict))
+                 If the method is called asynchronously,
+                 returns the request thread.
+        """
+
+        local_var_params = locals()
+
+        all_params = [
+            'cloud_id',
+            'is_enabled'
+        ]
+        all_params.extend(
+            [
+                'async_req',
+                '_return_http_data_only',
+                '_preload_content',
+                '_request_timeout'
+            ]
+        )
+
+        for key, val in six.iteritems(local_var_params['kwargs']):
+            if key not in all_params:
+                raise ApiTypeError(
+                    "Got an unexpected keyword argument '%s'"
+                    " to method update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put" % key
+                )
+            local_var_params[key] = val
+        del local_var_params['kwargs']
+        # verify the required parameter 'cloud_id' is set
+        if self.api_client.client_side_validation and ('cloud_id' not in local_var_params or  # noqa: E501
+                                                        local_var_params['cloud_id'] is None):  # noqa: E501
+            raise ApiValueError("Missing the required parameter `cloud_id` when calling `update_system_cluster_config_api_v2_clouds_cloud_id_update_system_cluster_config_put`")  # noqa: E501
+
+        collection_formats = {}
+
+        path_params = {}
+        if 'cloud_id' in local_var_params:
+            path_params['cloud_id'] = local_var_params['cloud_id']  # noqa: E501
+
+        query_params = []
+        if 'is_enabled' in local_var_params and local_var_params['is_enabled'] is not None:  # noqa: E501
+            query_params.append(('is_enabled', local_var_params['is_enabled']))  # noqa: E501
+
+        header_params = {}
+
+        form_params = []
+        local_var_files = {}
+
+        body_params = None
+        # HTTP header `Accept`
+        header_params['Accept'] = self.api_client.select_header_accept(
+            ['application/json'])  # noqa: E501
+
+        # Authentication setting
+        auth_settings = []  # noqa: E501
+
+        return self.api_client.call_api(
+            '/api/v2/clouds/{cloud_id}/update_system_cluster_config', 'PUT',
+            path_params,
+            query_params,
+            header_params,
+            body=body_params,
+            post_params=form_params,
+            files=local_var_files,
+            response_type='object',  # noqa: E501
+            auth_settings=auth_settings,
+            async_req=local_var_params.get('async_req'),
+            _return_http_data_only=local_var_params.get('_return_http_data_only'),  # noqa: E501
+            _preload_content=local_var_params.get('_preload_content', True),
+            _request_timeout=local_var_params.get('_request_timeout'),
+            collection_formats=collection_formats)
+
     def upload_session_command_logs_api_v2_session_commands_session_command_id_upload_logs_post(self, session_command_id, **kwargs):  # noqa: E501
         """Upload Session Command Logs  # noqa: E501
 

anyscale/client/openapi_client/models/__init__.py

@@ -28,7 +28,7 @@ from openapi_client.models.aggregatedinstanceusagecsv_list_response import Aggre
 from openapi_client.models.aggregatedusage_response import AggregatedusageResponse
 from openapi_client.models.aioa_cloud_waitlist_record import AioaCloudWaitlistRecord
 from openapi_client.models.aioacloudwaitlistrecord_response import AioacloudwaitlistrecordResponse
-from openapi_client.models.alert_type import AlertType
+from openapi_client.models.alert_issue_type import AlertIssueType
 from openapi_client.models.anyscale_aws_account import AnyscaleAWSAccount
 from openapi_client.models.anyscale_service_account import AnyscaleServiceAccount
 from openapi_client.models.anyscaleawsaccount_response import AnyscaleawsaccountResponse
@@ -417,6 +417,7 @@ from openapi_client.models.operation_result import OperationResult
 from openapi_client.models.operator_branch import OperatorBranch
 from openapi_client.models.operator_id import OperatorId
 from openapi_client.models.operator_metrics import OperatorMetrics
+from openapi_client.models.operator_status import OperatorStatus
 from openapi_client.models.organization import Organization
 from openapi_client.models.organization_collaborator import OrganizationCollaborator
 from openapi_client.models.organization_configuration import OrganizationConfiguration

anyscale/client/openapi_client/models/{alert_type.py → alert_issue_type.py}

@@ -18,7 +18,7 @@ import six
 from openapi_client.configuration import Configuration
 
 
-class AlertType(object):
+class AlertIssueType(object):
     """NOTE: This class is auto generated by OpenAPI Generator.
     Ref: https://openapi-generator.tech
 
@@ -28,22 +28,10 @@ class AlertType(object):
     """
     allowed enum values
     """
-    LOW_CREDIT_BALANCE_REACHED = "low_credit_balance_reached"
-    SPEND_THRESHOLD_REACHED = "spend_threshold_reached"
-    LOW_REMAINING_DAYS_IN_PLAN_REACHED = "low_remaining_days_in_plan_reached"
-    LOW_REMAINING_CREDIT_PERCENTAGE_REACHED = "low_remaining_credit_percentage_reached"
-    USAGE_THRESHOLD_REACHED = "usage_threshold_reached"
-    LOW_REMAINING_COMMIT_BALANCE_REACHED = "low_remaining_commit_balance_reached"
-    LOW_REMAINING_COMMIT_PERCENTAGE_REACHED = "low_remaining_commit_percentage_reached"
-    LOW_REMAINING_DAYS_FOR_COMMIT_SEGMENT_REACHED = "low_remaining_days_for_commit_segment_reached"
-    LOW_REMAINING_CONTRACT_CREDIT_BALANCE_REACHED = "low_remaining_contract_credit_balance_reached"
-    LOW_REMAINING_CONTRACT_CREDIT_PERCENTAGE_REACHED = "low_remaining_contract_credit_percentage_reached"
-    LOW_REMAINING_DAYS_FOR_CONTRACT_CREDIT_SEGMENT_REACHED = "low_remaining_days_for_contract_credit_segment_reached"
-    LOW_REMAINING_CONTRACT_CREDIT_AND_COMMIT_BALANCE_REACHED = "low_remaining_contract_credit_and_commit_balance_reached"
-    MONTHLY_INVOICE_TOTAL_SPEND_THRESHOLD_REACHED = "monthly_invoice_total_spend_threshold_reached"
-    INVOICE_TOTAL_REACHED = "invoice_total_reached"
-
-    allowable_values = [LOW_CREDIT_BALANCE_REACHED, SPEND_THRESHOLD_REACHED, LOW_REMAINING_DAYS_IN_PLAN_REACHED, LOW_REMAINING_CREDIT_PERCENTAGE_REACHED, USAGE_THRESHOLD_REACHED, LOW_REMAINING_COMMIT_BALANCE_REACHED, LOW_REMAINING_COMMIT_PERCENTAGE_REACHED, LOW_REMAINING_DAYS_FOR_COMMIT_SEGMENT_REACHED, LOW_REMAINING_CONTRACT_CREDIT_BALANCE_REACHED, LOW_REMAINING_CONTRACT_CREDIT_PERCENTAGE_REACHED, LOW_REMAINING_DAYS_FOR_CONTRACT_CREDIT_SEGMENT_REACHED, LOW_REMAINING_CONTRACT_CREDIT_AND_COMMIT_BALANCE_REACHED, MONTHLY_INVOICE_TOTAL_SPEND_THRESHOLD_REACHED, INVOICE_TOTAL_REACHED]  # noqa: E501
+    SPEND = "spend"
+    CREDIT = "credit"
+
+    allowable_values = [SPEND, CREDIT]  # noqa: E501
 
     """
     Attributes:
@@ -59,7 +47,7 @@ class AlertType(object):
     }
 
     def __init__(self, local_vars_configuration=None):  # noqa: E501
-        """AlertType - a model defined in OpenAPI"""  # noqa: E501
+        """AlertIssueType - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
         self.local_vars_configuration = local_vars_configuration
@@ -99,14 +87,14 @@ class AlertType(object):
 
     def __eq__(self, other):
         """Returns true if both objects are equal"""
-        if not isinstance(other, AlertType):
+        if not isinstance(other, AlertIssueType):
             return False
 
         return self.to_dict() == other.to_dict()
 
     def __ne__(self, other):
         """Returns true if both objects are not equal"""
-        if not isinstance(other, AlertType):
+        if not isinstance(other, AlertIssueType):
             return True
 
         return self.to_dict() != other.to_dict()