anyscale 0.26.31__py3-none-any.whl → 0.26.32__py3-none-any.whl

anyscale/cloud/_private/cloud_sdk.py

@@ -1,6 +1,7 @@
 from typing import List, Optional
 
 from anyscale._private.sdk.base_sdk import BaseSDK
+from anyscale.cli_logger import BlockLogger
 from anyscale.client.openapi_client.models import (
     Cloud as CloudModel,
     CreateCloudCollaborator as CreateCloudCollaboratorModel,
@@ -11,6 +12,10 @@ from anyscale.cloud.models import (
     ComputeStack,
     CreateCloudCollaborator,
 )
+from anyscale.sdk.anyscale_client.models import ClusterState
+
+
+logger = BlockLogger()
 
 
 class PrivateCloudSDK(BaseSDK):
@@ -80,3 +85,31 @@ class PrivateCloudSDK(BaseSDK):
             is_default=openapi_cloud.is_default,
             compute_stack=compute_stack,
         )
+
+    def terminate_system_cluster(self, cloud_id: str, wait: bool) -> str:
+        resp = self.client.terminate_system_cluster(cloud_id)
+        if wait:
+            self._wait_for_system_cluster_status(cloud_id, ClusterState.TERMINATED)
+        else:
+            logger.info(f"System cluster termination initiated for cloud {cloud_id}.")
+        return resp.result.cluster_id
+
+    def _wait_for_system_cluster_status(
+        self,
+        cloud_id: str,
+        goal_status: str,
+        timeout_s: int = 500,
+        interval_s: int = 10,
+    ) -> bool:
+        self.logger.info("Waiting for system cluster termination...", end="")
+        for _ in self.timer.poll(timeout_s=timeout_s, interval_s=interval_s):
+            status = self.client.describe_system_workload_get_status(cloud_id)
+            if status == goal_status:
+                print(".")
+                self.logger.info(f"System cluster for cloud '{cloud_id}' is {status}.")
+                return True
+            else:
+                print(".", end="")
+        raise TimeoutError(
+            f"Timed out waiting for system cluster termination for cloud '{cloud_id}'. Last seen status: {status}."
+        )

anyscale/cloud/commands.py

@@ -114,3 +114,38 @@ def get_default(*, _private_sdk: Optional[PrivateCloudSDK] = None) -> Optional[C
     :return: The default `Cloud` object if it exists, otherwise `None`.
     """
     return _private_sdk.get_default()  # type: ignore
+
+
+_TERMINATE_SYSTEM_CLUSTER_EXAMPLE = """
+import anyscale
+
+# Terminate the system cluster for the cloud with the specified ID
+anyscale.cloud.terminate_system_cluster(cloud_id="cloud_id", wait=True)
+"""
+
+_TERMINATE_SYSTEM_CLUSTER_ARG_DOCSTRINGS = {
+    "cloud_id": "The ID of the cloud whose system cluster should be terminated.",
+    "wait": "If True, wait for the system cluster to be terminated before returning. Defaults to False.",
+}
+
+
+@sdk_command(
+    _CLOUD_SDK_SINGLETON_KEY,
+    PrivateCloudSDK,
+    doc_py_example=_TERMINATE_SYSTEM_CLUSTER_EXAMPLE,
+    arg_docstrings=_TERMINATE_SYSTEM_CLUSTER_ARG_DOCSTRINGS,
+)
+def terminate_system_cluster(
+    cloud_id: str,
+    wait: Optional[bool] = False,
+    *,
+    _private_sdk: Optional[PrivateCloudSDK] = None,
+) -> str:
+    """
+    Terminate the system cluster for the specified cloud.
+
+    :param cloud: The name of the cloud whose system cluster should be terminated.
+    :param wait: If True, wait for the system cluster to be terminated before returning. Defaults to False.
+    :return: ID of the terminated system cluster.
+    """
+    return _private_sdk.terminate_system_cluster(cloud_id, wait)  # type: ignore

anyscale/commands/cloud_commands.py

@@ -1269,3 +1269,38 @@ def generate_jobs_report(
         )
     except ValueError as e:
         log.error(f"Error generating jobs report: {e}")
+
+
+@cloud_cli.command(
+    name="terminate-system-cluster",
+    help="Terminate the system cluster for a specific given cloud.",
+    cls=AnyscaleCommand,
+    example=command_examples.CLOUD_TERMINATE_SYSTEM_CLUSTER_EXAMPLE,
+)
+@click.option(
+    "--cloud-id",
+    "--id",
+    help="ID of the cloud to terminate the system cluster for.",
+    type=str,
+    required=True,
+)
+@click.option(
+    "-w",
+    "--wait",
+    required=False,
+    default=False,
+    type=bool,
+    is_flag=True,
+    help="Block this CLI command and print logs until the job finishes.",
+)
+def terminate_system_cluster(cloud_id: str, wait: Optional[bool]) -> None:
+    """
+    Terminate the system cluster for a specific cloud.
+
+    :param cloud_id: The ID of the cloud to terminate the system cluster for.
+    :param wait: If True, wait for the system cluster to be terminated before returning. Defaults to False.
+    """
+    try:
+        anyscale.cloud.terminate_system_cluster(cloud_id, wait)
+    except ValueError as e:
+        log.error(f"Error terminating system cluster: {e}")

anyscale/commands/command_examples.py

@@ -673,6 +673,12 @@ is_default: true
 compute_stack: VM
 """
 
+CLOUD_TERMINATE_SYSTEM_CLUSTER_EXAMPLE = """\
+$ anyscale cloud terminate-system-cluster --cloud-id cloud_id --wait
+(anyscale +1.3s) Waiting for system cluster termination............
+(anyscale +1m22.9s) System cluster for cloud 'cloud_id' is Terminated.
+"""
+
 SERVICE_ARCHIVE_EXAMPLE = """\
 $ anyscale service archive --name my_service
 """