modal 0.74.55__tar.gz → 0.74.56__tar.gz

{modal-0.74.55 → modal-0.74.56}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modal
-Version: 0.74.55
+Version: 0.74.56
 Summary: Python client library for Modal
 Author-email: Modal Labs <support@modal.com>
 License: Apache-2.0

modal-0.74.56/modal/cli/cluster.py

@@ -0,0 +1,82 @@
+# Copyright Modal Labs 2022
+from typing import Optional, Union
+
+import typer
+from rich.console import Console
+from rich.text import Text
+
+from modal._object import _get_environment_name
+from modal._pty import get_pty_info
+from modal._utils.async_utils import synchronizer
+from modal.cli.utils import ENV_OPTION, display_table, is_tty, timestamp_to_local
+from modal.client import _Client
+from modal.config import config
+from modal.container_process import _ContainerProcess
+from modal.environments import ensure_env
+from modal.stream_type import StreamType
+from modal_proto import api_pb2
+
+cluster_cli = typer.Typer(
+    name="cluster", help="Manage and connect to running multi-node clusters.", no_args_is_help=True
+)
+
+
+@cluster_cli.command("list")
+@synchronizer.create_blocking
+async def list_(env: Optional[str] = ENV_OPTION, json: bool = False):
+    """List all clusters that are currently running."""
+    env = ensure_env(env)
+    client = await _Client.from_env()
+    environment_name = _get_environment_name(env)
+    res: api_pb2.ClusterListResponse = await client.stub.ClusterList(
+        api_pb2.ClusterListRequest(environment_name=environment_name)
+    )
+
+    column_names = ["Cluster ID", "App ID", "Start Time", "Nodes"]
+    rows: list[list[Union[Text, str]]] = []
+    res.clusters.sort(key=lambda c: c.started_at, reverse=True)
+
+    for c in res.clusters:
+        rows.append(
+            [
+                c.cluster_id,
+                c.app_id,
+                timestamp_to_local(c.started_at, json) if c.started_at else "Pending",
+                str(len(c.task_ids)),
+            ]
+        )
+
+    display_table(column_names, rows, json=json, title=f"Active Multi-node Clusters in environment: {environment_name}")
+
+
+@cluster_cli.command("shell")
+@synchronizer.create_blocking
+async def shell(
+    cluster_id: str = typer.Argument(help="Cluster ID"),
+    rank: int = typer.Option(default=0, help="Rank of the node to shell into"),
+):
+    """Open a shell to a multi-node cluster node."""
+    client = await _Client.from_env()
+    res: api_pb2.ClusterGetResponse = await client.stub.ClusterGet(api_pb2.ClusterGetRequest(cluster_id=cluster_id))
+    if len(res.cluster.task_ids) <= rank:
+        raise typer.Abort(f"No node with rank {rank} in cluster {cluster_id}")
+    task_id = res.cluster.task_ids[rank]
+    console = Console()
+    is_main = "(main)" if rank == 0 else ""
+    console.print(
+        f"Opening shell to node {rank} {is_main} of cluster {cluster_id} (container {task_id})", style="green"
+    )
+
+    pty = is_tty()
+    req = api_pb2.ContainerExecRequest(
+        task_id=task_id,
+        command=["/bin/bash"],
+        pty_info=get_pty_info(shell=True) if pty else None,
+        runtime_debug=config.get("function_runtime_debug"),
+    )
+    exec_res: api_pb2.ContainerExecResponse = await client.stub.ContainerExec(req)
+    if pty:
+        await _ContainerProcess(exec_res.exec_id, client).attach()
+    else:
+        # TODO: redirect stderr to its own stream?
+        await _ContainerProcess(exec_res.exec_id, client, stdout=StreamType.STDOUT, stderr=StreamType.STDOUT).wait()

{modal-0.74.55 → modal-0.74.56}/modal/cli/entry_point.py

@@ -10,6 +10,7 @@ from modal._utils.async_utils import synchronizer
 
 from . import run
 from .app import app_cli
+from .cluster import cluster_cli
 from .config import config_cli
 from .container import container_cli
 from .dict import dict_cli
@@ -92,6 +93,8 @@ entrypoint_cli_typer.add_typer(launch_cli)
 # Deployments
 entrypoint_cli_typer.add_typer(app_cli, rich_help_panel="Deployments")
 entrypoint_cli_typer.add_typer(container_cli, rich_help_panel="Deployments")
+# TODO: cluster is hidden while multi-node is in beta/experimental
+entrypoint_cli_typer.add_typer(cluster_cli, rich_help_panel="Deployments", hidden=True)
 
 # Storage
 entrypoint_cli_typer.add_typer(dict_cli, rich_help_panel="Storage")

{modal-0.74.55 → modal-0.74.56}/modal/client.pyi

@@ -27,7 +27,7 @@ class _Client:
     _snapshotted: bool
 
     def __init__(
-        self, server_url: str, client_type: int, credentials: typing.Optional[tuple[str, str]], version: str = "0.74.55"
+        self, server_url: str, client_type: int, credentials: typing.Optional[tuple[str, str]], version: str = "0.74.56"
     ): ...
     def is_closed(self) -> bool: ...
     @property
@@ -86,7 +86,7 @@ class Client:
     _snapshotted: bool
 
     def __init__(
-        self, server_url: str, client_type: int, credentials: typing.Optional[tuple[str, str]], version: str = "0.74.55"
+        self, server_url: str, client_type: int, credentials: typing.Optional[tuple[str, str]], version: str = "0.74.56"
     ): ...
     def is_closed(self) -> bool: ...
     @property

{modal-0.74.55 → modal-0.74.56}/modal.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modal
-Version: 0.74.55
+Version: 0.74.56
 Summary: Python client library for Modal
 Author-email: Modal Labs <support@modal.com>
 License: Apache-2.0

{modal-0.74.55 → modal-0.74.56}/modal.egg-info/SOURCES.txt

@@ -127,6 +127,7 @@ modal/cli/__init__.py
 modal/cli/_download.py
 modal/cli/_traceback.py
 modal/cli/app.py
+modal/cli/cluster.py
 modal/cli/config.py
 modal/cli/container.py
 modal/cli/dict.py

{modal-0.74.55 → modal-0.74.56}/modal_version/_version_generated.py

@@ -1,4 +1,4 @@
 # Copyright Modal Labs 2025
 
 # Note: Reset this value to -1 whenever you make a minor `0.X` release of the client.
-build_number = 55  # git: aa1b399
+build_number = 56  # git: 9749b47