anyscale 0.26.23__py3-none-any.whl → 0.26.25__py3-none-any.whl

anyscale/client/openapi_client/models/decorated_compute_template_config.py

@@ -34,6 +34,8 @@ class DecoratedComputeTemplateConfig(object):
     """
     openapi_types = {
         'cloud_id': 'str',
+        'maximum_uptime_minutes': 'int',
+        'deployment_configs': 'list[CloudDeploymentComputeConfig]',
         'max_workers': 'int',
         'region': 'str',
         'allowed_azs': 'list[str]',
@@ -42,7 +44,6 @@ class DecoratedComputeTemplateConfig(object):
         'aws_advanced_configurations_json': 'object',
         'gcp_advanced_configurations_json': 'object',
         'advanced_configurations_json': 'object',
-        'maximum_uptime_minutes': 'int',
         'auto_select_worker_config': 'bool',
         'flags': 'object',
         'idle_termination_minutes': 'int',
@@ -51,6 +52,8 @@ class DecoratedComputeTemplateConfig(object):
 
     attribute_map = {
         'cloud_id': 'cloud_id',
+        'maximum_uptime_minutes': 'maximum_uptime_minutes',
+        'deployment_configs': 'deployment_configs',
         'max_workers': 'max_workers',
         'region': 'region',
         'allowed_azs': 'allowed_azs',
@@ -59,20 +62,21 @@ class DecoratedComputeTemplateConfig(object):
         'aws_advanced_configurations_json': 'aws_advanced_configurations_json',
         'gcp_advanced_configurations_json': 'gcp_advanced_configurations_json',
         'advanced_configurations_json': 'advanced_configurations_json',
-        'maximum_uptime_minutes': 'maximum_uptime_minutes',
         'auto_select_worker_config': 'auto_select_worker_config',
         'flags': 'flags',
         'idle_termination_minutes': 'idle_termination_minutes',
         'cloud': 'cloud'
     }
 
-    def __init__(self, cloud_id=None, max_workers=None, region=None, allowed_azs=None, head_node_type=None, worker_node_types=None, aws_advanced_configurations_json=None, gcp_advanced_configurations_json=None, advanced_configurations_json=None, maximum_uptime_minutes=None, auto_select_worker_config=False, flags=None, idle_termination_minutes=None, cloud=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, cloud_id=None, maximum_uptime_minutes=None, deployment_configs=None, max_workers=None, region=None, allowed_azs=None, head_node_type=None, worker_node_types=None, aws_advanced_configurations_json=None, gcp_advanced_configurations_json=None, advanced_configurations_json=None, auto_select_worker_config=False, flags=None, idle_termination_minutes=None, cloud=None, local_vars_configuration=None):  # noqa: E501
         """DecoratedComputeTemplateConfig - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
         self.local_vars_configuration = local_vars_configuration
 
         self._cloud_id = None
+        self._maximum_uptime_minutes = None
+        self._deployment_configs = None
         self._max_workers = None
         self._region = None
         self._allowed_azs = None
@@ -81,7 +85,6 @@ class DecoratedComputeTemplateConfig(object):
         self._aws_advanced_configurations_json = None
         self._gcp_advanced_configurations_json = None
         self._advanced_configurations_json = None
-        self._maximum_uptime_minutes = None
         self._auto_select_worker_config = None
         self._flags = None
         self._idle_termination_minutes = None
@@ -89,9 +92,14 @@ class DecoratedComputeTemplateConfig(object):
         self.discriminator = None
 
         self.cloud_id = cloud_id
+        if maximum_uptime_minutes is not None:
+            self.maximum_uptime_minutes = maximum_uptime_minutes
+        if deployment_configs is not None:
+            self.deployment_configs = deployment_configs
         if max_workers is not None:
             self.max_workers = max_workers
-        self.region = region
+        if region is not None:
+            self.region = region
         if allowed_azs is not None:
             self.allowed_azs = allowed_azs
         self.head_node_type = head_node_type
@@ -103,8 +111,6 @@ class DecoratedComputeTemplateConfig(object):
             self.gcp_advanced_configurations_json = gcp_advanced_configurations_json
         if advanced_configurations_json is not None:
             self.advanced_configurations_json = advanced_configurations_json
-        if maximum_uptime_minutes is not None:
-            self.maximum_uptime_minutes = maximum_uptime_minutes
         if auto_select_worker_config is not None:
             self.auto_select_worker_config = auto_select_worker_config
         if flags is not None:
@@ -138,11 +144,57 @@ class DecoratedComputeTemplateConfig(object):
 
         self._cloud_id = cloud_id
 
+    @property
+    def maximum_uptime_minutes(self):
+        """Gets the maximum_uptime_minutes of this DecoratedComputeTemplateConfig.  # noqa: E501
+
+        If set to a positive number, Anyscale will terminate the cluster this many minutes after cluster start.  # noqa: E501
+
+        :return: The maximum_uptime_minutes of this DecoratedComputeTemplateConfig.  # noqa: E501
+        :rtype: int
+        """
+        return self._maximum_uptime_minutes
+
+    @maximum_uptime_minutes.setter
+    def maximum_uptime_minutes(self, maximum_uptime_minutes):
+        """Sets the maximum_uptime_minutes of this DecoratedComputeTemplateConfig.
+
+        If set to a positive number, Anyscale will terminate the cluster this many minutes after cluster start.  # noqa: E501
+
+        :param maximum_uptime_minutes: The maximum_uptime_minutes of this DecoratedComputeTemplateConfig.  # noqa: E501
+        :type: int
+        """
+
+        self._maximum_uptime_minutes = maximum_uptime_minutes
+
+    @property
+    def deployment_configs(self):
+        """Gets the deployment_configs of this DecoratedComputeTemplateConfig.  # noqa: E501
+
+        A list of cloud deployment-specific configs to use.  # noqa: E501
+
+        :return: The deployment_configs of this DecoratedComputeTemplateConfig.  # noqa: E501
+        :rtype: list[CloudDeploymentComputeConfig]
+        """
+        return self._deployment_configs
+
+    @deployment_configs.setter
+    def deployment_configs(self, deployment_configs):
+        """Sets the deployment_configs of this DecoratedComputeTemplateConfig.
+
+        A list of cloud deployment-specific configs to use.  # noqa: E501
+
+        :param deployment_configs: The deployment_configs of this DecoratedComputeTemplateConfig.  # noqa: E501
+        :type: list[CloudDeploymentComputeConfig]
+        """
+
+        self._deployment_configs = deployment_configs
+
     @property
     def max_workers(self):
         """Gets the max_workers of this DecoratedComputeTemplateConfig.  # noqa: E501
 
-        Desired limit on total running workers for this session.  # noqa: E501
+        DEPRECATED. This attribute will be ignored - please use the \"max_resources\" flag to configure resource limits.  # noqa: E501
 
         :return: The max_workers of this DecoratedComputeTemplateConfig.  # noqa: E501
         :rtype: int
@@ -153,7 +205,7 @@ class DecoratedComputeTemplateConfig(object):
     def max_workers(self, max_workers):
         """Sets the max_workers of this DecoratedComputeTemplateConfig.
 
-        Desired limit on total running workers for this session.  # noqa: E501
+        DEPRECATED. This attribute will be ignored - please use the \"max_resources\" flag to configure resource limits.  # noqa: E501
 
         :param max_workers: The max_workers of this DecoratedComputeTemplateConfig.  # noqa: E501
         :type: int
@@ -181,8 +233,6 @@ class DecoratedComputeTemplateConfig(object):
         :param region: The region of this DecoratedComputeTemplateConfig.  # noqa: E501
         :type: str
         """
-        if self.local_vars_configuration.client_side_validation and region is None:  # noqa: E501
-            raise ValueError("Invalid value for `region`, must not be `None`")  # noqa: E501
 
         self._region = region
 
@@ -326,29 +376,6 @@ class DecoratedComputeTemplateConfig(object):
 
         self._advanced_configurations_json = advanced_configurations_json
 
-    @property
-    def maximum_uptime_minutes(self):
-        """Gets the maximum_uptime_minutes of this DecoratedComputeTemplateConfig.  # noqa: E501
-
-        If set to a positive number, Anyscale will terminate the cluster this many minutes after cluster start.  # noqa: E501
-
-        :return: The maximum_uptime_minutes of this DecoratedComputeTemplateConfig.  # noqa: E501
-        :rtype: int
-        """
-        return self._maximum_uptime_minutes
-
-    @maximum_uptime_minutes.setter
-    def maximum_uptime_minutes(self, maximum_uptime_minutes):
-        """Sets the maximum_uptime_minutes of this DecoratedComputeTemplateConfig.
-
-        If set to a positive number, Anyscale will terminate the cluster this many minutes after cluster start.  # noqa: E501
-
-        :param maximum_uptime_minutes: The maximum_uptime_minutes of this DecoratedComputeTemplateConfig.  # noqa: E501
-        :type: int
-        """
-
-        self._maximum_uptime_minutes = maximum_uptime_minutes
-
     @property
     def auto_select_worker_config(self):
         """Gets the auto_select_worker_config of this DecoratedComputeTemplateConfig.  # noqa: E501

anyscale/client/openapi_client/models/decorated_session.py

@@ -60,6 +60,7 @@ class DecoratedSession(object):
         'train_metrics_dashboard_url': 'str',
         'serve_metrics_dashboard_url': 'str',
         'serve_deployment_metrics_dashboard_url': 'str',
+        'serve_llm_metrics_dashboard_url': 'str',
         'persistent_metrics_url': 'str',
         'connect_url': 'str',
         'jupyter_notebook_url': 'str',
@@ -121,6 +122,7 @@ class DecoratedSession(object):
         'train_metrics_dashboard_url': 'train_metrics_dashboard_url',
         'serve_metrics_dashboard_url': 'serve_metrics_dashboard_url',
         'serve_deployment_metrics_dashboard_url': 'serve_deployment_metrics_dashboard_url',
+        'serve_llm_metrics_dashboard_url': 'serve_llm_metrics_dashboard_url',
         'persistent_metrics_url': 'persistent_metrics_url',
         'connect_url': 'connect_url',
         'jupyter_notebook_url': 'jupyter_notebook_url',
@@ -154,7 +156,7 @@ class DecoratedSession(object):
         'project': 'project'
     }
 
-    def __init__(self, name=None, project_id=None, cloud_id=None, cluster_config=None, build_id=None, compute_template_id=None, idle_timeout=120, uses_app_config=False, allow_public_internet_traffic=False, user_service_access=None, user_service_token=None, ha_job_id=None, id=None, state=None, pending_state=None, state_data=None, status=None, status_details=None, creator_id=None, created_at=None, archived_at=None, webterminal_auth_url=None, metrics_dashboard_url=None, data_metrics_dashboard_url=None, train_metrics_dashboard_url=None, serve_metrics_dashboard_url=None, serve_deployment_metrics_dashboard_url=None, persistent_metrics_url=None, connect_url=None, jupyter_notebook_url=None, ray_dashboard_url=None, access_token=None, service_proxy_url=None, tensorboard_available=None, cluster_config_last_modified_at=None, host_name=None, head_node_ip=None, ssh_authorized_keys=None, ssh_private_key=None, anyscaled_config=None, anyscaled_config_generated_at=None, default_build_id=None, idle_timeout_last_activity_at=None, ray_version=None, ray_version_last_updated_at=None, user_service_url=None, ray_component_activities_last_reported_at=None, activity_details=None, maximum_uptime_will_terminate_cluster_at=None, idle_termination_status=None, ray_dashboard_snapshot_last_reported_at=None, build=None, cloud=None, creator=None, compute_template=None, idle_time_remaining_seconds=None, access=None, project=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, name=None, project_id=None, cloud_id=None, cluster_config=None, build_id=None, compute_template_id=None, idle_timeout=120, uses_app_config=False, allow_public_internet_traffic=False, user_service_access=None, user_service_token=None, ha_job_id=None, id=None, state=None, pending_state=None, state_data=None, status=None, status_details=None, creator_id=None, created_at=None, archived_at=None, webterminal_auth_url=None, metrics_dashboard_url=None, data_metrics_dashboard_url=None, train_metrics_dashboard_url=None, serve_metrics_dashboard_url=None, serve_deployment_metrics_dashboard_url=None, serve_llm_metrics_dashboard_url=None, persistent_metrics_url=None, connect_url=None, jupyter_notebook_url=None, ray_dashboard_url=None, access_token=None, service_proxy_url=None, tensorboard_available=None, cluster_config_last_modified_at=None, host_name=None, head_node_ip=None, ssh_authorized_keys=None, ssh_private_key=None, anyscaled_config=None, anyscaled_config_generated_at=None, default_build_id=None, idle_timeout_last_activity_at=None, ray_version=None, ray_version_last_updated_at=None, user_service_url=None, ray_component_activities_last_reported_at=None, activity_details=None, maximum_uptime_will_terminate_cluster_at=None, idle_termination_status=None, ray_dashboard_snapshot_last_reported_at=None, build=None, cloud=None, creator=None, compute_template=None, idle_time_remaining_seconds=None, access=None, project=None, local_vars_configuration=None):  # noqa: E501
         """DecoratedSession - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -187,6 +189,7 @@ class DecoratedSession(object):
         self._train_metrics_dashboard_url = None
         self._serve_metrics_dashboard_url = None
         self._serve_deployment_metrics_dashboard_url = None
+        self._serve_llm_metrics_dashboard_url = None
         self._persistent_metrics_url = None
         self._connect_url = None
         self._jupyter_notebook_url = None
@@ -266,6 +269,8 @@ class DecoratedSession(object):
             self.serve_metrics_dashboard_url = serve_metrics_dashboard_url
         if serve_deployment_metrics_dashboard_url is not None:
             self.serve_deployment_metrics_dashboard_url = serve_deployment_metrics_dashboard_url
+        if serve_llm_metrics_dashboard_url is not None:
+            self.serve_llm_metrics_dashboard_url = serve_llm_metrics_dashboard_url
         if persistent_metrics_url is not None:
             self.persistent_metrics_url = persistent_metrics_url
         if connect_url is not None:
@@ -959,6 +964,29 @@ class DecoratedSession(object):
 
         self._serve_deployment_metrics_dashboard_url = serve_deployment_metrics_dashboard_url
 
+    @property
+    def serve_llm_metrics_dashboard_url(self):
+        """Gets the serve_llm_metrics_dashboard_url of this DecoratedSession.  # noqa: E501
+
+        URL for Serve LLM Grafana dashboard for this Session. This field will only be populated after the Session finishes starting.  # noqa: E501
+
+        :return: The serve_llm_metrics_dashboard_url of this DecoratedSession.  # noqa: E501
+        :rtype: str
+        """
+        return self._serve_llm_metrics_dashboard_url
+
+    @serve_llm_metrics_dashboard_url.setter
+    def serve_llm_metrics_dashboard_url(self, serve_llm_metrics_dashboard_url):
+        """Sets the serve_llm_metrics_dashboard_url of this DecoratedSession.
+
+        URL for Serve LLM Grafana dashboard for this Session. This field will only be populated after the Session finishes starting.  # noqa: E501
+
+        :param serve_llm_metrics_dashboard_url: The serve_llm_metrics_dashboard_url of this DecoratedSession.  # noqa: E501
+        :type: str
+        """
+
+        self._serve_llm_metrics_dashboard_url = serve_llm_metrics_dashboard_url
+
     @property
     def persistent_metrics_url(self):
         """Gets the persistent_metrics_url of this DecoratedSession.  # noqa: E501
@@ -1055,7 +1083,7 @@ class DecoratedSession(object):
     def access_token(self):
         """Gets the access_token of this DecoratedSession.  # noqa: E501
 
-        Access token for web based services (e.g. jupyter, tensorboard, etc). This field will be populated when the web based services are available after the Session finishes starting.  # noqa: E501
+        [DEPRECATED] Call GET /api/v2/authentication/{cluster_id}/cluster_access_token to get this.  # noqa: E501
 
         :return: The access_token of this DecoratedSession.  # noqa: E501
         :rtype: str
@@ -1066,7 +1094,7 @@ class DecoratedSession(object):
     def access_token(self, access_token):
         """Sets the access_token of this DecoratedSession.
 
-        Access token for web based services (e.g. jupyter, tensorboard, etc). This field will be populated when the web based services are available after the Session finishes starting.  # noqa: E501
+        [DEPRECATED] Call GET /api/v2/authentication/{cluster_id}/cluster_access_token to get this.  # noqa: E501
 
         :param access_token: The access_token of this DecoratedSession.  # noqa: E501
         :type: str

anyscale/client/openapi_client/models/workspace_dataplane_proxied_artifacts.py

@@ -37,7 +37,8 @@ class WorkspaceDataplaneProxiedArtifacts(object):
         'skip_packages_tracking': 'str',
         'environment_variables': 'list[str]',
         'dockerfile': 'str',
-        'dockerfile_draft': 'str'
+        'dockerfile_draft': 'str',
+        'latest_snapshot_uri': 'str'
     }
 
     attribute_map = {
@@ -45,10 +46,11 @@ class WorkspaceDataplaneProxiedArtifacts(object):
         'skip_packages_tracking': 'skip_packages_tracking',
         'environment_variables': 'environment_variables',
         'dockerfile': 'dockerfile',
-        'dockerfile_draft': 'dockerfile_draft'
+        'dockerfile_draft': 'dockerfile_draft',
+        'latest_snapshot_uri': 'latest_snapshot_uri'
     }
 
-    def __init__(self, requirements=None, skip_packages_tracking=None, environment_variables=None, dockerfile=None, dockerfile_draft=None, local_vars_configuration=None):  # noqa: E501
+    def __init__(self, requirements=None, skip_packages_tracking=None, environment_variables=None, dockerfile=None, dockerfile_draft=None, latest_snapshot_uri=None, local_vars_configuration=None):  # noqa: E501
         """WorkspaceDataplaneProxiedArtifacts - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:
             local_vars_configuration = Configuration()
@@ -59,6 +61,7 @@ class WorkspaceDataplaneProxiedArtifacts(object):
         self._environment_variables = None
         self._dockerfile = None
         self._dockerfile_draft = None
+        self._latest_snapshot_uri = None
         self.discriminator = None
 
         if requirements is not None:
@@ -71,6 +74,8 @@ class WorkspaceDataplaneProxiedArtifacts(object):
             self.dockerfile = dockerfile
         if dockerfile_draft is not None:
             self.dockerfile_draft = dockerfile_draft
+        if latest_snapshot_uri is not None:
+            self.latest_snapshot_uri = latest_snapshot_uri
 
     @property
     def requirements(self):
@@ -187,6 +192,29 @@ class WorkspaceDataplaneProxiedArtifacts(object):
 
         self._dockerfile_draft = dockerfile_draft
 
+    @property
+    def latest_snapshot_uri(self):
+        """Gets the latest_snapshot_uri of this WorkspaceDataplaneProxiedArtifacts.  # noqa: E501
+
+        The URI of the latest snapshot of the workspace.  # noqa: E501
+
+        :return: The latest_snapshot_uri of this WorkspaceDataplaneProxiedArtifacts.  # noqa: E501
+        :rtype: str
+        """
+        return self._latest_snapshot_uri
+
+    @latest_snapshot_uri.setter
+    def latest_snapshot_uri(self, latest_snapshot_uri):
+        """Sets the latest_snapshot_uri of this WorkspaceDataplaneProxiedArtifacts.
+
+        The URI of the latest snapshot of the workspace.  # noqa: E501
+
+        :param latest_snapshot_uri: The latest_snapshot_uri of this WorkspaceDataplaneProxiedArtifacts.  # noqa: E501
+        :type: str
+        """
+
+        self._latest_snapshot_uri = latest_snapshot_uri
+
     def to_dict(self):
         """Returns the model properties as a dict"""
         result = {}

anyscale/cluster_compute.py

@@ -4,20 +4,20 @@ from typing import Optional, Tuple, Union
 from anyscale.authenticate import get_auth_api_client
 from anyscale.cli_logger import BlockLogger
 from anyscale.client.openapi_client.api.default_api import DefaultApi
+from anyscale.client.openapi_client.models import (
+    ComputeTemplateQuery,
+    CreateComputeTemplate,
+)
 from anyscale.cloud_utils import get_cloud_id_and_name, get_last_used_cloud
 from anyscale.sdk.anyscale_client import (
     ArchiveStatus,
     ComputeTemplateConfig,
-    CreateComputeTemplate,
 )
 from anyscale.sdk.anyscale_client.api.default_api import DefaultApi as SDKDefaultApi
 from anyscale.sdk.anyscale_client.models.cluster_compute_config import (
     ClusterComputeConfig,
 )
 from anyscale.sdk.anyscale_client.models.compute_template import ComputeTemplate
-from anyscale.sdk.anyscale_client.models.compute_template_query import (
-    ComputeTemplateQuery,
-)
 from anyscale.utils.cloud_utils import get_organization_default_cloud
 
 

anyscale/commands/anyscale_api/sessions_commands.py

@@ -7,12 +7,8 @@ from typing import Optional
 import click
 
 from anyscale.authenticate import get_auth_api_client
+from anyscale.client.openapi_client.models import StartSessionOptions
 from anyscale.formatters import common_formatter
-from anyscale.sdk.anyscale_client import (
-    StartSessionOptions,
-    TerminateSessionOptions,
-    UpdateSession,
-)
 
 
 @click.group(
@@ -56,29 +52,6 @@ def get_session(session_id: str) -> None:
     print(common_formatter.prettify_json(response.to_dict()))
 
 
-@sessions.command(name="update", short_help="Updates a Session.")
-@click.argument("session_id", required=True)
-@click.argument("cluster_config", required=False)
-@click.option(
-    "--idle-timeout", required=False, help="Idle timeout (in minutes)", type=int,
-)
-def update_session(
-    session_id: str, cluster_config: Optional[str], idle_timeout: Optional[int]
-) -> None:
-    """
-    Updates Session SESSION_ID with CLUSTER_CONFIG and/or idle-timeout.
-    """
-
-    api_client = get_auth_api_client().anyscale_api_client
-    update_data = UpdateSession(
-        cluster_config=cluster_config, idle_timeout=idle_timeout
-    )
-    response = api_client.update_session(session_id, update_data)
-    print(response)
-
-    print(common_formatter.prettify_json(response.to_dict()))
-
-
 @sessions.command(name="delete", short_help="Deletes a Session.")
 @click.argument("session_id", required=True)
 def delete_session(session_id: str) -> None:
@@ -108,45 +81,3 @@ def start_session(session_id: str, cluster_config: Optional[str]) -> None:
     start_session_options = StartSessionOptions(cluster_config=cluster_config)
     response = api_client.start_session(session_id, start_session_options)
     print(common_formatter.prettify_json(response.to_dict()))
-
-
-@sessions.command(name="terminate", short_help="Sets session goal state to Terminated.")
-@click.argument("session_id", required=True)
-@click.argument("workers_only", required=False, default=False)
-@click.argument("keep_min_workers", required=False, default=False)
-@click.argument("delete", required=False, default=False)
-@click.argument("take_snapshot", required=False, default=True)
-def terminate_session(
-    session_id: str,
-    workers_only: bool,
-    keep_min_workers: bool,
-    delete: bool,
-    take_snapshot: bool,
-) -> None:
-    """Sets session goal state to Terminated.
-    A session with goal state Terminated will eventually
-    transition from its current state to Terminated, or
-    remain Terminated if its current state is already Terminated.
-    Retrieves the corresponding session operation.
-
-    workers_only: Default False. Only destroy the workers when stopping.
-
-    keep_min_workers: Default False. Retain the minimal amount of workers
-    specified in the config when stopping.
-
-    delete: Default False. Delete the session after terminating.
-
-    take_snapshot: Default True. Takes a snapshot to preserve the state of
-    the session before terminating. The state will be restored
-    the next time the session is started.
-    """
-
-    api_client = get_auth_api_client().anyscale_api_client
-    terminate_session_options = TerminateSessionOptions(
-        workers_only=workers_only,
-        keep_min_workers=keep_min_workers,
-        delete=delete,
-        take_snapshot=take_snapshot,
-    )
-    response = api_client.terminate_session(session_id, terminate_session_options)
-    print(common_formatter.prettify_json(response.to_dict()))

anyscale/commands/cloud_commands.py

@@ -436,7 +436,7 @@ def cloud_config_update(
     "--provider",
     help="The cloud provider type.",
     required=True,
-    type=click.Choice(["aws", "gcp", "generic"], case_sensitive=False),
+    type=click.Choice(["aws", "gcp", "azure", "generic"], case_sensitive=False),
 )
 @click.option(
     "--region",
@@ -861,18 +861,19 @@ def register_cloud(  # noqa: PLR0913, PLR0912, C901
             skip_verifications=skip_verifications,
             auto_add_user=enable_auto_add_user,
         )
-    elif provider == "generic":
+    elif provider in ("azure", "generic"):
         # For the 'generic' provider type, for the time being, most fields are optional; only 'name', 'provider', and 'compute-stack' are required.
         if not name:
             raise click.ClickException("Please provide a value for --name.")
 
         if compute_stack != ComputeStack.K8S:
             raise click.ClickException(
-                "--compute-stack=k8s must be passed for registering generic Anyscale clouds."
+                "--compute-stack=k8s must be passed to register this Anyscale cloud."
             )
 
-        CloudController().register_generic_cloud(
+        CloudController().register_azure_or_generic_cloud(
             name=name,
+            provider=provider,
             auto_add_user=enable_auto_add_user,
             region=region,
             cloud_storage_bucket_name=cloud_storage_bucket_name,

anyscale/commands/image_commands.py

@@ -39,7 +39,7 @@ def image_cli() -> None:
 @click.option(
     "--ray-version",
     "-r",
-    help="The Ray version (X.Y.Z) specified for this image specified by either an image URI or a containerfile. If not provided, the latest Ray version will be used.",
+    help="The Ray version (X.Y.Z) specified for this image specified by either an image URI or a containerfile. If you don't specify a Ray version, Anyscale defaults to the latest Ray version available at the time of the Anyscale CLI/SDK release.",
     type=str,
     default=None,
 )
@@ -100,7 +100,7 @@ def get(name: str) -> None:
 @click.option(
     "--ray-version",
     "-r",
-    help="The Ray version (X.Y.Z) specified for this image specified by either an image URI or a containerfile. If not provided, the latest Ray version will be used.",
+    help="The Ray version (X.Y.Z) specified for this image specified by either an image URI or a containerfile. If you don't specify a Ray version, Anyscale defaults to the latest Ray version available at the time of the Anyscale CLI/SDK release.",
     type=str,
     default=None,
 )

anyscale/commands/job_commands.py

@@ -148,7 +148,7 @@ def job_cli() -> None:
     required=False,
     default=None,
     type=str,
-    help="The Ray version (X.Y.Z) to the image specified by --image-uri. This is only used when --image-uri is provided. If not provided, the latest Ray version will be used.",
+    help="The Ray version (X.Y.Z) to the image specified by --image-uri. This is only used when --image-uri is provided. If you don't specify a Ray version, Anyscale defaults to the latest Ray version available at the time of the Anyscale CLI/SDK release.",
 )
 @click.option(
     "--max-retries",

anyscale/commands/service_commands.py

@@ -113,7 +113,7 @@ def _read_name_from_config_file(path: str):
     required=False,
     default=None,
     type=str,
-    help="The Ray version (X.Y.Z) to the image specified by --image-uri. This is only used when --image-uri is provided. If not provided, the latest Ray version will be used.",
+    help="The Ray version (X.Y.Z) to the image specified by --image-uri. This is only used when --image-uri is provided. If you don't specify a Ray version, Anyscale defaults to the latest Ray version available at the time of the Anyscale CLI/SDK release.",
 )
 @click.option(
     "--compute-config",

anyscale/commands/workspace_commands_v2.py

@@ -86,7 +86,7 @@ def workspace_cli() -> None:
     required=False,
     default=None,
     type=str,
-    help="The Ray version (X.Y.Z) to the image specified by --image-uri. This is only used when --image-uri is provided. If not provided, the latest Ray version will be used.",
+    help="The Ray version (X.Y.Z) to the image specified by --image-uri. This is only used when --image-uri is provided. If you don't specify a Ray version, Anyscale defaults to the latest Ray version available at the time of the Anyscale CLI/SDK release.",
 )
 @click.option(
     "--compute-config",

anyscale/connect_utils/prepare_cluster.py

@@ -8,8 +8,7 @@ from packaging import version
 
 from anyscale.authenticate import get_auth_api_client
 from anyscale.cli_logger import BlockLogger
-from anyscale.client.openapi_client.models.build import Build
-from anyscale.client.openapi_client.models.session import Session
+from anyscale.client.openapi_client.models import Build, CreateComputeTemplate, Session
 from anyscale.cluster_compute import (
     get_cluster_compute_from_name,
     get_selected_cloud_id_or_default,
@@ -19,7 +18,6 @@ from anyscale.links import DOCS_CLUSTER
 from anyscale.sdk.anyscale_client import (
     ComputeTemplateConfig,
     CreateCluster,
-    CreateComputeTemplate,
     StartClusterOptions,
     UpdateCluster,
 )

anyscale/controllers/cloud_controller.py

@@ -2184,9 +2184,10 @@ class CloudController(BaseController):
                 f"{quota_error_str}\n\nFor instructions on how to increase quotas, visit this link: https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/ec2-resource-limits.html#request-increase"
             )
 
-    def register_generic_cloud(  # noqa: PLR0913
+    def register_azure_or_generic_cloud(  # noqa: PLR0913
         self,
         name: str,
+        provider: str,
         auto_add_user: bool = False,
         # Optional cloud-resource-scoped parameters.
         # Some of these are conditionally required.
@@ -2198,8 +2199,11 @@ class CloudController(BaseController):
         nfs_mount_path: Optional[str] = None,
         kubernetes_zones: Optional[List[str]] = None,
     ) -> None:
+        cloud_provider = (
+            CloudProviders.AZURE if provider == "azure" else CloudProviders.GENERIC
+        )
         self.cloud_event_producer.init_trace_context(
-            CloudAnalyticsEventCommandName.REGISTER, CloudProviders.GENERIC
+            CloudAnalyticsEventCommandName.REGISTER, cloud_provider,
         )
         self.cloud_event_producer.produce(
             CloudAnalyticsEventName.COMMAND_START, succeeded=True
@@ -2228,7 +2232,7 @@ class CloudController(BaseController):
                 write_cloud=WriteCloud(
                     name=name,
                     region=region,
-                    provider="GENERIC",
+                    provider=cloud_provider,
                     is_bring_your_own_resource=True,
                     cluster_management_stack_version=ClusterManagementStackVersions.V2,
                     auto_add_user=auto_add_user,
@@ -2984,6 +2988,7 @@ class CloudController(BaseController):
         assert cloud_provider in (
             CloudProviders.AWS,
             CloudProviders.GCP,
+            CloudProviders.AZURE,
             CloudProviders.GENERIC,
         ), f"Cloud provider {cloud_provider} not supported yet"
 
@@ -3018,9 +3023,10 @@ class CloudController(BaseController):
 
         # set cloud state to DELETING
         try:
-            if (
-                cloud_provider == CloudProviders.GENERIC
-                or cloud_provider == CloudProviders.AWS
+            if cloud_provider in (
+                CloudProviders.AWS,
+                CloudProviders.AZURE,
+                CloudProviders.GENERIC,
             ):
                 with self.log.spinner("Preparing to delete Anyscale cloud..."):
                     response = self.api_client.update_cloud_with_cloud_resource_api_v2_clouds_with_cloud_resource_router_cloud_id_put(

anyscale/controllers/cloud_functional_verification_controller.py

@@ -25,6 +25,7 @@ from anyscale.client.openapi_client.models import (
     ApplyProductionServiceV2Model,
     CloudAnalyticsEventName,
     CloudProviders,
+    ComputeTemplateQuery,
     CreateExperimentalWorkspace,
     ServiceEventCurrentState,
     SessionState,
@@ -33,7 +34,6 @@ from anyscale.controllers.base_controller import BaseController
 from anyscale.project_utils import get_default_project
 from anyscale.sdk.anyscale_client.models import (
     ComputeNodeType,
-    ComputeTemplateQuery,
     CreateClusterCompute,
     CreateClusterComputeConfig,
     ServiceConfig,
@@ -50,7 +50,7 @@ SERVICE_VERIFICATION_TIMEOUT_MINUTES = 30  # for a single rollout
 MAXIMUM_UPTIME_MINUTES = 15
 IDLE_TERMINATION_MINUTES = 5
 HEAD_NODE_TYPE_AWS = "m5.xlarge"  # on demand price ~$0.20 per hour
-HEAD_NODE_TYPE_GCP = "n1-standard-2"  # on demand price ~$0.11 per hour
+HEAD_NODE_TYPE_GCP = "n2-highmem-2"  # on demand price ~$0.13 per hour
 CREATE_COMPUTE_CONFIG_TIMEOUT_SECONDS = 600  # 10 minutes
 
 # Workspace verification will fail fast if any of the following logs are found

anyscale/image/commands.py

@@ -85,7 +85,7 @@ image_uri: str = anyscale.image.register("docker.io/myuser/myimage:v2", name="my
 _REGISTER_ARG_DOCSTRINGS = {
     "image_uri": "The URI of the BYOD image to register.",
     "name": "Name for the container image. If the name already exists, a new version will be built. Otherwise, a new container image will be created.",
-    "ray_version": "The Ray version (X.Y.Z) specified for this image specified by either an image URI or a containerfile. If not provided, the latest Ray version will be used.",
+    "ray_version": "The Ray version (X.Y.Z) specified for this image specified by either an image URI or a containerfile. If you don't specify a Ray version, Anyscale defaults to the latest Ray version available at the time of the Anyscale CLI/SDK release.",
     "registry_login_secret": "Name or identifier of the secret containing credentials to authenticate to the docker registry hosting the image.",  # pragma: allowlist secret
 }