anyscale 0.26.66__py3-none-any.whl → 0.26.67__py3-none-any.whl

anyscale/commands/cloud_commands.py

@@ -23,6 +23,7 @@ from anyscale.client.openapi_client.models import (
 from anyscale.client.openapi_client.models.compute_stack import ComputeStack
 from anyscale.cloud.models import CreateCloudCollaborator, CreateCloudCollaborators
 from anyscale.commands import command_examples
+from anyscale.commands.setup_k8s import setup_kubernetes_cloud
 from anyscale.commands.util import AnyscaleCommand, OptionPromptNull
 from anyscale.controllers.cloud_controller import CloudController
 from anyscale.util import (
@@ -127,6 +128,27 @@ def default_region(provider: str) -> str:
     show_default=True,
 )
 @click.option("--name", "-n", help="Name of the cloud.", required=True, prompt="Name")
+@click.option(
+    "--stack",
+    help="The compute stack to use (vm or k8s).",
+    required=False,
+    type=click.Choice(["vm", "k8s"], case_sensitive=False),
+    default="vm",
+    show_default=True,
+)
+@click.option(
+    "--cluster-name",
+    help="Kubernetes cluster name (required for k8s stack).",
+    required=False,
+    type=str,
+)
+@click.option(
+    "--namespace",
+    help="Kubernetes namespace for Anyscale operator (for k8s stack).",
+    required=False,
+    type=str,
+    default="anyscale-operator",
+)
 @click.option(
     "--project-id",
     help="Globally Unique project ID for GCP clouds (e.g., my-project-abc123)",
@@ -173,10 +195,22 @@ def default_region(provider: str) -> str:
     show_default=True,
     help="The type of shared storage to use for the cloud. Use 'object-storage' for cloud bucket-based storage (e.g., S3, GCS), or 'nfs' for network file systems.",
 )
+@click.option(
+    "--values-file",
+    help="Path to save the generated Helm values file (for k8s stack, default: auto-generated with timestamp).",
+    required=False,
+    type=str,
+)
+@click.option(
+    "--debug", is_flag=True, default=False, help="Enable debug logging.",
+)
 def setup_cloud(  # noqa: PLR0913
     provider: str,
     region: str,
     name: str,
+    stack: str,
+    cluster_name: Optional[str],
+    namespace: str,
     project_id: str,
     functional_verify: Optional[str],
     anyscale_managed: bool,  # noqa: ARG001
@@ -184,9 +218,33 @@ def setup_cloud(  # noqa: PLR0913
     yes: bool,
     disable_auto_add_user: bool,
     shared_storage: str,
+    values_file: Optional[str],
+    debug: bool,
 ) -> None:
     # TODO (congding): remove `anyscale_managed` in the future, now keeping it for compatibility
 
+    # Handle Kubernetes stack
+    if stack == "k8s":
+        if not cluster_name:
+            raise click.ClickException(
+                "--cluster-name is required when using --stack=k8s"
+            )
+
+        setup_kubernetes_cloud(
+            provider=provider,
+            region=region,
+            name=name,
+            cluster_name=cluster_name,
+            namespace=namespace,
+            project_id=project_id,
+            functional_verify=bool(functional_verify),
+            yes=yes,
+            values_file=values_file,
+            debug=debug,
+        )
+        return
+
+    # Handle VM stack
     # Convert string to enum for type safety
     shared_storage_type = SharedStorageType(shared_storage)
     if provider == "aws":