anyscale 0.5.127__tar.gz → 0.5.128__tar.gz

{anyscale-0.5.127/anyscale.egg-info → anyscale-0.5.128}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.5.127
+Version: 0.5.128
 Summary: Command Line Interface for Anyscale
 Home-page: UNKNOWN
 Author: Anyscale Inc.

{anyscale-0.5.127 → anyscale-0.5.128}/anyscale/client/README.md

@@ -192,7 +192,6 @@ Class | Method | HTTP request | Description
 *DefaultApi* | [**get_monitor_logs_stream_api_v2_logs_monitor_logs_stream_session_id_get**](docs/DefaultApi.md#get_monitor_logs_stream_api_v2_logs_monitor_logs_stream_session_id_get) | **GET** /api/v2/logs/monitor_logs_stream/{session_id} | Get Monitor Logs Stream
 *DefaultApi* | [**get_node_id_by_external_ip_api_v2_instances_external_ip_get_node_id_by_external_ip_get**](docs/DefaultApi.md#get_node_id_by_external_ip_api_v2_instances_external_ip_get_node_id_by_external_ip_get) | **GET** /api/v2/instances/{external_ip}/get_node_id_by_external_ip | Get Node Id By External Ip
 *DefaultApi* | [**get_node_id_by_internal_ip_api_v2_instances_internal_ip_get_node_id_by_internal_ip_get**](docs/DefaultApi.md#get_node_id_by_internal_ip_api_v2_instances_internal_ip_get_node_id_by_internal_ip_get) | **GET** /api/v2/instances/{internal_ip}/get_node_id_by_internal_ip | Get Node Id By Internal Ip
-*DefaultApi* | [**get_or_create_cluster_monitor_api_v2_sessions_session_id_get_or_create_cluster_monitor_post**](docs/DefaultApi.md#get_or_create_cluster_monitor_api_v2_sessions_session_id_get_or_create_cluster_monitor_post) | **POST** /api/v2/sessions/{session_id}/get_or_create_cluster_monitor | Get Or Create Cluster Monitor
 *DefaultApi* | [**get_ping_external_services_api_v2_sessions_session_id_ping_external_services_get**](docs/DefaultApi.md#get_ping_external_services_api_v2_sessions_session_id_ping_external_services_get) | **GET** /api/v2/sessions/{session_id}/ping_external_services | Get Ping External Services
 *DefaultApi* | [**get_pool_instances_api_v2_instances_pool_instances_get**](docs/DefaultApi.md#get_pool_instances_api_v2_instances_pool_instances_get) | **GET** /api/v2/instances/pool/instances | Get Pool Instances
 *DefaultApi* | [**get_project_api_v2_projects_project_id_get**](docs/DefaultApi.md#get_project_api_v2_projects_project_id_get) | **GET** /api/v2/projects/{project_id} | Get Project
@@ -420,13 +419,11 @@ Class | Method | HTTP request | Description
  - [ClusterConfigWithSessionIdleTimeout](docs/ClusterConfigWithSessionIdleTimeout.md)
  - [ClusterFeatures](docs/ClusterFeatures.md)
  - [ClusterManagementStackVersions](docs/ClusterManagementStackVersions.md)
- - [ClusterMonitorResponse](docs/ClusterMonitorResponse.md)
  - [ClusterStartup](docs/ClusterStartup.md)
  - [ClusterauthresponseResponse](docs/ClusterauthresponseResponse.md)
  - [ClusterconfigResponse](docs/ClusterconfigResponse.md)
  - [ClusterconfigwithsessionidletimeoutResponse](docs/ClusterconfigwithsessionidletimeoutResponse.md)
  - [ClusterfeaturesResponse](docs/ClusterfeaturesResponse.md)
- - [ClustermonitorresponseResponse](docs/ClustermonitorresponseResponse.md)
  - [ComputeNodeType](docs/ComputeNodeType.md)
  - [ComputeTemplate](docs/ComputeTemplate.md)
  - [ComputeTemplateConfig](docs/ComputeTemplateConfig.md)

{anyscale-0.5.127 → anyscale-0.5.128}/anyscale/client/openapi_client/__init__.py

@@ -108,13 +108,11 @@ from openapi_client.models.cluster_config import ClusterConfig
 from openapi_client.models.cluster_config_with_session_idle_timeout import ClusterConfigWithSessionIdleTimeout
 from openapi_client.models.cluster_features import ClusterFeatures
 from openapi_client.models.cluster_management_stack_versions import ClusterManagementStackVersions
-from openapi_client.models.cluster_monitor_response import ClusterMonitorResponse
 from openapi_client.models.cluster_startup import ClusterStartup
 from openapi_client.models.clusterauthresponse_response import ClusterauthresponseResponse
 from openapi_client.models.clusterconfig_response import ClusterconfigResponse
 from openapi_client.models.clusterconfigwithsessionidletimeout_response import ClusterconfigwithsessionidletimeoutResponse
 from openapi_client.models.clusterfeatures_response import ClusterfeaturesResponse
-from openapi_client.models.clustermonitorresponse_response import ClustermonitorresponseResponse
 from openapi_client.models.compute_node_type import ComputeNodeType
 from openapi_client.models.compute_template import ComputeTemplate
 from openapi_client.models.compute_template_config import ComputeTemplateConfig

{anyscale-0.5.127 → anyscale-0.5.128}/anyscale/client/openapi_client/api/default_api.py

@@ -12440,7 +12440,7 @@ class DefaultApi(object):
     def get_job_logs_download_v2_api_v2_logs_job_logs_download_v2_job_id_get(self, job_id, **kwargs):  # noqa: E501
         """Get Job Logs Download V2  # noqa: E501
 
-        Fetches the job logs of a single job run. This is a single API that supports job logs no matter if the job is still running or not.  If the job is still running, the logs will be fetched from the streaming logs s3 path. If the job is not running, the logs will be fetched from downloaded logs s3 path. If a pagination token is provided, we use the streaming logs or downloaded logs path depending on the format of the pagination token.  TODO(aguo): Implement job alive or dead detection TODO(aguo): Implement pagination  # noqa: E501
+        Fetches the job logs of a single job run. This is a single API that supports job logs no matter if the job is still running or not.  If the job is still running, the logs will be fetched from the streaming logs s3 path. If the job is not running, the logs will be fetched from downloaded logs s3 path. If a pagination token is provided, we use the streaming logs or downloaded logs path depending on the format of the pagination token.  TODO(aguo): Implement job alive or dead detection  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
         >>> thread = api.get_job_logs_download_v2_api_v2_logs_job_logs_download_v2_job_id_get(job_id, async_req=True)
@@ -12449,8 +12449,8 @@ class DefaultApi(object):
         :param async_req bool: execute request asynchronously
         :param str job_id: (required)
         :param int page_size: Number of chunks to fetch. Defaults to 400. Max of 1000.
-        :param str next_token: Pagination token for getting newer logs.
-        :param str prev_token: Pagination token for getting older logs.
+        :param str next_page_token: Pagination token for getting newer logs.
+        :param str previous_page_token: Pagination token for getting older logs.
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -12468,7 +12468,7 @@ class DefaultApi(object):
     def get_job_logs_download_v2_api_v2_logs_job_logs_download_v2_job_id_get_with_http_info(self, job_id, **kwargs):  # noqa: E501
         """Get Job Logs Download V2  # noqa: E501
 
-        Fetches the job logs of a single job run. This is a single API that supports job logs no matter if the job is still running or not.  If the job is still running, the logs will be fetched from the streaming logs s3 path. If the job is not running, the logs will be fetched from downloaded logs s3 path. If a pagination token is provided, we use the streaming logs or downloaded logs path depending on the format of the pagination token.  TODO(aguo): Implement job alive or dead detection TODO(aguo): Implement pagination  # noqa: E501
+        Fetches the job logs of a single job run. This is a single API that supports job logs no matter if the job is still running or not.  If the job is still running, the logs will be fetched from the streaming logs s3 path. If the job is not running, the logs will be fetched from downloaded logs s3 path. If a pagination token is provided, we use the streaming logs or downloaded logs path depending on the format of the pagination token.  TODO(aguo): Implement job alive or dead detection  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
         >>> thread = api.get_job_logs_download_v2_api_v2_logs_job_logs_download_v2_job_id_get_with_http_info(job_id, async_req=True)
@@ -12477,8 +12477,8 @@ class DefaultApi(object):
         :param async_req bool: execute request asynchronously
         :param str job_id: (required)
         :param int page_size: Number of chunks to fetch. Defaults to 400. Max of 1000.
-        :param str next_token: Pagination token for getting newer logs.
-        :param str prev_token: Pagination token for getting older logs.
+        :param str next_page_token: Pagination token for getting newer logs.
+        :param str previous_page_token: Pagination token for getting older logs.
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -12498,8 +12498,8 @@ class DefaultApi(object):
         all_params = [
             'job_id',
             'page_size',
-            'next_token',
-            'prev_token'
+            'next_page_token',
+            'previous_page_token'
         ]
         all_params.extend(
             [
@@ -12534,10 +12534,10 @@ class DefaultApi(object):
         query_params = []
         if 'page_size' in local_var_params and local_var_params['page_size'] is not None:  # noqa: E501
             query_params.append(('page_size', local_var_params['page_size']))  # noqa: E501
-        if 'next_token' in local_var_params and local_var_params['next_token'] is not None:  # noqa: E501
-            query_params.append(('next_token', local_var_params['next_token']))  # noqa: E501
-        if 'prev_token' in local_var_params and local_var_params['prev_token'] is not None:  # noqa: E501
-            query_params.append(('prev_token', local_var_params['prev_token']))  # noqa: E501
+        if 'next_page_token' in local_var_params and local_var_params['next_page_token'] is not None:  # noqa: E501
+            query_params.append(('next_page_token', local_var_params['next_page_token']))  # noqa: E501
+        if 'previous_page_token' in local_var_params and local_var_params['previous_page_token'] is not None:  # noqa: E501
+            query_params.append(('previous_page_token', local_var_params['previous_page_token']))  # noqa: E501
 
         header_params = {}
 
@@ -13741,120 +13741,6 @@ class DefaultApi(object):
             _request_timeout=local_var_params.get('_request_timeout'),
             collection_formats=collection_formats)
 
-    def get_or_create_cluster_monitor_api_v2_sessions_session_id_get_or_create_cluster_monitor_post(self, session_id, **kwargs):  # noqa: E501
-        """Get Or Create Cluster Monitor  # noqa: E501
-
-        Get or create a cluster monitor object for the session.  Returns the URL of the cluster monitor and its status.  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.get_or_create_cluster_monitor_api_v2_sessions_session_id_get_or_create_cluster_monitor_post(session_id, async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool: execute request asynchronously
-        :param str session_id: (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: ClustermonitorresponseResponse
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-        kwargs['_return_http_data_only'] = True
-        return self.get_or_create_cluster_monitor_api_v2_sessions_session_id_get_or_create_cluster_monitor_post_with_http_info(session_id, **kwargs)  # noqa: E501
-
-    def get_or_create_cluster_monitor_api_v2_sessions_session_id_get_or_create_cluster_monitor_post_with_http_info(self, session_id, **kwargs):  # noqa: E501
-        """Get Or Create Cluster Monitor  # noqa: E501
-
-        Get or create a cluster monitor object for the session.  Returns the URL of the cluster monitor and its status.  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-        >>> thread = api.get_or_create_cluster_monitor_api_v2_sessions_session_id_get_or_create_cluster_monitor_post_with_http_info(session_id, async_req=True)
-        >>> result = thread.get()
-
-        :param async_req bool: execute request asynchronously
-        :param str session_id: (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(ClustermonitorresponseResponse, status_code(int), headers(HTTPHeaderDict))
-                 If the method is called asynchronously,
-                 returns the request thread.
-        """
-
-        local_var_params = locals()
-
-        all_params = [
-            'session_id'
-        ]
-        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 get_or_create_cluster_monitor_api_v2_sessions_session_id_get_or_create_cluster_monitor_post" % key
-                )
-            local_var_params[key] = val
-        del local_var_params['kwargs']
-        # verify the required parameter 'session_id' is set
-        if self.api_client.client_side_validation and ('session_id' not in local_var_params or  # noqa: E501
-                                                        local_var_params['session_id'] is None):  # noqa: E501
-            raise ApiValueError("Missing the required parameter `session_id` when calling `get_or_create_cluster_monitor_api_v2_sessions_session_id_get_or_create_cluster_monitor_post`")  # noqa: E501
-
-        collection_formats = {}
-
-        path_params = {}
-        if 'session_id' in local_var_params:
-            path_params['session_id'] = local_var_params['session_id']  # noqa: E501
-
-        query_params = []
-
-        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/sessions/{session_id}/get_or_create_cluster_monitor', 'POST',
-            path_params,
-            query_params,
-            header_params,
-            body=body_params,
-            post_params=form_params,
-            files=local_var_files,
-            response_type='ClustermonitorresponseResponse',  # 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 get_ping_external_services_api_v2_sessions_session_id_ping_external_services_get(self, session_id, **kwargs):  # noqa: E501
         """Get Ping External Services  # noqa: E501
 
@@ -18220,6 +18106,7 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str cluster_id: (required)
+        :param str original_href:
         :param _preload_content: if False, the urllib3.HTTPResponse object will
                                  be returned without reading/decoding response
                                  data. Default is True.
@@ -18245,6 +18132,7 @@ class DefaultApi(object):
 
         :param async_req bool: execute request asynchronously
         :param str cluster_id: (required)
+        :param str original_href:
         :param _return_http_data_only: response data without head status code
                                        and headers
         :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -18262,7 +18150,8 @@ class DefaultApi(object):
         local_var_params = locals()
 
         all_params = [
-            'cluster_id'
+            'cluster_id',
+            'original_href'
         ]
         all_params.extend(
             [
@@ -18293,6 +18182,8 @@ class DefaultApi(object):
             path_params['cluster_id'] = local_var_params['cluster_id']  # noqa: E501
 
         query_params = []
+        if 'original_href' in local_var_params and local_var_params['original_href'] is not None:  # noqa: E501
+            query_params.append(('original_href', local_var_params['original_href']))  # noqa: E501
 
         header_params = {}
 

{anyscale-0.5.127 → anyscale-0.5.128}/anyscale/client/openapi_client/models/__init__.py

@@ -94,13 +94,11 @@ from openapi_client.models.cluster_config import ClusterConfig
 from openapi_client.models.cluster_config_with_session_idle_timeout import ClusterConfigWithSessionIdleTimeout
 from openapi_client.models.cluster_features import ClusterFeatures
 from openapi_client.models.cluster_management_stack_versions import ClusterManagementStackVersions
-from openapi_client.models.cluster_monitor_response import ClusterMonitorResponse
 from openapi_client.models.cluster_startup import ClusterStartup
 from openapi_client.models.clusterauthresponse_response import ClusterauthresponseResponse
 from openapi_client.models.clusterconfig_response import ClusterconfigResponse
 from openapi_client.models.clusterconfigwithsessionidletimeout_response import ClusterconfigwithsessionidletimeoutResponse
 from openapi_client.models.clusterfeatures_response import ClusterfeaturesResponse
-from openapi_client.models.clustermonitorresponse_response import ClustermonitorresponseResponse
 from openapi_client.models.compute_node_type import ComputeNodeType
 from openapi_client.models.compute_template import ComputeTemplate
 from openapi_client.models.compute_template_config import ComputeTemplateConfig

{anyscale-0.5.127 → anyscale-0.5.128}/anyscale/client/openapi_client/models/log_download_result.py

@@ -35,16 +35,18 @@ class LogDownloadResult(object):
     openapi_types = {
         'log_chunks': 'list[LogFileChunk]',
         'next_page_token': 'str',
+        'previous_page_token': 'str',
         'bearer_token': 'str'
     }
 
     attribute_map = {
         'log_chunks': 'log_chunks',
         'next_page_token': 'next_page_token',
+        'previous_page_token': 'previous_page_token',
         'bearer_token': 'bearer_token'
     }
 
-    def __init__(self, log_chunks=None, next_page_token=None, bearer_token=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, log_chunks=None, next_page_token=None, previous_page_token=None, bearer_token=None, local_vars_configuration=None):  # noqa: E501
         """LogDownloadResult - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -52,12 +54,15 @@ class LogDownloadResult(object):
 
         self._log_chunks = None
         self._next_page_token = None
+        self._previous_page_token = None
         self._bearer_token = None
         self.discriminator = None
 
         self.log_chunks = log_chunks
         if next_page_token is not None:
             self.next_page_token = next_page_token
+        if previous_page_token is not None:
+            self.previous_page_token = previous_page_token
         if bearer_token is not None:
             self.bearer_token = bearer_token
 
@@ -90,7 +95,7 @@ class LogDownloadResult(object):
     def next_page_token(self):
         """Gets the next_page_token of this LogDownloadResult.  # noqa: E501
 
-        Next page token.  # noqa: E501
+        Page token to fetch newer logs.  # noqa: E501
 
         :return: The next_page_token of this LogDownloadResult.  # noqa: E501
         :rtype: str
@@ -101,7 +106,7 @@ class LogDownloadResult(object):
     def next_page_token(self, next_page_token):
         """Sets the next_page_token of this LogDownloadResult.
 
-        Next page token.  # noqa: E501
+        Page token to fetch newer logs.  # noqa: E501
 
         :param next_page_token: The next_page_token of this LogDownloadResult.  # noqa: E501
         :type: str
@@ -109,6 +114,29 @@ class LogDownloadResult(object):
 
         self._next_page_token = next_page_token
 
+    @property
+    def previous_page_token(self):
+        """Gets the previous_page_token of this LogDownloadResult.  # noqa: E501
+
+        Page token to fetch older logs.  # noqa: E501
+
+        :return: The previous_page_token of this LogDownloadResult.  # noqa: E501
+        :rtype: str
+        """
+        return self._previous_page_token
+
+    @previous_page_token.setter
+    def previous_page_token(self, previous_page_token):
+        """Sets the previous_page_token of this LogDownloadResult.
+
+        Page token to fetch older logs.  # noqa: E501
+
+        :param previous_page_token: The previous_page_token of this LogDownloadResult.  # noqa: E501
+        :type: str
+        """
+
+        self._previous_page_token = previous_page_token
+
     @property
     def bearer_token(self):
         """Gets the bearer_token of this LogDownloadResult.  # noqa: E501

{anyscale-0.5.127 → anyscale-0.5.128}/anyscale/cloud_resource.py

@@ -479,7 +479,7 @@ def verify_aws_s3(  # noqa: PLR0911, PLR0912
     except ClientError as e:
         if e.response["Error"]["Code"] == "NoSuchCORSConfiguration":
             logger.warning(
-                f"S3 bucket {bucket_name} does not have CORS rules. This is safe to ignore if you are not using Anyscale UI. Otherwise please create the correct CORS rule for Anyscale according to https://docs.anyscale.com/anyscale-cloud-administration/deploy-on-aws#appendix-detailed-resource-requirements"
+                f"S3 bucket {bucket_name} does not have CORS rules. This is safe to ignore if you are not using Anyscale UI. Otherwise please create the correct CORS rule for Anyscale according to https://docs.anyscale.com/cloud-deployment/aws/manage-clouds#s3"
             )
             cors_rules = []
             if strict:

{anyscale-0.5.127 → anyscale-0.5.128}/anyscale/commands/service_commands.py

@@ -1,5 +1,4 @@
-import sys
-from typing import List, Optional
+from typing import Optional
 
 import click
 from typing_extensions import Literal
@@ -19,41 +18,6 @@ def service_cli() -> None:
     pass
 
 
-@service_cli.command(name="deploy", help="Deploy a v1 service to Anyscale.")
-@click.argument("service-config-file", required=True)
-@click.option("--name", "-n", required=False, default=None, help="Name of service.")
-@click.option(
-    "--description", required=False, default=None, help="Description of service."
-)
-@click.option(
-    "--healthcheck-url", required=False, help="Healthcheck URL for service.",
-)
-@click.argument(
-    "entrypoint", required=False, nargs=-1,
-)
-def deploy(
-    service_config_file: str,
-    entrypoint: List[str],
-    name: Optional[str],
-    description: Optional[str],
-    healthcheck_url: Optional[str],
-) -> None:
-    # TODO[Bruce]: Remove update flag once fully test apply function.
-    """Deploy a service (create or update).
-
-    This is *only* supported for v1 services.
-    """
-    service_controller = ServiceController()
-    service_controller.deploy(
-        service_config_file,
-        name=name,
-        description=description,
-        healthcheck_url=healthcheck_url,
-        entrypoint=entrypoint,
-        is_entrypoint_cmd="--" in sys.argv,
-    )
-
-
 @service_cli.command(
     name="rollout",
     help="Rollout a v2 service to Anyscale. Please contact support for details.",

{anyscale-0.5.127 → anyscale-0.5.128}/anyscale/controllers/cloud_controller.py

@@ -24,20 +24,18 @@ from anyscale.aws_iam_policies import (
 from anyscale.cli_logger import BlockLogger, LogsLogger
 from anyscale.client.openapi_client.models import (
     CloudConfig,
+    CloudProviders,
+    CloudState,
     CloudWithCloudResource,
     CloudWithCloudResourceGCP,
     ClusterManagementStackVersions,
     CreateCloudResource,
     CreateCloudResourceGCP,
+    SubnetIdWithAvailabilityZoneAWS,
     UpdateCloudWithCloudResource,
     UpdateCloudWithCloudResourceGCP,
     WriteCloud,
 )
-from anyscale.client.openapi_client.models.cloud_providers import CloudProviders
-from anyscale.client.openapi_client.models.cloud_state import CloudState
-from anyscale.client.openapi_client.models.subnet_id_with_availability_zone_aws import (
-    SubnetIdWithAvailabilityZoneAWS,
-)
 from anyscale.cloud import (
     get_cloud_id_and_name,
     get_cloud_json_from_id,
@@ -58,6 +56,7 @@ from anyscale.conf import ANYSCALE_IAM_ROLE_NAME
 from anyscale.controllers.base_controller import BaseController
 from anyscale.controllers.cloud_functional_verification_controller import (
     CloudFunctionalVerificationController,
+    CloudFunctionalVerificationType,
 )
 from anyscale.formatters import clouds_formatter
 from anyscale.shared_anyscale_utils.aws import AwsRoleArn
@@ -811,18 +810,19 @@ class CloudController(BaseController):
 
     def _validate_functional_verification_args(
         self, functional_verify: Optional[str]
-    ) -> List[str]:
+    ) -> List[CloudFunctionalVerificationType]:
         if functional_verify is None:
             return []
         # validate functional_verify
-        supported_functions = "workspace"  # TODO (congding): add service
         functions_to_verify = set()
         for function in functional_verify.split(","):
-            if function.lower() not in supported_functions:
+            fn = getattr(CloudFunctionalVerificationType, function.upper(), None)
+            if fn is None:
                 raise ClickException(
-                    f"Unsupported function {function} for --functional-verify. Supported functions are {list(supported_functions)}"
+                    f"Unsupported function {function} for --functional-verify. "
+                    f"Supported functions: {[function.lower() for function in CloudFunctionalVerificationType]}"
                 )
-            functions_to_verify.add(function.lower())
+            functions_to_verify.add(fn)
         return list(functions_to_verify)
 
     def verify_cloud(  # noqa: PLR0911