anyscale 0.26.22__py3-none-any.whl → 0.26.24__py3-none-any.whl

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
 }
 

anyscale/sdk/anyscale_client/__init__.py

@@ -37,13 +37,12 @@ from anyscale_client.models.archive_status import ArchiveStatus
 from anyscale_client.models.baseimagesenum import BASEIMAGESENUM
 from anyscale_client.models.base_job_status import BaseJobStatus
 from anyscale_client.models.build import Build
-from anyscale_client.models.build_list_response import BuildListResponse
-from anyscale_client.models.build_log_response import BuildLogResponse
 from anyscale_client.models.build_response import BuildResponse
 from anyscale_client.models.build_status import BuildStatus
-from anyscale_client.models.buildlogresponse_response import BuildlogresponseResponse
 from anyscale_client.models.cloud import Cloud
 from anyscale_client.models.cloud_config import CloudConfig
+from anyscale_client.models.cloud_deployment_compute_config import CloudDeploymentComputeConfig
+from anyscale_client.models.cloud_deployment_selector import CloudDeploymentSelector
 from anyscale_client.models.cloud_list_response import CloudListResponse
 from anyscale_client.models.cloud_providers import CloudProviders
 from anyscale_client.models.cloud_response import CloudResponse
@@ -87,8 +86,6 @@ from anyscale_client.models.compute_node_type import ComputeNodeType
 from anyscale_client.models.compute_stack import ComputeStack
 from anyscale_client.models.compute_template import ComputeTemplate
 from anyscale_client.models.compute_template_config import ComputeTemplateConfig
-from anyscale_client.models.compute_template_query import ComputeTemplateQuery
-from anyscale_client.models.computetemplate_list_response import ComputetemplateListResponse
 from anyscale_client.models.computetemplate_response import ComputetemplateResponse
 from anyscale_client.models.computetemplateconfig_response import ComputetemplateconfigResponse
 from anyscale_client.models.create_app_config_configuration_schema import CreateAppConfigConfigurationSchema
@@ -96,7 +93,6 @@ from anyscale_client.models.create_byod_app_config_configuration_schema import C
 from anyscale_client.models.create_byod_cluster_environment import CreateBYODClusterEnvironment
 from anyscale_client.models.create_byod_cluster_environment_build import CreateBYODClusterEnvironmentBuild
 from anyscale_client.models.create_byod_cluster_environment_configuration_schema import CreateBYODClusterEnvironmentConfigurationSchema
-from anyscale_client.models.create_build import CreateBuild
 from anyscale_client.models.create_cloud import CreateCloud
 from anyscale_client.models.create_cluster import CreateCluster
 from anyscale_client.models.create_cluster_compute import CreateClusterCompute
@@ -104,8 +100,6 @@ from anyscale_client.models.create_cluster_compute_config import CreateClusterCo
 from anyscale_client.models.create_cluster_environment import CreateClusterEnvironment
 from anyscale_client.models.create_cluster_environment_build import CreateClusterEnvironmentBuild
 from anyscale_client.models.create_cluster_environment_configuration_schema import CreateClusterEnvironmentConfigurationSchema
-from anyscale_client.models.create_compute_template import CreateComputeTemplate
-from anyscale_client.models.create_compute_template_config import CreateComputeTemplateConfig
 from anyscale_client.models.create_job_queue_config import CreateJobQueueConfig
 from anyscale_client.models.create_production_job import CreateProductionJob
 from anyscale_client.models.create_production_job_config import CreateProductionJobConfig
@@ -184,7 +178,6 @@ from anyscale_client.models.service_version_state import ServiceVersionState
 from anyscale_client.models.servicemodel_list_response import ServicemodelListResponse
 from anyscale_client.models.servicemodel_response import ServicemodelResponse
 from anyscale_client.models.session import Session
-from anyscale_client.models.session_list_response import SessionListResponse
 from anyscale_client.models.session_operation import SessionOperation
 from anyscale_client.models.session_operation_type import SessionOperationType
 from anyscale_client.models.session_response import SessionResponse
@@ -193,26 +186,20 @@ from anyscale_client.models.session_state import SessionState
 from anyscale_client.models.session_state_data import SessionStateData
 from anyscale_client.models.session_stopping_data import SessionStoppingData
 from anyscale_client.models.sessionoperation_response import SessionoperationResponse
-from anyscale_client.models.sessions_query import SessionsQuery
 from anyscale_client.models.sort_by_clause_jobs_sort_field import SortByClauseJobsSortField
 from anyscale_client.models.sort_order import SortOrder
 from anyscale_client.models.ssoconfig_response import SsoconfigResponse
 from anyscale_client.models.start_cluster_options import StartClusterOptions
-from anyscale_client.models.start_session_options import StartSessionOptions
 from anyscale_client.models.static_sso_config import StaticSSOConfig
 from anyscale_client.models.system_workload_name import SystemWorkloadName
 from anyscale_client.models.terminate_cluster_options import TerminateClusterOptions
-from anyscale_client.models.terminate_session_options import TerminateSessionOptions
 from anyscale_client.models.text_query import TextQuery
 from anyscale_client.models.tracing_config import TracingConfig
 from anyscale_client.models.ux_instance import UXInstance
 from anyscale_client.models.update_cloud import UpdateCloud
 from anyscale_client.models.update_cluster import UpdateCluster
-from anyscale_client.models.update_compute_template import UpdateComputeTemplate
-from anyscale_client.models.update_compute_template_config import UpdateComputeTemplateConfig
 from anyscale_client.models.update_organization import UpdateOrganization
 from anyscale_client.models.update_project import UpdateProject
-from anyscale_client.models.update_session import UpdateSession
 from anyscale_client.models.user_service_access_types import UserServiceAccessTypes
 from anyscale_client.models.validation_error import ValidationError
 from anyscale_client.models.worker_node_type import WorkerNodeType