modAL 1.4.3.dev7__py3-none-any.whl → 1.4.3.dev8__py3-none-any.whl

modal/cli/cluster.py

@@ -1,4 +1,5 @@
 # Copyright Modal Labs 2022
+import uuid
 from typing import Optional, Union
 
 import click
@@ -8,15 +9,17 @@ from rich.text import Text
 from modal._object import _get_environment_name
 from modal._output.pty import get_pty_info
 from modal._utils.async_utils import synchronizer
+from modal._utils.task_command_router_client import TaskCommandRouterClient
 from modal._utils.time_utils import timestamp_to_localized_str
 from modal.cli.utils import display_table, env_option, is_tty
 from modal.client import _Client
 from modal.config import config
 from modal.container_process import _ContainerProcess
 from modal.environments import ensure_env
+from modal.exception import InvalidError
 from modal.output import OutputManager
 from modal.stream_type import StreamType
-from modal_proto import api_pb2
+from modal_proto import api_pb2, task_command_router_pb2 as sr_pb2
 
 from ._help import ModalGroup
 
@@ -75,17 +78,32 @@ async def shell(cluster_id: str, rank: int = 0):
     )
 
     pty = is_tty()
-    req = api_pb2.ContainerExecRequest(
+
+    command_router_client = await TaskCommandRouterClient.try_init(client, task_id)
+    if command_router_client is None:
+        raise InvalidError(f"Command router access is not available for container {task_id}")
+
+    process_id = str(uuid.uuid4())
+
+    start_req = sr_pb2.TaskExecStartRequest(
         task_id=task_id,
-        command=["/bin/bash"],
+        exec_id=process_id,
+        command_args=["/bin/bash"],
+        stdout_config=sr_pb2.TaskExecStdoutConfig.TASK_EXEC_STDOUT_CONFIG_PIPE,
+        stderr_config=sr_pb2.TaskExecStderrConfig.TASK_EXEC_STDERR_CONFIG_PIPE,
         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)
+    await command_router_client.exec_start(start_req)
+
     if pty:
-        await _ContainerProcess(exec_res.exec_id, task_id, client).attach()
+        await _ContainerProcess(process_id, task_id, client, command_router_client=command_router_client).attach()
     else:
-        # TODO: redirect stderr to its own stream?
         await _ContainerProcess(
-            exec_res.exec_id, task_id, client, stdout=StreamType.STDOUT, stderr=StreamType.STDOUT
+            process_id,
+            task_id,
+            client,
+            command_router_client=command_router_client,
+            stdout=StreamType.STDOUT,
+            stderr=StreamType.STDOUT,
         ).wait()

modal/cli/container.py

@@ -1,4 +1,5 @@
 # Copyright Modal Labs 2022
+import uuid
 import warnings
 from datetime import datetime, timezone
 from typing import Optional, Union
@@ -12,6 +13,7 @@ from modal._logs import _FETCH_LIMIT, _MAX_FETCH_RANGE, LogsFilters
 from modal._object import _get_environment_name
 from modal._output.pty import get_pty_info
 from modal._utils.async_utils import synchronizer
+from modal._utils.task_command_router_client import TaskCommandRouterClient
 from modal._utils.time_utils import timestamp_to_localized_str
 from modal.cli.app import _DEFAULT_LOGS_TAIL, _SOURCE_OPTIONS, _parse_time_arg
 from modal.cli.utils import (
@@ -30,7 +32,7 @@ from modal.container_process import _ContainerProcess
 from modal.environments import ensure_env
 from modal.exception import InvalidError
 from modal.stream_type import StreamType
-from modal_proto import api_pb2
+from modal_proto import api_pb2, task_command_router_pb2 as sr_pb2
 
 from ._help import ModalGroup
 
@@ -276,20 +278,33 @@ async def _exec_impl(
 
     client = await _Client.from_env()
 
-    req = api_pb2.ContainerExecRequest(
+    command_router_client = await TaskCommandRouterClient.try_init(client, container_id)
+    if command_router_client is None:
+        raise InvalidError(f"Command router access is not available for container {container_id}")
+
+    process_id = str(uuid.uuid4())
+
+    start_req = sr_pb2.TaskExecStartRequest(
         task_id=container_id,
-        command=command,
+        exec_id=process_id,
+        command_args=command,
+        stdout_config=sr_pb2.TaskExecStdoutConfig.TASK_EXEC_STDOUT_CONFIG_PIPE,
+        stderr_config=sr_pb2.TaskExecStderrConfig.TASK_EXEC_STDERR_CONFIG_PIPE,
         pty_info=get_pty_info(shell=True) if pty else None,
         runtime_debug=config.get("function_runtime_debug"),
     )
-    res: api_pb2.ContainerExecResponse = await client.stub.ContainerExec(req)
+    await command_router_client.exec_start(start_req)
 
     if pty:
-        await _ContainerProcess(res.exec_id, container_id, client).attach()
+        await _ContainerProcess(process_id, container_id, client, command_router_client=command_router_client).attach()
     else:
-        # TODO: redirect stderr to its own stream?
         await _ContainerProcess(
-            res.exec_id, container_id, client, stdout=StreamType.STDOUT, stderr=StreamType.STDOUT
+            process_id,
+            container_id,
+            client,
+            command_router_client=command_router_client,
+            stdout=StreamType.STDOUT,
+            stderr=StreamType.STDOUT,
         ).wait()
 
 

modal/client.pyi

@@ -35,7 +35,7 @@ class _Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.4.3.dev7",
+        version: str = "1.4.3.dev8",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.
@@ -175,7 +175,7 @@ class Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.4.3.dev7",
+        version: str = "1.4.3.dev8",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.

modal/functions.pyi

@@ -347,7 +347,7 @@ class Function(
 
     _call_generator: ___call_generator_spec
 
-    class __remote_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER]):
+    class __remote_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER]):
         def __call__(self, /, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> ReturnType_INNER:
             """Calls the function remotely, executing it with the given arguments and returning the execution's result."""
             ...
@@ -356,7 +356,7 @@ class Function(
             """Calls the function remotely, executing it with the given arguments and returning the execution's result."""
             ...
 
-    remote: __remote_spec[modal._functions.P, modal._functions.ReturnType]
+    remote: __remote_spec[modal._functions.ReturnType, modal._functions.P]
 
     class __remote_gen_spec(typing_extensions.Protocol):
         def __call__(self, /, *args, **kwargs) -> typing.Generator[typing.Any, None, None]:
@@ -383,7 +383,7 @@ class Function(
         """
         ...
 
-    class ___experimental_spawn_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER]):
+    class ___experimental_spawn_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER]):
         def __call__(self, /, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]:
             """[Experimental] Calls the function with the given arguments, without waiting for the results.
 
@@ -406,7 +406,7 @@ class Function(
             """
             ...
 
-    _experimental_spawn: ___experimental_spawn_spec[modal._functions.P, modal._functions.ReturnType]
+    _experimental_spawn: ___experimental_spawn_spec[modal._functions.ReturnType, modal._functions.P]
 
     class ___spawn_map_inner_spec(typing_extensions.Protocol[P_INNER]):
         def __call__(self, /, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> None: ...
@@ -414,7 +414,7 @@ class Function(
 
     _spawn_map_inner: ___spawn_map_inner_spec[modal._functions.P]
 
-    class __spawn_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER]):
+    class __spawn_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER]):
         def __call__(self, /, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]:
             """Calls the function with the given arguments, without waiting for the results.
 
@@ -435,7 +435,7 @@ class Function(
             """
             ...
 
-    spawn: __spawn_spec[modal._functions.P, modal._functions.ReturnType]
+    spawn: __spawn_spec[modal._functions.ReturnType, modal._functions.P]
 
     def get_raw_f(self) -> collections.abc.Callable[..., typing.Any]:
         """Return the inner Python object wrapped by this Modal Function."""

modal/sandbox.py

@@ -628,6 +628,7 @@ class _Sandbox(_Object, type_prefix="sb"):
         idle_timeout: Optional[int] = None,
         workdir: Optional[str] = None,
         cpu: Optional[float] = None,
+        memory: Optional[int] = None,
         cloud: Optional[str] = None,
         region: Optional[Union[str, Sequence[str]]] = None,
         block_network: bool = False,
@@ -642,7 +643,6 @@ class _Sandbox(_Object, type_prefix="sb"):
     ) -> "_Sandbox":
         """Create a sandbox using the V2 backend.
 
-        Only CPU is configurable; memory is derived as a fixed ratio of CPU.
         Features like tags, snapshots, exec, volumes, network file systems,
         GPUs, custom domains, and proxies are not supported.
         """
@@ -714,7 +714,7 @@ class _Sandbox(_Object, type_prefix="sb"):
                 timeout_secs=timeout,
                 idle_timeout_secs=idle_timeout,
                 workdir=workdir,
-                resources=convert_fn_config_to_resources_config(cpu=cpu, memory=0, gpu=None, ephemeral_disk=None),
+                resources=convert_fn_config_to_resources_config(cpu=cpu, memory=memory, gpu=None, ephemeral_disk=None),
                 cloud_provider_str=cloud if cloud else None,
                 runtime=config.get("function_runtime"),
                 runtime_debug=config.get("function_runtime_debug"),

modal/sandbox.pyi

@@ -304,6 +304,7 @@ class _Sandbox(modal._object._Object):
         idle_timeout: typing.Optional[int] = None,
         workdir: typing.Optional[str] = None,
         cpu: typing.Optional[float] = None,
+        memory: typing.Optional[int] = None,
         cloud: typing.Optional[str] = None,
         region: typing.Union[str, collections.abc.Sequence[str], None] = None,
         block_network: bool = False,
@@ -318,7 +319,6 @@ class _Sandbox(modal._object._Object):
     ) -> _Sandbox:
         """Create a sandbox using the V2 backend.
 
-        Only CPU is configurable; memory is derived as a fixed ratio of CPU.
         Features like tags, snapshots, exec, volumes, network file systems,
         GPUs, custom domains, and proxies are not supported.
         """
@@ -1190,6 +1190,7 @@ class Sandbox(modal.object.Object):
             idle_timeout: typing.Optional[int] = None,
             workdir: typing.Optional[str] = None,
             cpu: typing.Optional[float] = None,
+            memory: typing.Optional[int] = None,
             cloud: typing.Optional[str] = None,
             region: typing.Union[str, collections.abc.Sequence[str], None] = None,
             block_network: bool = False,
@@ -1204,7 +1205,6 @@ class Sandbox(modal.object.Object):
         ) -> Sandbox:
             """Create a sandbox using the V2 backend.
 
-            Only CPU is configurable; memory is derived as a fixed ratio of CPU.
             Features like tags, snapshots, exec, volumes, network file systems,
             GPUs, custom domains, and proxies are not supported.
             """
@@ -1223,6 +1223,7 @@ class Sandbox(modal.object.Object):
             idle_timeout: typing.Optional[int] = None,
             workdir: typing.Optional[str] = None,
             cpu: typing.Optional[float] = None,
+            memory: typing.Optional[int] = None,
             cloud: typing.Optional[str] = None,
             region: typing.Union[str, collections.abc.Sequence[str], None] = None,
             block_network: bool = False,
@@ -1237,7 +1238,6 @@ class Sandbox(modal.object.Object):
         ) -> Sandbox:
             """Create a sandbox using the V2 backend.
 
-            Only CPU is configurable; memory is derived as a fixed ratio of CPU.
             Features like tags, snapshots, exec, volumes, network file systems,
             GPUs, custom domains, and proxies are not supported.
             """

{modal-1.4.3.dev7.dist-info → modal-1.4.3.dev8.dist-info}/METADATA

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

{modal-1.4.3.dev7.dist-info → modal-1.4.3.dev8.dist-info}/RECORD

@@ -26,7 +26,7 @@ modal/app.pyi,sha256=mysbOCD6OjZi938NeMY40RpdpQbnI43Bb4s5P79x5OQ,53773
 modal/billing.py,sha256=B2zXa7KeMnpqi2AQINycEk-KSXVYaLTTTAQTtuf2Gfs,257
 modal/call_graph.py,sha256=1g2DGcMIJvRy-xKicuf63IVE98gJSnQsr8R_NVMptNc,2581
 modal/client.py,sha256=jjT4eLFHGcvIF6Xw4kPptIx-VmExtphRN8XJ1JA7fWI,15502
-modal/client.pyi,sha256=TuZfFnS_-XxtHnhR7ZtOGru-1nIAUPwX-C8qQBZ2hHE,15701
+modal/client.pyi,sha256=xx85z1MlGyiGvYhlGkJ-ypngp-jblaCKIaUVJ5TZ31c,15701
 modal/cloud_bucket_mount.py,sha256=KkaJVy5vpO_u2NtDYtv1JM-M3y4J0UtWBapTDKpQRcA,6034
 modal/cloud_bucket_mount.pyi,sha256=mGDeEd1XvxGpCE6O5jRBSfIvMXvZ5AtIybea5k-0w9w,7560
 modal/cls.py,sha256=Yrl5b_tQhZMhc5avInM-i5UlGuqeog6Gs4rX2d9lLjI,40390
@@ -43,7 +43,7 @@ modal/file_io.py,sha256=45gO6ScEPKrE-KTQ619RwejaqoHUregwMgKcVktfoAk,22275
 modal/file_io.pyi,sha256=DPaq0BhySaiCX3g56QHhyIMdU1izbfWsjpSp30JoJUU,15745
 modal/file_pattern_matcher.py,sha256=A_Kdkej6q7YQyhM_2-BvpFmPqJ0oHb54B6yf9VqvPVE,8116
 modal/functions.py,sha256=EZRcHFfnoYbvawUJnAHF5nIvoo9GX2Djy7mUmOCXmXc,258
-modal/functions.pyi,sha256=m2uNU9DMi9LWprk9PJykCM1Zs_oEG2-Vl2FaJMGSNpY,34255
+modal/functions.pyi,sha256=_sZ27MFClkElRg0iyVXmn9pGXwm9F-r49sHjrYhH4Sg,34255
 modal/image.py,sha256=EHdzQMPBXtbwHBajZJAC-uLCypF-DoNbB0vMWVfaZ5c,112579
 modal/image.pyi,sha256=zYiJsc5tEhGev_rT9R3cGpyQJf-Hm08OJl37sV4E_fk,79377
 modal/io_streams.py,sha256=JcYbvvjMZFugv53yHc9Id0N4nIvDMwV63ZtnN1bPmZc,30940
@@ -68,8 +68,8 @@ modal/retries.py,sha256=vPWWKP-FRolJwTBuLPXdWqHV7aJfSQFgEW7vhlBBXRQ,5067
 modal/runner.py,sha256=yIB1PE5EGdvLWPPymyT4P0KYliREnFf-MAtULl3tVN0,30062
 modal/runner.pyi,sha256=ySefjmlMQ5FsvJgJkCWiUZKvcPNfEqdB5hwxI1fGtDo,9445
 modal/running_app.py,sha256=v61mapYNV1-O-Uaho5EfJlryMLvIT9We0amUOSvSGx8,1188
-modal/sandbox.py,sha256=ilVVQoCW4T_Ow5GQ-khqBdU1DSZuf2uqpg73myLpyMw,83656
-modal/sandbox.pyi,sha256=0rLb_VT1ALXcbrNUFlUhC0Z6pkh_-H6il8Layt5Z5Hc,86642
+modal/sandbox.py,sha256=byesQYe1QWRNB6yA3eTpr8TUIlb398U9kSoSm77X9P0,83622
+modal/sandbox.pyi,sha256=fDUIw3NsJPBRZ198lEo_Bw9WfESsqmUWcFLRnow8F-0,86546
 modal/sandbox_fs.py,sha256=IfsbUfYO1whNENRLP1UukH5TdYsdctveE1hvIPoKoGA,19789
 modal/sandbox_fs.pyi,sha256=0PlbdZmBERFbQ2ZgwHnqBZybD6SVOmWecBDuHQa3ZY8,31826
 modal/schedule.py,sha256=ng0g0AqNY5GQI9KhkXZQ5Wam5G42glbkqVQsNpBtbDE,3078
@@ -149,9 +149,9 @@ modal/cli/app.py,sha256=9mTkT2JZaGs2_GomWMk_WIXAuP4HelmgyqLUWhc-gwk,21990
 modal/cli/billing.py,sha256=afXmnVmak5gzPF9n-3v_WQ7jTr2idyZCKk7gKRKuYmo,5727
 modal/cli/bootstrap.py,sha256=tnEHGvLWc3oLy5hv2spB2-8AIcck8gLWgHDEI126Wl8,5300
 modal/cli/changelog.py,sha256=RSem_IBx4Zb6BsJbHUFs50PElm37O7fEjcL2f6FalSc,5848
-modal/cli/cluster.py,sha256=c33Yv3vgxYxWn1h9oVQHBrhHQpB4HcgSDimvMwBzVgY,3403
+modal/cli/cluster.py,sha256=3oh_3FNnyeoEXMQnNSHyDYhkjqtWaMGid2lyKm79CiY,4079
 modal/cli/config.py,sha256=2_PV3TKd7BA-5EhhskGS3_tUVMGUptimjw8KBEHvSQY,1857
-modal/cli/container.py,sha256=1vzF2yBWF0B6wxw0mMZ26PMlA6I8SnhxS8TXY1jdW1c,11553
+modal/cli/container.py,sha256=0aJFZnZ-Mck8G5Pdk6N4j1CwKKVOWvS-ndzAklQvyzY,12211
 modal/cli/dashboard.py,sha256=u3HzFNb4BgSBTW8ve034XNO0Tdt_z5P1kr8RcGFMxcw,941
 modal/cli/dict.py,sha256=DwZ-fbunPAgY9M678u4urL_SI1TrPxjs1Vavm52AKG0,5087
 modal/cli/entry_point.py,sha256=0g7C_BGDg6GPJSCKEOty9GPGXhkjlRI6UoP9oZVQdDI,4647
@@ -176,7 +176,7 @@ modal/experimental/__init__.py,sha256=Wb1YoAKAnNxH_0c2wA0T12I7bSu9PSO2nS_LuLska7
 modal/experimental/flash.py,sha256=ATGhVO8AleJ-TOe2PGolI5T7zfCEtj0kl_ivMPp7Rqg,33362
 modal/experimental/flash.pyi,sha256=R4h2nQl7-oiwqq_twCp0YqID_LnPjPGqxywXvxbiLIk,17011
 modal/experimental/ipython.py,sha256=TrCfmol9LGsRZMeDoeMPx3Hv3BFqQhYnmD_iH0pqdhk,2904
-modal-1.4.3.dev7.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
+modal-1.4.3.dev8.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
 modal_docs/__init__.py,sha256=svYKtV8HDwDCN86zbdWqyq5T8sMdGDj0PVlzc2tIxDM,28
 modal_docs/gen_cli_docs.py,sha256=bfUSLscHfoQoKkKjIqJvQMe3fURzqcQK7gF1h9WQLtM,3742
 modal_docs/gen_cli_docs_main.py,sha256=UPGr7O1DYTIRLwfINe2rbdh89J3cPVTU1OWL-SjxnOQ,198
@@ -193,15 +193,15 @@ modal_proto/api_pb2_grpc.py,sha256=dWnojBtfuaqHvBos1FL-JMNEB7LsNHM1UGUpKXO5IFY,3
 modal_proto/api_pb2_grpc.pyi,sha256=qDqqKIs18olWaoBjLOJ8Z6Gn8_GjpNn3Z1Pw1yG_M4I,77024
 modal_proto/modal_api_grpc.py,sha256=bBPhTehvCriGANMHxXNxKa58ew_2QbSvO88zhKCr4CE,24177
 modal_proto/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-modal_proto/task_command_router_grpc.py,sha256=wbRjiCyx9MlvkmPKM_6UVU1KmJaynUO4Eyt6tDKtmRk,12572
-modal_proto/task_command_router_pb2.py,sha256=ZrDKDK9Kw2Ks4l9-gWXjUG8nW50pyDY5G-DOwTGfF1U,24363
-modal_proto/task_command_router_pb2.pyi,sha256=qDvQBePeIQZcLmpYgEEiT5-7LnkpWTP9h-Lkzan5LXE,27289
-modal_proto/task_command_router_pb2_grpc.py,sha256=CW52Dj00YLRRAc2-KIz_aimGe7RuKvx0cROMs2No-b0,25002
-modal_proto/task_command_router_pb2_grpc.pyi,sha256=j4-VU3QFp8nUyx5_6F04T4IIU28KWHpAFMnY8Eg521g,8018
-modal_version/__init__.py,sha256=fJ4FvwhIrc4NwEftoGhdqyBjJ2OQuONLUSjj5FcuGz8,120
+modal_proto/task_command_router_grpc.py,sha256=i5ho152oU3hgfY3DZLXdT44ogcMy7rQUfGcEbmi0vXc,14477
+modal_proto/task_command_router_pb2.py,sha256=cFzU-s7Y1cnqUHIVrSFauBQ37Xv_jEQRDTHSNb-70t0,28015
+modal_proto/task_command_router_pb2.pyi,sha256=0fjfaucVSziP8VCLMalnqTaSfPCDjwMsWlD3EDQA_sk,30466
+modal_proto/task_command_router_pb2_grpc.py,sha256=XVmcdxLnK_IY3trlF6nLmzFhqSG-bbYNABC7xIs6fT8,28732
+modal_proto/task_command_router_pb2_grpc.pyi,sha256=og_J8CxYeAV0ikpHXZMvr_uC15UD3I3fgyyAO2pGm1E,8981
+modal_version/__init__.py,sha256=640BRe09B3s_LEYg_H83MOwI1-q8f4ES2t4l3udAsw4,120
 modal_version/__main__.py,sha256=nLUbubxKx7GXNzJIkTB7M7_u8080LyLM0IS0rzFkC_c,119
-modal-1.4.3.dev7.dist-info/METADATA,sha256=QLrVNmly1BjJoKEOm5y0DVrdyUbRagzvtsQ6tfyGT_8,2338
-modal-1.4.3.dev7.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-modal-1.4.3.dev7.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
-modal-1.4.3.dev7.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
-modal-1.4.3.dev7.dist-info/RECORD,,
+modal-1.4.3.dev8.dist-info/METADATA,sha256=MQF4REj1Xl3ByWCz5vMaifctLfGFRdzeFHEGQRU3GeE,2338
+modal-1.4.3.dev8.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+modal-1.4.3.dev8.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
+modal-1.4.3.dev8.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
+modal-1.4.3.dev8.dist-info/RECORD,,

modal_proto/task_command_router_grpc.py

@@ -16,6 +16,14 @@ import modal_proto.task_command_router_pb2
 
 class TaskCommandRouterBase(abc.ABC):
 
+    @abc.abstractmethod
+    async def SandboxStdinWriteV2(self, stream: 'grpclib.server.Stream[modal_proto.task_command_router_pb2.SandboxStdinWriteV2Request, modal_proto.task_command_router_pb2.SandboxStdinWriteV2Response]') -> None:
+        pass
+
+    @abc.abstractmethod
+    async def SandboxStdioReadV2(self, stream: 'grpclib.server.Stream[modal_proto.task_command_router_pb2.SandboxStdioReadV2Request, modal_proto.task_command_router_pb2.SandboxStdioReadV2Response]') -> None:
+        pass
+
     @abc.abstractmethod
     async def TaskContainerCreate(self, stream: 'grpclib.server.Stream[modal_proto.task_command_router_pb2.TaskContainerCreateRequest, modal_proto.task_command_router_pb2.TaskContainerCreateResponse]') -> None:
         pass
@@ -70,6 +78,18 @@ class TaskCommandRouterBase(abc.ABC):
 
     def __mapping__(self) -> typing.Dict[str, grpclib.const.Handler]:
         return {
+            '/modal.task_command_router.TaskCommandRouter/SandboxStdinWriteV2': grpclib.const.Handler(
+                self.SandboxStdinWriteV2,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.task_command_router_pb2.SandboxStdinWriteV2Request,
+                modal_proto.task_command_router_pb2.SandboxStdinWriteV2Response,
+            ),
+            '/modal.task_command_router.TaskCommandRouter/SandboxStdioReadV2': grpclib.const.Handler(
+                self.SandboxStdioReadV2,
+                grpclib.const.Cardinality.UNARY_STREAM,
+                modal_proto.task_command_router_pb2.SandboxStdioReadV2Request,
+                modal_proto.task_command_router_pb2.SandboxStdioReadV2Response,
+            ),
             '/modal.task_command_router.TaskCommandRouter/TaskContainerCreate': grpclib.const.Handler(
                 self.TaskContainerCreate,
                 grpclib.const.Cardinality.UNARY_UNARY,
@@ -154,6 +174,18 @@ class TaskCommandRouterBase(abc.ABC):
 class TaskCommandRouterStub:
 
     def __init__(self, channel: grpclib.client.Channel) -> None:
+        self.SandboxStdinWriteV2 = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.task_command_router.TaskCommandRouter/SandboxStdinWriteV2',
+            modal_proto.task_command_router_pb2.SandboxStdinWriteV2Request,
+            modal_proto.task_command_router_pb2.SandboxStdinWriteV2Response,
+        )
+        self.SandboxStdioReadV2 = grpclib.client.UnaryStreamMethod(
+            channel,
+            '/modal.task_command_router.TaskCommandRouter/SandboxStdioReadV2',
+            modal_proto.task_command_router_pb2.SandboxStdioReadV2Request,
+            modal_proto.task_command_router_pb2.SandboxStdioReadV2Response,
+        )
         self.TaskContainerCreate = grpclib.client.UnaryUnaryMethod(
             channel,
             '/modal.task_command_router.TaskCommandRouter/TaskContainerCreate',

modal_proto/task_command_router_pb2.py

@@ -17,14 +17,18 @@ from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
 from modal_proto import api_pb2 as modal__proto_dot_api__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%modal_proto/task_command_router.proto\x12\x19modal.task_command_router\x1a\x1bgoogle/protobuf/empty.proto\x1a\x15modal_proto/api.proto\"\x83\x02\n\x1aTaskContainerCreateRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x16\n\x0e\x63ontainer_name\x18\x02 \x01(\t\x12\x10\n\x08image_id\x18\x03 \x01(\t\x12\x0c\n\x04\x61rgs\x18\x05 \x03(\t\x12K\n\x03\x65nv\x18\x06 \x03(\x0b\x32>.modal.task_command_router.TaskContainerCreateRequest.EnvEntry\x12\x0f\n\x07workdir\x18\x07 \x01(\t\x12\x12\n\nsecret_ids\x18\x08 \x03(\t\x1a*\n\x08\x45nvEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"K\n\x1bTaskContainerCreateResponse\x12\x14\n\x0c\x63ontainer_id\x18\x01 \x01(\t\x12\x16\n\x0e\x63ontainer_name\x18\x02 \x01(\t\"^\n\x17TaskContainerGetRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x16\n\x0e\x63ontainer_name\x18\x02 \x01(\t\x12\x1a\n\x12include_terminated\x18\x03 \x01(\x08\"[\n\x18TaskContainerGetResponse\x12?\n\tcontainer\x18\x01 \x01(\x0b\x32,.modal.task_command_router.TaskContainerInfo\"~\n\x11TaskContainerInfo\x12\x14\n\x0c\x63ontainer_id\x18\x01 \x01(\t\x12\x16\n\x0e\x63ontainer_name\x18\x02 \x01(\t\x12\x0e\n\x06status\x18\x03 \x01(\t\x12+\n\x06result\x18\x04 \x01(\x0b\x32\x1b.modal.client.GenericResult\"G\n\x18TaskContainerListRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x1a\n\x12include_terminated\x18\x02 \x01(\x08\"]\n\x19TaskContainerListResponse\x12@\n\ncontainers\x18\x01 \x03(\x0b\x32,.modal.task_command_router.TaskContainerInfo\"F\n\x1dTaskContainerTerminateRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x14\n\x0c\x63ontainer_id\x18\x02 \x01(\t\" \n\x1eTaskContainerTerminateResponse\"R\n\x18TaskContainerWaitRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x14\n\x0c\x63ontainer_id\x18\x02 \x01(\t\x12\x0f\n\x07timeout\x18\x03 \x01(\x02\"H\n\x19TaskContainerWaitResponse\x12+\n\x06result\x18\x01 \x01(\x0b\x32\x1b.modal.client.GenericResult\"7\n\x13TaskExecPollRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x65xec_id\x18\x02 \x01(\t\"G\n\x14TaskExecPollResponse\x12\x0e\n\x04\x63ode\x18\x01 \x01(\x05H\x00\x12\x10\n\x06signal\x18\x02 \x01(\x05H\x00\x42\r\n\x0b\x65xit_status\"\x9b\x04\n\x14TaskExecStartRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x65xec_id\x18\x02 \x01(\t\x12\x14\n\x0c\x63ommand_args\x18\x03 \x03(\t\x12\x46\n\rstdout_config\x18\x04 \x01(\x0e\x32/.modal.task_command_router.TaskExecStdoutConfig\x12\x46\n\rstderr_config\x18\x05 \x01(\x0e\x32/.modal.task_command_router.TaskExecStderrConfig\x12\x19\n\x0ctimeout_secs\x18\x06 \x01(\rH\x00\x88\x01\x01\x12\x14\n\x07workdir\x18\x07 \x01(\tH\x01\x88\x01\x01\x12\x12\n\nsecret_ids\x18\x08 \x03(\t\x12,\n\x08pty_info\x18\t \x01(\x0b\x32\x15.modal.client.PTYInfoH\x02\x88\x01\x01\x12\x15\n\rruntime_debug\x18\n \x01(\x08\x12\x14\n\x0c\x63ontainer_id\x18\x0b \x01(\t\x12\x45\n\x03\x65nv\x18\x0c \x03(\x0b\x32\x38.modal.task_command_router.TaskExecStartRequest.EnvEntry\x1a*\n\x08\x45nvEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x0f\n\r_timeout_secsB\n\n\x08_workdirB\x0b\n\t_pty_info\"\x17\n\x15TaskExecStartResponse\"h\n\x19TaskExecStdinWriteRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x65xec_id\x18\x02 \x01(\t\x12\x0e\n\x06offset\x18\x03 \x01(\x04\x12\x0c\n\x04\x64\x61ta\x18\x04 \x01(\x0c\x12\x0b\n\x03\x65of\x18\x05 \x01(\x08\"\x1c\n\x1aTaskExecStdinWriteResponse\"\x9d\x01\n\x18TaskExecStdioReadRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x65xec_id\x18\x02 \x01(\t\x12\x0e\n\x06offset\x18\x03 \x01(\x04\x12O\n\x0f\x66ile_descriptor\x18\x04 \x01(\x0e\x32\x36.modal.task_command_router.TaskExecStdioFileDescriptor\")\n\x19TaskExecStdioReadResponse\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\"7\n\x13TaskExecWaitRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x65xec_id\x18\x02 \x01(\t\"G\n\x14TaskExecWaitResponse\x12\x0e\n\x04\x63ode\x18\x01 \x01(\x05H\x00\x12\x10\n\x06signal\x18\x02 \x01(\x05H\x00\x42\r\n\x0b\x65xit_status\"L\n\x19TaskMountDirectoryRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\x0c\x12\x10\n\x08image_id\x18\x03 \x01(\t\"R\n\x1cTaskSnapshotDirectoryRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\x0c\x12\x13\n\x0bsnapshot_id\x18\x03 \x01(\t\"1\n\x1dTaskSnapshotDirectoryResponse\x12\x10\n\x08image_id\x18\x01 \x01(\t\"<\n\x1bTaskUnmountDirectoryRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\x0c*\x81\x01\n\x14TaskExecStderrConfig\x12#\n\x1fTASK_EXEC_STDERR_CONFIG_DEVNULL\x10\x00\x12 \n\x1cTASK_EXEC_STDERR_CONFIG_PIPE\x10\x01\x12\"\n\x1eTASK_EXEC_STDERR_CONFIG_STDOUT\x10\x02*u\n\x1bTaskExecStdioFileDescriptor\x12*\n&TASK_EXEC_STDIO_FILE_DESCRIPTOR_STDOUT\x10\x00\x12*\n&TASK_EXEC_STDIO_FILE_DESCRIPTOR_STDERR\x10\x01*]\n\x14TaskExecStdoutConfig\x12#\n\x1fTASK_EXEC_STDOUT_CONFIG_DEVNULL\x10\x00\x12 \n\x1cTASK_EXEC_STDOUT_CONFIG_PIPE\x10\x01\x32\xdd\x0c\n\x11TaskCommandRouter\x12\x84\x01\n\x13TaskContainerCreate\x12\x35.modal.task_command_router.TaskContainerCreateRequest\x1a\x36.modal.task_command_router.TaskContainerCreateResponse\x12{\n\x10TaskContainerGet\x12\x32.modal.task_command_router.TaskContainerGetRequest\x1a\x33.modal.task_command_router.TaskContainerGetResponse\x12~\n\x11TaskContainerList\x12\x33.modal.task_command_router.TaskContainerListRequest\x1a\x34.modal.task_command_router.TaskContainerListResponse\x12\x8d\x01\n\x16TaskContainerTerminate\x12\x38.modal.task_command_router.TaskContainerTerminateRequest\x1a\x39.modal.task_command_router.TaskContainerTerminateResponse\x12~\n\x11TaskContainerWait\x12\x33.modal.task_command_router.TaskContainerWaitRequest\x1a\x34.modal.task_command_router.TaskContainerWaitResponse\x12o\n\x0cTaskExecPoll\x12..modal.task_command_router.TaskExecPollRequest\x1a/.modal.task_command_router.TaskExecPollResponse\x12r\n\rTaskExecStart\x12/.modal.task_command_router.TaskExecStartRequest\x1a\x30.modal.task_command_router.TaskExecStartResponse\x12\x81\x01\n\x12TaskExecStdinWrite\x12\x34.modal.task_command_router.TaskExecStdinWriteRequest\x1a\x35.modal.task_command_router.TaskExecStdinWriteResponse\x12\x80\x01\n\x11TaskExecStdioRead\x12\x33.modal.task_command_router.TaskExecStdioReadRequest\x1a\x34.modal.task_command_router.TaskExecStdioReadResponse0\x01\x12o\n\x0cTaskExecWait\x12..modal.task_command_router.TaskExecWaitRequest\x1a/.modal.task_command_router.TaskExecWaitResponse\x12\x62\n\x12TaskMountDirectory\x12\x34.modal.task_command_router.TaskMountDirectoryRequest\x1a\x16.google.protobuf.Empty\x12\x8a\x01\n\x15TaskSnapshotDirectory\x12\x37.modal.task_command_router.TaskSnapshotDirectoryRequest\x1a\x38.modal.task_command_router.TaskSnapshotDirectoryResponse\x12\x66\n\x14TaskUnmountDirectory\x12\x36.modal.task_command_router.TaskUnmountDirectoryRequest\x1a\x16.google.protobuf.EmptyB&Z$github.com/modal-labs/modal/go/protob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%modal_proto/task_command_router.proto\x12\x19modal.task_command_router\x1a\x1bgoogle/protobuf/empty.proto\x1a\x15modal_proto/api.proto\"X\n\x1aSandboxStdinWriteV2Request\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0e\n\x06offset\x18\x02 \x01(\x04\x12\x0c\n\x04\x64\x61ta\x18\x03 \x01(\x0c\x12\x0b\n\x03\x65of\x18\x04 \x01(\x08\"\x1d\n\x1bSandboxStdinWriteV2Response\"\x8c\x01\n\x19SandboxStdioReadV2Request\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0e\n\x06offset\x18\x02 \x01(\x04\x12N\n\x0f\x66ile_descriptor\x18\x03 \x01(\x0e\x32\x35.modal.task_command_router.SandboxStdioFileDescriptor\"*\n\x1aSandboxStdioReadV2Response\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\"\x83\x02\n\x1aTaskContainerCreateRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x16\n\x0e\x63ontainer_name\x18\x02 \x01(\t\x12\x10\n\x08image_id\x18\x03 \x01(\t\x12\x0c\n\x04\x61rgs\x18\x05 \x03(\t\x12K\n\x03\x65nv\x18\x06 \x03(\x0b\x32>.modal.task_command_router.TaskContainerCreateRequest.EnvEntry\x12\x0f\n\x07workdir\x18\x07 \x01(\t\x12\x12\n\nsecret_ids\x18\x08 \x03(\t\x1a*\n\x08\x45nvEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"K\n\x1bTaskContainerCreateResponse\x12\x14\n\x0c\x63ontainer_id\x18\x01 \x01(\t\x12\x16\n\x0e\x63ontainer_name\x18\x02 \x01(\t\"^\n\x17TaskContainerGetRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x16\n\x0e\x63ontainer_name\x18\x02 \x01(\t\x12\x1a\n\x12include_terminated\x18\x03 \x01(\x08\"[\n\x18TaskContainerGetResponse\x12?\n\tcontainer\x18\x01 \x01(\x0b\x32,.modal.task_command_router.TaskContainerInfo\"~\n\x11TaskContainerInfo\x12\x14\n\x0c\x63ontainer_id\x18\x01 \x01(\t\x12\x16\n\x0e\x63ontainer_name\x18\x02 \x01(\t\x12\x0e\n\x06status\x18\x03 \x01(\t\x12+\n\x06result\x18\x04 \x01(\x0b\x32\x1b.modal.client.GenericResult\"G\n\x18TaskContainerListRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x1a\n\x12include_terminated\x18\x02 \x01(\x08\"]\n\x19TaskContainerListResponse\x12@\n\ncontainers\x18\x01 \x03(\x0b\x32,.modal.task_command_router.TaskContainerInfo\"F\n\x1dTaskContainerTerminateRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x14\n\x0c\x63ontainer_id\x18\x02 \x01(\t\" \n\x1eTaskContainerTerminateResponse\"R\n\x18TaskContainerWaitRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x14\n\x0c\x63ontainer_id\x18\x02 \x01(\t\x12\x0f\n\x07timeout\x18\x03 \x01(\x02\"H\n\x19TaskContainerWaitResponse\x12+\n\x06result\x18\x01 \x01(\x0b\x32\x1b.modal.client.GenericResult\"7\n\x13TaskExecPollRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x65xec_id\x18\x02 \x01(\t\"G\n\x14TaskExecPollResponse\x12\x0e\n\x04\x63ode\x18\x01 \x01(\x05H\x00\x12\x10\n\x06signal\x18\x02 \x01(\x05H\x00\x42\r\n\x0b\x65xit_status\"\x9b\x04\n\x14TaskExecStartRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x65xec_id\x18\x02 \x01(\t\x12\x14\n\x0c\x63ommand_args\x18\x03 \x03(\t\x12\x46\n\rstdout_config\x18\x04 \x01(\x0e\x32/.modal.task_command_router.TaskExecStdoutConfig\x12\x46\n\rstderr_config\x18\x05 \x01(\x0e\x32/.modal.task_command_router.TaskExecStderrConfig\x12\x19\n\x0ctimeout_secs\x18\x06 \x01(\rH\x00\x88\x01\x01\x12\x14\n\x07workdir\x18\x07 \x01(\tH\x01\x88\x01\x01\x12\x12\n\nsecret_ids\x18\x08 \x03(\t\x12,\n\x08pty_info\x18\t \x01(\x0b\x32\x15.modal.client.PTYInfoH\x02\x88\x01\x01\x12\x15\n\rruntime_debug\x18\n \x01(\x08\x12\x14\n\x0c\x63ontainer_id\x18\x0b \x01(\t\x12\x45\n\x03\x65nv\x18\x0c \x03(\x0b\x32\x38.modal.task_command_router.TaskExecStartRequest.EnvEntry\x1a*\n\x08\x45nvEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x0f\n\r_timeout_secsB\n\n\x08_workdirB\x0b\n\t_pty_info\"\x17\n\x15TaskExecStartResponse\"h\n\x19TaskExecStdinWriteRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x65xec_id\x18\x02 \x01(\t\x12\x0e\n\x06offset\x18\x03 \x01(\x04\x12\x0c\n\x04\x64\x61ta\x18\x04 \x01(\x0c\x12\x0b\n\x03\x65of\x18\x05 \x01(\x08\"\x1c\n\x1aTaskExecStdinWriteResponse\"\x9d\x01\n\x18TaskExecStdioReadRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x65xec_id\x18\x02 \x01(\t\x12\x0e\n\x06offset\x18\x03 \x01(\x04\x12O\n\x0f\x66ile_descriptor\x18\x04 \x01(\x0e\x32\x36.modal.task_command_router.TaskExecStdioFileDescriptor\")\n\x19TaskExecStdioReadResponse\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\"7\n\x13TaskExecWaitRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0f\n\x07\x65xec_id\x18\x02 \x01(\t\"G\n\x14TaskExecWaitResponse\x12\x0e\n\x04\x63ode\x18\x01 \x01(\x05H\x00\x12\x10\n\x06signal\x18\x02 \x01(\x05H\x00\x42\r\n\x0b\x65xit_status\"L\n\x19TaskMountDirectoryRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\x0c\x12\x10\n\x08image_id\x18\x03 \x01(\t\"R\n\x1cTaskSnapshotDirectoryRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\x0c\x12\x13\n\x0bsnapshot_id\x18\x03 \x01(\t\"1\n\x1dTaskSnapshotDirectoryResponse\x12\x10\n\x08image_id\x18\x01 \x01(\t\"<\n\x1bTaskUnmountDirectoryRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\x0c*p\n\x1aSandboxStdioFileDescriptor\x12(\n$SANDBOX_STDIO_FILE_DESCRIPTOR_STDOUT\x10\x00\x12(\n$SANDBOX_STDIO_FILE_DESCRIPTOR_STDERR\x10\x01*\x81\x01\n\x14TaskExecStderrConfig\x12#\n\x1fTASK_EXEC_STDERR_CONFIG_DEVNULL\x10\x00\x12 \n\x1cTASK_EXEC_STDERR_CONFIG_PIPE\x10\x01\x12\"\n\x1eTASK_EXEC_STDERR_CONFIG_STDOUT\x10\x02*u\n\x1bTaskExecStdioFileDescriptor\x12*\n&TASK_EXEC_STDIO_FILE_DESCRIPTOR_STDOUT\x10\x00\x12*\n&TASK_EXEC_STDIO_FILE_DESCRIPTOR_STDERR\x10\x01*]\n\x14TaskExecStdoutConfig\x12#\n\x1fTASK_EXEC_STDOUT_CONFIG_DEVNULL\x10\x00\x12 \n\x1cTASK_EXEC_STDOUT_CONFIG_PIPE\x10\x01\x32\xea\x0e\n\x11TaskCommandRouter\x12\x84\x01\n\x13SandboxStdinWriteV2\x12\x35.modal.task_command_router.SandboxStdinWriteV2Request\x1a\x36.modal.task_command_router.SandboxStdinWriteV2Response\x12\x83\x01\n\x12SandboxStdioReadV2\x12\x34.modal.task_command_router.SandboxStdioReadV2Request\x1a\x35.modal.task_command_router.SandboxStdioReadV2Response0\x01\x12\x84\x01\n\x13TaskContainerCreate\x12\x35.modal.task_command_router.TaskContainerCreateRequest\x1a\x36.modal.task_command_router.TaskContainerCreateResponse\x12{\n\x10TaskContainerGet\x12\x32.modal.task_command_router.TaskContainerGetRequest\x1a\x33.modal.task_command_router.TaskContainerGetResponse\x12~\n\x11TaskContainerList\x12\x33.modal.task_command_router.TaskContainerListRequest\x1a\x34.modal.task_command_router.TaskContainerListResponse\x12\x8d\x01\n\x16TaskContainerTerminate\x12\x38.modal.task_command_router.TaskContainerTerminateRequest\x1a\x39.modal.task_command_router.TaskContainerTerminateResponse\x12~\n\x11TaskContainerWait\x12\x33.modal.task_command_router.TaskContainerWaitRequest\x1a\x34.modal.task_command_router.TaskContainerWaitResponse\x12o\n\x0cTaskExecPoll\x12..modal.task_command_router.TaskExecPollRequest\x1a/.modal.task_command_router.TaskExecPollResponse\x12r\n\rTaskExecStart\x12/.modal.task_command_router.TaskExecStartRequest\x1a\x30.modal.task_command_router.TaskExecStartResponse\x12\x81\x01\n\x12TaskExecStdinWrite\x12\x34.modal.task_command_router.TaskExecStdinWriteRequest\x1a\x35.modal.task_command_router.TaskExecStdinWriteResponse\x12\x80\x01\n\x11TaskExecStdioRead\x12\x33.modal.task_command_router.TaskExecStdioReadRequest\x1a\x34.modal.task_command_router.TaskExecStdioReadResponse0\x01\x12o\n\x0cTaskExecWait\x12..modal.task_command_router.TaskExecWaitRequest\x1a/.modal.task_command_router.TaskExecWaitResponse\x12\x62\n\x12TaskMountDirectory\x12\x34.modal.task_command_router.TaskMountDirectoryRequest\x1a\x16.google.protobuf.Empty\x12\x8a\x01\n\x15TaskSnapshotDirectory\x12\x37.modal.task_command_router.TaskSnapshotDirectoryRequest\x1a\x38.modal.task_command_router.TaskSnapshotDirectoryResponse\x12\x66\n\x14TaskUnmountDirectory\x12\x36.modal.task_command_router.TaskUnmountDirectoryRequest\x1a\x16.google.protobuf.EmptyB&Z$github.com/modal-labs/modal/go/protob\x06proto3')
 
+_SANDBOXSTDIOFILEDESCRIPTOR = DESCRIPTOR.enum_types_by_name['SandboxStdioFileDescriptor']
+SandboxStdioFileDescriptor = enum_type_wrapper.EnumTypeWrapper(_SANDBOXSTDIOFILEDESCRIPTOR)
 _TASKEXECSTDERRCONFIG = DESCRIPTOR.enum_types_by_name['TaskExecStderrConfig']
 TaskExecStderrConfig = enum_type_wrapper.EnumTypeWrapper(_TASKEXECSTDERRCONFIG)
 _TASKEXECSTDIOFILEDESCRIPTOR = DESCRIPTOR.enum_types_by_name['TaskExecStdioFileDescriptor']
 TaskExecStdioFileDescriptor = enum_type_wrapper.EnumTypeWrapper(_TASKEXECSTDIOFILEDESCRIPTOR)
 _TASKEXECSTDOUTCONFIG = DESCRIPTOR.enum_types_by_name['TaskExecStdoutConfig']
 TaskExecStdoutConfig = enum_type_wrapper.EnumTypeWrapper(_TASKEXECSTDOUTCONFIG)
+SANDBOX_STDIO_FILE_DESCRIPTOR_STDOUT = 0
+SANDBOX_STDIO_FILE_DESCRIPTOR_STDERR = 1
 TASK_EXEC_STDERR_CONFIG_DEVNULL = 0
 TASK_EXEC_STDERR_CONFIG_PIPE = 1
 TASK_EXEC_STDERR_CONFIG_STDOUT = 2
@@ -34,6 +38,10 @@ TASK_EXEC_STDOUT_CONFIG_DEVNULL = 0
 TASK_EXEC_STDOUT_CONFIG_PIPE = 1
 
 
+_SANDBOXSTDINWRITEV2REQUEST = DESCRIPTOR.message_types_by_name['SandboxStdinWriteV2Request']
+_SANDBOXSTDINWRITEV2RESPONSE = DESCRIPTOR.message_types_by_name['SandboxStdinWriteV2Response']
+_SANDBOXSTDIOREADV2REQUEST = DESCRIPTOR.message_types_by_name['SandboxStdioReadV2Request']
+_SANDBOXSTDIOREADV2RESPONSE = DESCRIPTOR.message_types_by_name['SandboxStdioReadV2Response']
 _TASKCONTAINERCREATEREQUEST = DESCRIPTOR.message_types_by_name['TaskContainerCreateRequest']
 _TASKCONTAINERCREATEREQUEST_ENVENTRY = _TASKCONTAINERCREATEREQUEST.nested_types_by_name['EnvEntry']
 _TASKCONTAINERCREATERESPONSE = DESCRIPTOR.message_types_by_name['TaskContainerCreateResponse']
@@ -61,6 +69,34 @@ _TASKMOUNTDIRECTORYREQUEST = DESCRIPTOR.message_types_by_name['TaskMountDirector
 _TASKSNAPSHOTDIRECTORYREQUEST = DESCRIPTOR.message_types_by_name['TaskSnapshotDirectoryRequest']
 _TASKSNAPSHOTDIRECTORYRESPONSE = DESCRIPTOR.message_types_by_name['TaskSnapshotDirectoryResponse']
 _TASKUNMOUNTDIRECTORYREQUEST = DESCRIPTOR.message_types_by_name['TaskUnmountDirectoryRequest']
+SandboxStdinWriteV2Request = _reflection.GeneratedProtocolMessageType('SandboxStdinWriteV2Request', (_message.Message,), {
+  'DESCRIPTOR' : _SANDBOXSTDINWRITEV2REQUEST,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.SandboxStdinWriteV2Request)
+  })
+_sym_db.RegisterMessage(SandboxStdinWriteV2Request)
+
+SandboxStdinWriteV2Response = _reflection.GeneratedProtocolMessageType('SandboxStdinWriteV2Response', (_message.Message,), {
+  'DESCRIPTOR' : _SANDBOXSTDINWRITEV2RESPONSE,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.SandboxStdinWriteV2Response)
+  })
+_sym_db.RegisterMessage(SandboxStdinWriteV2Response)
+
+SandboxStdioReadV2Request = _reflection.GeneratedProtocolMessageType('SandboxStdioReadV2Request', (_message.Message,), {
+  'DESCRIPTOR' : _SANDBOXSTDIOREADV2REQUEST,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.SandboxStdioReadV2Request)
+  })
+_sym_db.RegisterMessage(SandboxStdioReadV2Request)
+
+SandboxStdioReadV2Response = _reflection.GeneratedProtocolMessageType('SandboxStdioReadV2Response', (_message.Message,), {
+  'DESCRIPTOR' : _SANDBOXSTDIOREADV2RESPONSE,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.SandboxStdioReadV2Response)
+  })
+_sym_db.RegisterMessage(SandboxStdioReadV2Response)
+
 TaskContainerCreateRequest = _reflection.GeneratedProtocolMessageType('TaskContainerCreateRequest', (_message.Message,), {
 
   'EnvEntry' : _reflection.GeneratedProtocolMessageType('EnvEntry', (_message.Message,), {
@@ -261,66 +297,76 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _TASKCONTAINERCREATEREQUEST_ENVENTRY._serialized_options = b'8\001'
   _TASKEXECSTARTREQUEST_ENVENTRY._options = None
   _TASKEXECSTARTREQUEST_ENVENTRY._serialized_options = b'8\001'
-  _TASKEXECSTDERRCONFIG._serialized_start=2650
-  _TASKEXECSTDERRCONFIG._serialized_end=2779
-  _TASKEXECSTDIOFILEDESCRIPTOR._serialized_start=2781
-  _TASKEXECSTDIOFILEDESCRIPTOR._serialized_end=2898
-  _TASKEXECSTDOUTCONFIG._serialized_start=2900
-  _TASKEXECSTDOUTCONFIG._serialized_end=2993
-  _TASKCONTAINERCREATEREQUEST._serialized_start=121
-  _TASKCONTAINERCREATEREQUEST._serialized_end=380
-  _TASKCONTAINERCREATEREQUEST_ENVENTRY._serialized_start=338
-  _TASKCONTAINERCREATEREQUEST_ENVENTRY._serialized_end=380
-  _TASKCONTAINERCREATERESPONSE._serialized_start=382
-  _TASKCONTAINERCREATERESPONSE._serialized_end=457
-  _TASKCONTAINERGETREQUEST._serialized_start=459
-  _TASKCONTAINERGETREQUEST._serialized_end=553
-  _TASKCONTAINERGETRESPONSE._serialized_start=555
-  _TASKCONTAINERGETRESPONSE._serialized_end=646
-  _TASKCONTAINERINFO._serialized_start=648
-  _TASKCONTAINERINFO._serialized_end=774
-  _TASKCONTAINERLISTREQUEST._serialized_start=776
-  _TASKCONTAINERLISTREQUEST._serialized_end=847
-  _TASKCONTAINERLISTRESPONSE._serialized_start=849
-  _TASKCONTAINERLISTRESPONSE._serialized_end=942
-  _TASKCONTAINERTERMINATEREQUEST._serialized_start=944
-  _TASKCONTAINERTERMINATEREQUEST._serialized_end=1014
-  _TASKCONTAINERTERMINATERESPONSE._serialized_start=1016
-  _TASKCONTAINERTERMINATERESPONSE._serialized_end=1048
-  _TASKCONTAINERWAITREQUEST._serialized_start=1050
-  _TASKCONTAINERWAITREQUEST._serialized_end=1132
-  _TASKCONTAINERWAITRESPONSE._serialized_start=1134
-  _TASKCONTAINERWAITRESPONSE._serialized_end=1206
-  _TASKEXECPOLLREQUEST._serialized_start=1208
-  _TASKEXECPOLLREQUEST._serialized_end=1263
-  _TASKEXECPOLLRESPONSE._serialized_start=1265
-  _TASKEXECPOLLRESPONSE._serialized_end=1336
-  _TASKEXECSTARTREQUEST._serialized_start=1339
-  _TASKEXECSTARTREQUEST._serialized_end=1878
-  _TASKEXECSTARTREQUEST_ENVENTRY._serialized_start=338
-  _TASKEXECSTARTREQUEST_ENVENTRY._serialized_end=380
-  _TASKEXECSTARTRESPONSE._serialized_start=1880
-  _TASKEXECSTARTRESPONSE._serialized_end=1903
-  _TASKEXECSTDINWRITEREQUEST._serialized_start=1905
-  _TASKEXECSTDINWRITEREQUEST._serialized_end=2009
-  _TASKEXECSTDINWRITERESPONSE._serialized_start=2011
-  _TASKEXECSTDINWRITERESPONSE._serialized_end=2039
-  _TASKEXECSTDIOREADREQUEST._serialized_start=2042
-  _TASKEXECSTDIOREADREQUEST._serialized_end=2199
-  _TASKEXECSTDIOREADRESPONSE._serialized_start=2201
-  _TASKEXECSTDIOREADRESPONSE._serialized_end=2242
-  _TASKEXECWAITREQUEST._serialized_start=2244
-  _TASKEXECWAITREQUEST._serialized_end=2299
-  _TASKEXECWAITRESPONSE._serialized_start=2301
-  _TASKEXECWAITRESPONSE._serialized_end=2372
-  _TASKMOUNTDIRECTORYREQUEST._serialized_start=2374
-  _TASKMOUNTDIRECTORYREQUEST._serialized_end=2450
-  _TASKSNAPSHOTDIRECTORYREQUEST._serialized_start=2452
-  _TASKSNAPSHOTDIRECTORYREQUEST._serialized_end=2534
-  _TASKSNAPSHOTDIRECTORYRESPONSE._serialized_start=2536
-  _TASKSNAPSHOTDIRECTORYRESPONSE._serialized_end=2585
-  _TASKUNMOUNTDIRECTORYREQUEST._serialized_start=2587
-  _TASKUNMOUNTDIRECTORYREQUEST._serialized_end=2647
-  _TASKCOMMANDROUTER._serialized_start=2996
-  _TASKCOMMANDROUTER._serialized_end=4625
+  _SANDBOXSTDIOFILEDESCRIPTOR._serialized_start=2957
+  _SANDBOXSTDIOFILEDESCRIPTOR._serialized_end=3069
+  _TASKEXECSTDERRCONFIG._serialized_start=3072
+  _TASKEXECSTDERRCONFIG._serialized_end=3201
+  _TASKEXECSTDIOFILEDESCRIPTOR._serialized_start=3203
+  _TASKEXECSTDIOFILEDESCRIPTOR._serialized_end=3320
+  _TASKEXECSTDOUTCONFIG._serialized_start=3322
+  _TASKEXECSTDOUTCONFIG._serialized_end=3415
+  _SANDBOXSTDINWRITEV2REQUEST._serialized_start=120
+  _SANDBOXSTDINWRITEV2REQUEST._serialized_end=208
+  _SANDBOXSTDINWRITEV2RESPONSE._serialized_start=210
+  _SANDBOXSTDINWRITEV2RESPONSE._serialized_end=239
+  _SANDBOXSTDIOREADV2REQUEST._serialized_start=242
+  _SANDBOXSTDIOREADV2REQUEST._serialized_end=382
+  _SANDBOXSTDIOREADV2RESPONSE._serialized_start=384
+  _SANDBOXSTDIOREADV2RESPONSE._serialized_end=426
+  _TASKCONTAINERCREATEREQUEST._serialized_start=429
+  _TASKCONTAINERCREATEREQUEST._serialized_end=688
+  _TASKCONTAINERCREATEREQUEST_ENVENTRY._serialized_start=646
+  _TASKCONTAINERCREATEREQUEST_ENVENTRY._serialized_end=688
+  _TASKCONTAINERCREATERESPONSE._serialized_start=690
+  _TASKCONTAINERCREATERESPONSE._serialized_end=765
+  _TASKCONTAINERGETREQUEST._serialized_start=767
+  _TASKCONTAINERGETREQUEST._serialized_end=861
+  _TASKCONTAINERGETRESPONSE._serialized_start=863
+  _TASKCONTAINERGETRESPONSE._serialized_end=954
+  _TASKCONTAINERINFO._serialized_start=956
+  _TASKCONTAINERINFO._serialized_end=1082
+  _TASKCONTAINERLISTREQUEST._serialized_start=1084
+  _TASKCONTAINERLISTREQUEST._serialized_end=1155
+  _TASKCONTAINERLISTRESPONSE._serialized_start=1157
+  _TASKCONTAINERLISTRESPONSE._serialized_end=1250
+  _TASKCONTAINERTERMINATEREQUEST._serialized_start=1252
+  _TASKCONTAINERTERMINATEREQUEST._serialized_end=1322
+  _TASKCONTAINERTERMINATERESPONSE._serialized_start=1324
+  _TASKCONTAINERTERMINATERESPONSE._serialized_end=1356
+  _TASKCONTAINERWAITREQUEST._serialized_start=1358
+  _TASKCONTAINERWAITREQUEST._serialized_end=1440
+  _TASKCONTAINERWAITRESPONSE._serialized_start=1442
+  _TASKCONTAINERWAITRESPONSE._serialized_end=1514
+  _TASKEXECPOLLREQUEST._serialized_start=1516
+  _TASKEXECPOLLREQUEST._serialized_end=1571
+  _TASKEXECPOLLRESPONSE._serialized_start=1573
+  _TASKEXECPOLLRESPONSE._serialized_end=1644
+  _TASKEXECSTARTREQUEST._serialized_start=1647
+  _TASKEXECSTARTREQUEST._serialized_end=2186
+  _TASKEXECSTARTREQUEST_ENVENTRY._serialized_start=646
+  _TASKEXECSTARTREQUEST_ENVENTRY._serialized_end=688
+  _TASKEXECSTARTRESPONSE._serialized_start=2188
+  _TASKEXECSTARTRESPONSE._serialized_end=2211
+  _TASKEXECSTDINWRITEREQUEST._serialized_start=2213
+  _TASKEXECSTDINWRITEREQUEST._serialized_end=2317
+  _TASKEXECSTDINWRITERESPONSE._serialized_start=2319
+  _TASKEXECSTDINWRITERESPONSE._serialized_end=2347
+  _TASKEXECSTDIOREADREQUEST._serialized_start=2350
+  _TASKEXECSTDIOREADREQUEST._serialized_end=2507
+  _TASKEXECSTDIOREADRESPONSE._serialized_start=2509
+  _TASKEXECSTDIOREADRESPONSE._serialized_end=2550
+  _TASKEXECWAITREQUEST._serialized_start=2552
+  _TASKEXECWAITREQUEST._serialized_end=2607
+  _TASKEXECWAITRESPONSE._serialized_start=2609
+  _TASKEXECWAITRESPONSE._serialized_end=2680
+  _TASKMOUNTDIRECTORYREQUEST._serialized_start=2682
+  _TASKMOUNTDIRECTORYREQUEST._serialized_end=2758
+  _TASKSNAPSHOTDIRECTORYREQUEST._serialized_start=2760
+  _TASKSNAPSHOTDIRECTORYREQUEST._serialized_end=2842
+  _TASKSNAPSHOTDIRECTORYRESPONSE._serialized_start=2844
+  _TASKSNAPSHOTDIRECTORYRESPONSE._serialized_end=2893
+  _TASKUNMOUNTDIRECTORYREQUEST._serialized_start=2895
+  _TASKUNMOUNTDIRECTORYREQUEST._serialized_end=2955
+  _TASKCOMMANDROUTER._serialized_start=3418
+  _TASKCOMMANDROUTER._serialized_end=5316
 # @@protoc_insertion_point(module_scope)

modal_proto/task_command_router_pb2.pyi

@@ -19,6 +19,21 @@ else:
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+class _SandboxStdioFileDescriptor:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _SandboxStdioFileDescriptorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SandboxStdioFileDescriptor.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    SANDBOX_STDIO_FILE_DESCRIPTOR_STDOUT: _SandboxStdioFileDescriptor.ValueType  # 0
+    SANDBOX_STDIO_FILE_DESCRIPTOR_STDERR: _SandboxStdioFileDescriptor.ValueType  # 1
+
+class SandboxStdioFileDescriptor(_SandboxStdioFileDescriptor, metaclass=_SandboxStdioFileDescriptorEnumTypeWrapper): ...
+
+SANDBOX_STDIO_FILE_DESCRIPTOR_STDOUT: SandboxStdioFileDescriptor.ValueType  # 0
+SANDBOX_STDIO_FILE_DESCRIPTOR_STDERR: SandboxStdioFileDescriptor.ValueType  # 1
+global___SandboxStdioFileDescriptor = SandboxStdioFileDescriptor
+
 class _TaskExecStderrConfig:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
@@ -84,6 +99,72 @@ TASK_EXEC_STDOUT_CONFIG_PIPE: TaskExecStdoutConfig.ValueType  # 1
 """The output will be streamed to the client."""
 global___TaskExecStdoutConfig = TaskExecStdoutConfig
 
+class SandboxStdinWriteV2Request(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    OFFSET_FIELD_NUMBER: builtins.int
+    DATA_FIELD_NUMBER: builtins.int
+    EOF_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    offset: builtins.int
+    data: builtins.bytes
+    eof: builtins.bool
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+        offset: builtins.int = ...,
+        data: builtins.bytes = ...,
+        eof: builtins.bool = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data", b"data", "eof", b"eof", "offset", b"offset", "task_id", b"task_id"]) -> None: ...
+
+global___SandboxStdinWriteV2Request = SandboxStdinWriteV2Request
+
+class SandboxStdinWriteV2Response(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___SandboxStdinWriteV2Response = SandboxStdinWriteV2Response
+
+class SandboxStdioReadV2Request(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    OFFSET_FIELD_NUMBER: builtins.int
+    FILE_DESCRIPTOR_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    offset: builtins.int
+    file_descriptor: global___SandboxStdioFileDescriptor.ValueType
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+        offset: builtins.int = ...,
+        file_descriptor: global___SandboxStdioFileDescriptor.ValueType = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["file_descriptor", b"file_descriptor", "offset", b"offset", "task_id", b"task_id"]) -> None: ...
+
+global___SandboxStdioReadV2Request = SandboxStdioReadV2Request
+
+class SandboxStdioReadV2Response(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATA_FIELD_NUMBER: builtins.int
+    data: builtins.bytes
+    def __init__(
+        self,
+        *,
+        data: builtins.bytes = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data", b"data"]) -> None: ...
+
+global___SandboxStdioReadV2Response = SandboxStdioReadV2Response
+
 class TaskContainerCreateRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

modal_proto/task_command_router_pb2_grpc.py

@@ -15,6 +15,16 @@ class TaskCommandRouterStub(object):
         Args:
             channel: A grpc.Channel.
         """
+        self.SandboxStdinWriteV2 = channel.unary_unary(
+                '/modal.task_command_router.TaskCommandRouter/SandboxStdinWriteV2',
+                request_serializer=modal__proto_dot_task__command__router__pb2.SandboxStdinWriteV2Request.SerializeToString,
+                response_deserializer=modal__proto_dot_task__command__router__pb2.SandboxStdinWriteV2Response.FromString,
+                )
+        self.SandboxStdioReadV2 = channel.unary_stream(
+                '/modal.task_command_router.TaskCommandRouter/SandboxStdioReadV2',
+                request_serializer=modal__proto_dot_task__command__router__pb2.SandboxStdioReadV2Request.SerializeToString,
+                response_deserializer=modal__proto_dot_task__command__router__pb2.SandboxStdioReadV2Response.FromString,
+                )
         self.TaskContainerCreate = channel.unary_unary(
                 '/modal.task_command_router.TaskCommandRouter/TaskContainerCreate',
                 request_serializer=modal__proto_dot_task__command__router__pb2.TaskContainerCreateRequest.SerializeToString,
@@ -85,6 +95,18 @@ class TaskCommandRouterStub(object):
 class TaskCommandRouterServicer(object):
     """Missing associated documentation comment in .proto file."""
 
+    def SandboxStdinWriteV2(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def SandboxStdioReadV2(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def TaskContainerCreate(self, request, context):
         """Create an additional container for a task.
         """
@@ -179,6 +201,16 @@ class TaskCommandRouterServicer(object):
 
 def add_TaskCommandRouterServicer_to_server(servicer, server):
     rpc_method_handlers = {
+            'SandboxStdinWriteV2': grpc.unary_unary_rpc_method_handler(
+                    servicer.SandboxStdinWriteV2,
+                    request_deserializer=modal__proto_dot_task__command__router__pb2.SandboxStdinWriteV2Request.FromString,
+                    response_serializer=modal__proto_dot_task__command__router__pb2.SandboxStdinWriteV2Response.SerializeToString,
+            ),
+            'SandboxStdioReadV2': grpc.unary_stream_rpc_method_handler(
+                    servicer.SandboxStdioReadV2,
+                    request_deserializer=modal__proto_dot_task__command__router__pb2.SandboxStdioReadV2Request.FromString,
+                    response_serializer=modal__proto_dot_task__command__router__pb2.SandboxStdioReadV2Response.SerializeToString,
+            ),
             'TaskContainerCreate': grpc.unary_unary_rpc_method_handler(
                     servicer.TaskContainerCreate,
                     request_deserializer=modal__proto_dot_task__command__router__pb2.TaskContainerCreateRequest.FromString,
@@ -254,6 +286,40 @@ def add_TaskCommandRouterServicer_to_server(servicer, server):
 class TaskCommandRouter(object):
     """Missing associated documentation comment in .proto file."""
 
+    @staticmethod
+    def SandboxStdinWriteV2(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/modal.task_command_router.TaskCommandRouter/SandboxStdinWriteV2',
+            modal__proto_dot_task__command__router__pb2.SandboxStdinWriteV2Request.SerializeToString,
+            modal__proto_dot_task__command__router__pb2.SandboxStdinWriteV2Response.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def SandboxStdioReadV2(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_stream(request, target, '/modal.task_command_router.TaskCommandRouter/SandboxStdioReadV2',
+            modal__proto_dot_task__command__router__pb2.SandboxStdioReadV2Request.SerializeToString,
+            modal__proto_dot_task__command__router__pb2.SandboxStdioReadV2Response.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def TaskContainerCreate(request,
             target,

modal_proto/task_command_router_pb2_grpc.pyi

@@ -10,6 +10,14 @@ import modal_proto.task_command_router_pb2
 
 class TaskCommandRouterStub:
     def __init__(self, channel: grpc.Channel) -> None: ...
+    SandboxStdinWriteV2: grpc.UnaryUnaryMultiCallable[
+        modal_proto.task_command_router_pb2.SandboxStdinWriteV2Request,
+        modal_proto.task_command_router_pb2.SandboxStdinWriteV2Response,
+    ]
+    SandboxStdioReadV2: grpc.UnaryStreamMultiCallable[
+        modal_proto.task_command_router_pb2.SandboxStdioReadV2Request,
+        modal_proto.task_command_router_pb2.SandboxStdioReadV2Response,
+    ]
     TaskContainerCreate: grpc.UnaryUnaryMultiCallable[
         modal_proto.task_command_router_pb2.TaskContainerCreateRequest,
         modal_proto.task_command_router_pb2.TaskContainerCreateResponse,
@@ -77,6 +85,18 @@ class TaskCommandRouterStub:
     """Unmount an image previously mounted at a directory in the container."""
 
 class TaskCommandRouterServicer(metaclass=abc.ABCMeta):
+    @abc.abstractmethod
+    def SandboxStdinWriteV2(
+        self,
+        request: modal_proto.task_command_router_pb2.SandboxStdinWriteV2Request,
+        context: grpc.ServicerContext,
+    ) -> modal_proto.task_command_router_pb2.SandboxStdinWriteV2Response: ...
+    @abc.abstractmethod
+    def SandboxStdioReadV2(
+        self,
+        request: modal_proto.task_command_router_pb2.SandboxStdioReadV2Request,
+        context: grpc.ServicerContext,
+    ) -> collections.abc.Iterator[modal_proto.task_command_router_pb2.SandboxStdioReadV2Response]: ...
     @abc.abstractmethod
     def TaskContainerCreate(
         self,

modal_version/__init__.py

@@ -1,4 +1,4 @@
 # Copyright Modal Labs 2026
 """Supplies the current version of the modal client library."""
 
-__version__ = "1.4.3.dev7"
+__version__ = "1.4.3.dev8"