modAL 1.2.5.dev28__tar.gz → 1.2.6__tar.gz

{modal-1.2.5.dev28 → modal-1.2.6}/PKG-INFO

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

{modal-1.2.5.dev28 → modal-1.2.6}/modal/_output.py

@@ -556,7 +556,7 @@ async def get_app_logs_loop(
     async def stop_pty_shell():
         nonlocal pty_shell_finish_event, pty_shell_input_task
         if pty_shell_finish_event:
-            print("\r", end="")  # move cursor to beginning of line
+            print("\r", end="")  # move cursor to beginning of line # noqa: T201
             pty_shell_finish_event.set()
             pty_shell_finish_event = None
 
@@ -623,7 +623,7 @@ async def get_app_logs_loop(
                 # This corresponds to the `modal run -i` use case where a breakpoint
                 # triggers and the task drops into an interactive PTY mode
                 if pty_shell_finish_event:
-                    print("ERROR: concurrent PTY shells are not supported.")
+                    print("ERROR: concurrent PTY shells are not supported.")  # noqa: T201
                 else:
                     pty_shell_stdout = output_mgr._stdout
                     pty_shell_finish_event = asyncio.Event()

{modal-1.2.5.dev28 → modal-1.2.6}/modal/_runtime/container_io_manager.py

@@ -1007,7 +1007,7 @@ class _ContainerIOManager:
         # Start a debugger if the worker tells us to
         if int(restored_state.get("snapshot_debug", 0)):
             logger.debug("Entering snapshot debugger")
-            breakpoint()
+            breakpoint()  # noqa: T100
 
         # Local ContainerIOManager state.
         for key in ["task_id", "function_id"]:

{modal-1.2.5.dev28 → modal-1.2.6}/modal/_runtime/user_code_imports.py

@@ -243,7 +243,7 @@ def create_breakpoint_wrapper(container_io_manager: "modal._runtime.container_io
     def breakpoint_wrapper():
         # note: it would be nice to not have breakpoint_wrapper() included in the backtrace
         container_io_manager.interact(from_breakpoint=True)
-        import pdb
+        import pdb  # noqa: T100
 
         current_frame = inspect.currentframe()
         if current_frame is not None:

{modal-1.2.5.dev28 → modal-1.2.6}/modal/_traceback.py

@@ -119,7 +119,7 @@ def print_exception(exc: Optional[type[BaseException]], value: Optional[BaseExce
     traceback.print_exception(exc, value, tb)
     if sys.version_info < (3, 11) and value is not None:  # type: ignore
         notes = getattr(value, "__notes__", [])
-        print(*notes, sep="\n", file=sys.stderr)
+        print(*notes, sep="\n", file=sys.stderr)  # noqa: T201
 
 
 def print_server_warnings(server_warnings: Iterable[api_pb2.Warning]):

{modal-1.2.5.dev28 → modal-1.2.6}/modal/_utils/task_command_router_client.py

@@ -403,7 +403,10 @@ class TaskCommandRouterClient:
                 )
                 return
 
+            logger.debug(f"Refreshing JWT for exec with task ID {self._task_id}")
             resp = await fetch_command_router_access(self._server_client, self._task_id)
+            logger.debug(f"Finished refreshing JWT for exec with task ID {self._task_id}")
+
             # Ensure the server URL remains stable for the lifetime of this client.
             assert resp.url == self._server_url, "Task router URL changed during session"
             self._jwt = resp.jwt
@@ -434,7 +437,9 @@ class TaskCommandRouterClient:
         delay_secs = self.stream_stdio_retry_delay_secs
         delay_factor = self.stream_stdio_retry_delay_factor
         num_retries_remaining = self.stream_stdio_max_retries
-        num_auth_retries = 0
+        # Flag to prevent infinite auth retries in the event that the JWT
+        # refresh yields an invalid JWT somehow or that the JWT is otherwise invalid.
+        did_auth_retry = False
 
         async def sleep_and_update_delay_and_num_retries_remaining(e: Exception):
             nonlocal delay_secs, num_retries_remaining
@@ -458,25 +463,28 @@ class TaskCommandRouterClient:
                         file_descriptor=file_descriptor,
                     )
 
-                    # Scope auth retry strictly to the initial send (where headers/auth are sent).
+                    # Auth retry is scoped to a single refresh per streaming attempt. While auth metadata is
+                    # sent on request start, UNAUTHENTICATED may sometimes surface during iteration,
+                    # so we handle it at both send and receive boundaries.
                     try:
                         await s.send_message(req, end=True)
+                        async for item in s:
+                            # We successfully authenticated after a JWT refresh, reset the auth retry flag.
+                            if did_auth_retry:
+                                did_auth_retry = False
+                            # Reset retry backoff after any successful chunk.
+                            delay_secs = self.stream_stdio_retry_delay_secs
+                            offset += len(item.data)
+                            yield item
                     except GRPCError as exc:
-                        if exc.status == Status.UNAUTHENTICATED and num_auth_retries < 1:
+                        if exc.status == Status.UNAUTHENTICATED and not did_auth_retry:
                             await self._refresh_jwt()
-                            num_auth_retries += 1
+                            # Mark that we've retried authentication for this streaming attempt, to
+                            # prevent subsequent retries.
+                            did_auth_retry = True
                             continue
                         raise
 
-                    # We successfully authenticated, reset the auth retry count.
-                    num_auth_retries = 0
-
-                    async for item in s:
-                        # Reset retry backoff after any successful chunk.
-                        delay_secs = self.stream_stdio_retry_delay_secs
-                        offset += len(item.data)
-                        yield item
-
                 # We successfully streamed all output.
                 return
             except GRPCError as e:

{modal-1.2.5.dev28 → modal-1.2.6}/modal/cli/network_file_system.py

@@ -102,7 +102,7 @@ async def ls(
         console.print(table)
     else:
         for entry in entries:
-            print(entry.path)
+            print(entry.path)  # noqa: T201
 
 
 @nfs_cli.command(

{modal-1.2.5.dev28 → modal-1.2.6}/modal/cli/run.py

@@ -292,7 +292,7 @@ def _get_click_command_for_local_entrypoint(app: App, entrypoint: LocalEntrypoin
     @click.pass_context
     def f(ctx, *args, **kwargs):
         if ctx.obj["detach"]:
-            print(
+            print(  # noqa: T201
                 "Note that running a local entrypoint in detached mode only keeps the last "
                 "triggered Modal function alive after the parent process has been killed or disconnected."
             )

{modal-1.2.5.dev28 → modal-1.2.6}/modal/cli/volume.py

@@ -148,7 +148,7 @@ async def ls(
     if not json and not sys.stdout.isatty():
         # Legacy behavior -- I am not sure why exactly we did this originally but I don't want to break it
         for entry in entries:
-            print(entry.path)
+            print(entry.path)  # noqa: T201
     else:
         rows = []
         for entry in entries:

{modal-1.2.5.dev28 → modal-1.2.6}/modal/client.pyi

@@ -28,11 +28,7 @@ class _Client:
     _snapshotted: bool
 
     def __init__(
-        self,
-        server_url: str,
-        client_type: int,
-        credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.2.5.dev28",
+        self, server_url: str, client_type: int, credentials: typing.Optional[tuple[str, str]], version: str = "1.2.6"
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.
@@ -159,11 +155,7 @@ class Client:
     _snapshotted: bool
 
     def __init__(
-        self,
-        server_url: str,
-        client_type: int,
-        credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.2.5.dev28",
+        self, server_url: str, client_type: int, credentials: typing.Optional[tuple[str, str]], version: str = "1.2.6"
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.

{modal-1.2.5.dev28 → modal-1.2.6}/modal/container_process.py

@@ -146,7 +146,7 @@ class _ContainerProcessThroughServer(Generic[T]):
     async def attach(self):
         """mdmd:hidden"""
         if platform.system() == "Windows":
-            print("interactive exec is not currently supported on Windows.")
+            print("interactive exec is not currently supported on Windows.")  # noqa: T201
             return
 
         from ._output import make_console
@@ -347,7 +347,7 @@ class _ContainerProcessThroughCommandRouter(Generic[T]):
 
     async def attach(self):
         if platform.system() == "Windows":
-            print("interactive exec is not currently supported on Windows.")
+            print("interactive exec is not currently supported on Windows.")  # noqa: T201
             return
 
         from ._output import make_console

{modal-1.2.5.dev28 → modal-1.2.6}/modal/io_streams.py

@@ -183,7 +183,7 @@ class _StreamReaderThroughServer(Generic[T]):
                 async for message, batch_index in iterator:
                     if self._stream_type == StreamType.STDOUT and message:
                         # TODO: rearchitect this, since these bytes aren't necessarily decodable
-                        print(message.decode("utf-8"), end="")
+                        print(message.decode("utf-8"), end="")  # noqa: T201
                     elif self._stream_type == StreamType.PIPE:
                         self._container_process_buffer.append(message)
 

{modal-1.2.5.dev28 → modal-1.2.6}/modal/mount.py

@@ -775,13 +775,13 @@ async def _create_single_client_dependency_mount(
     if check_if_exists:
         try:
             await Mount.from_name(mount_name, namespace=api_pb2.DEPLOYMENT_NAMESPACE_GLOBAL).hydrate.aio(client)
-            print(f"➖ Found existing mount {mount_name} in global namespace.")
+            print(f"➖ Found existing mount {mount_name} in global namespace.")  # noqa: T201
             return
         except modal.exception.NotFoundError:
             pass
 
     with tempfile.TemporaryDirectory(ignore_cleanup_errors=True) as tmpd:
-        print(f"📦 Building {mount_name}.")
+        print(f"📦 Building {mount_name}.")  # noqa: T201
         requirements = os.path.join(os.path.dirname(__file__), f"builder/{builder_version}.txt")
         cmd = " ".join(
             [
@@ -810,11 +810,11 @@ async def _create_single_client_dependency_mount(
         await proc.wait()
         if proc.returncode:
             stdout, stderr = await proc.communicate()
-            print(stdout.decode("utf-8"))
-            print(stderr.decode("utf-8"))
+            print(stdout.decode("utf-8"))  # noqa: T201
+            print(stderr.decode("utf-8"))  # noqa: T201
             raise RuntimeError(f"Subprocess failed with {proc.returncode}")
 
-        print(f"🌐 Downloaded and unpacked {mount_name} packages to {tmpd}.")
+        print(f"🌐 Downloaded and unpacked {mount_name} packages to {tmpd}.")  # noqa: T201
 
         python_mount = Mount._from_local_dir(tmpd, remote_path=REMOTE_PACKAGES_PATH)
 
@@ -838,11 +838,11 @@ async def _create_single_client_dependency_mount(
                         allow_overwrite=allow_overwrite,
                         client=client,
                     )
-                    print(f"✅ Deployed mount {mount_name} to global namespace.")
+                    print(f"✅ Deployed mount {mount_name} to global namespace.")  # noqa: T201
                 except GRPCError as e:
-                    print(f"⚠️ Mount creation failed with {e.status}: {e.message}")
+                    print(f"⚠️ Mount creation failed with {e.status}: {e.message}")  # noqa: T201
             else:
-                print(f"Dry run - skipping deployment of mount {mount_name}")
+                print(f"Dry run - skipping deployment of mount {mount_name}")  # noqa: T201
 
 
 async def _create_client_dependency_mounts(

{modal-1.2.5.dev28 → modal-1.2.6}/modal.egg-info/PKG-INFO

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

{modal-1.2.5.dev28 → modal-1.2.6}/modal_proto/task_command_router.proto

@@ -2,6 +2,7 @@ syntax = "proto3";
 
 option go_package = "github.com/modal-labs/modal/go/proto";
 
+import "google/protobuf/empty.proto";
 import "modal_proto/api.proto";
 
 package modal.task_command_router;
@@ -130,6 +131,21 @@ message TaskExecWaitResponse {
   // and termination by Modal (due to task timeout, exec exceeded deadline, etc.)
 }
 
+message TaskMountDirectoryRequest {
+  string task_id = 1;
+  bytes path = 2;
+  string image_id = 3;
+}
+
+message TaskSnapshotDirectoryRequest {
+  string task_id = 1;
+  bytes path = 2;
+}
+
+message TaskSnapshotDirectoryResponse {
+  string image_id = 1;
+}
+
 service TaskCommandRouter {
   // Poll for the exit status of an exec'd command.
   rpc TaskExecPoll(TaskExecPollRequest) returns (TaskExecPollResponse);
@@ -141,4 +157,8 @@ service TaskCommandRouter {
   rpc TaskExecStdioRead(TaskExecStdioReadRequest) returns (stream TaskExecStdioReadResponse);
   // Wait for an exec'd command to exit and return the exit code.
   rpc TaskExecWait(TaskExecWaitRequest) returns (TaskExecWaitResponse);
+  // Mount an image at a directory in the container.
+  rpc TaskMountDirectory(TaskMountDirectoryRequest) returns (google.protobuf.Empty);
+  // Snapshot a directory with a mounted image, including any local changes, into a new image.
+  rpc TaskSnapshotDirectory(TaskSnapshotDirectoryRequest) returns (TaskSnapshotDirectoryResponse);
 }

{modal-1.2.5.dev28 → modal-1.2.6}/modal_proto/task_command_router_grpc.py

@@ -9,6 +9,7 @@ import grpclib.client
 if typing.TYPE_CHECKING:
     import grpclib.server
 
+import google.protobuf.empty_pb2
 import modal_proto.api_pb2
 import modal_proto.task_command_router_pb2
 
@@ -35,6 +36,14 @@ class TaskCommandRouterBase(abc.ABC):
     async def TaskExecWait(self, stream: 'grpclib.server.Stream[modal_proto.task_command_router_pb2.TaskExecWaitRequest, modal_proto.task_command_router_pb2.TaskExecWaitResponse]') -> None:
         pass
 
+    @abc.abstractmethod
+    async def TaskMountDirectory(self, stream: 'grpclib.server.Stream[modal_proto.task_command_router_pb2.TaskMountDirectoryRequest, google.protobuf.empty_pb2.Empty]') -> None:
+        pass
+
+    @abc.abstractmethod
+    async def TaskSnapshotDirectory(self, stream: 'grpclib.server.Stream[modal_proto.task_command_router_pb2.TaskSnapshotDirectoryRequest, modal_proto.task_command_router_pb2.TaskSnapshotDirectoryResponse]') -> None:
+        pass
+
     def __mapping__(self) -> typing.Dict[str, grpclib.const.Handler]:
         return {
             '/modal.task_command_router.TaskCommandRouter/TaskExecPoll': grpclib.const.Handler(
@@ -67,6 +76,18 @@ class TaskCommandRouterBase(abc.ABC):
                 modal_proto.task_command_router_pb2.TaskExecWaitRequest,
                 modal_proto.task_command_router_pb2.TaskExecWaitResponse,
             ),
+            '/modal.task_command_router.TaskCommandRouter/TaskMountDirectory': grpclib.const.Handler(
+                self.TaskMountDirectory,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.task_command_router_pb2.TaskMountDirectoryRequest,
+                google.protobuf.empty_pb2.Empty,
+            ),
+            '/modal.task_command_router.TaskCommandRouter/TaskSnapshotDirectory': grpclib.const.Handler(
+                self.TaskSnapshotDirectory,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.task_command_router_pb2.TaskSnapshotDirectoryRequest,
+                modal_proto.task_command_router_pb2.TaskSnapshotDirectoryResponse,
+            ),
         }
 
 
@@ -103,3 +124,15 @@ class TaskCommandRouterStub:
             modal_proto.task_command_router_pb2.TaskExecWaitRequest,
             modal_proto.task_command_router_pb2.TaskExecWaitResponse,
         )
+        self.TaskMountDirectory = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.task_command_router.TaskCommandRouter/TaskMountDirectory',
+            modal_proto.task_command_router_pb2.TaskMountDirectoryRequest,
+            google.protobuf.empty_pb2.Empty,
+        )
+        self.TaskSnapshotDirectory = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.task_command_router.TaskCommandRouter/TaskSnapshotDirectory',
+            modal_proto.task_command_router_pb2.TaskSnapshotDirectoryRequest,
+            modal_proto.task_command_router_pb2.TaskSnapshotDirectoryResponse,
+        )

modal-1.2.6/modal_proto/task_command_router_pb2.py

@@ -0,0 +1,180 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: modal_proto/task_command_router.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+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\"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\"\x92\x03\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\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\"=\n\x1cTaskSnapshotDirectoryRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\x0c\"1\n\x1dTaskSnapshotDirectoryResponse\x12\x10\n\x08image_id\x18\x01 \x01(\t*\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\xe1\x06\n\x11TaskCommandRouter\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.TaskSnapshotDirectoryResponseB&Z$github.com/modal-labs/modal/go/protob\x06proto3')
+
+_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)
+TASK_EXEC_STDERR_CONFIG_DEVNULL = 0
+TASK_EXEC_STDERR_CONFIG_PIPE = 1
+TASK_EXEC_STDERR_CONFIG_STDOUT = 2
+TASK_EXEC_STDIO_FILE_DESCRIPTOR_STDOUT = 0
+TASK_EXEC_STDIO_FILE_DESCRIPTOR_STDERR = 1
+TASK_EXEC_STDOUT_CONFIG_DEVNULL = 0
+TASK_EXEC_STDOUT_CONFIG_PIPE = 1
+
+
+_TASKEXECPOLLREQUEST = DESCRIPTOR.message_types_by_name['TaskExecPollRequest']
+_TASKEXECPOLLRESPONSE = DESCRIPTOR.message_types_by_name['TaskExecPollResponse']
+_TASKEXECSTARTREQUEST = DESCRIPTOR.message_types_by_name['TaskExecStartRequest']
+_TASKEXECSTARTRESPONSE = DESCRIPTOR.message_types_by_name['TaskExecStartResponse']
+_TASKEXECSTDINWRITEREQUEST = DESCRIPTOR.message_types_by_name['TaskExecStdinWriteRequest']
+_TASKEXECSTDINWRITERESPONSE = DESCRIPTOR.message_types_by_name['TaskExecStdinWriteResponse']
+_TASKEXECSTDIOREADREQUEST = DESCRIPTOR.message_types_by_name['TaskExecStdioReadRequest']
+_TASKEXECSTDIOREADRESPONSE = DESCRIPTOR.message_types_by_name['TaskExecStdioReadResponse']
+_TASKEXECWAITREQUEST = DESCRIPTOR.message_types_by_name['TaskExecWaitRequest']
+_TASKEXECWAITRESPONSE = DESCRIPTOR.message_types_by_name['TaskExecWaitResponse']
+_TASKMOUNTDIRECTORYREQUEST = DESCRIPTOR.message_types_by_name['TaskMountDirectoryRequest']
+_TASKSNAPSHOTDIRECTORYREQUEST = DESCRIPTOR.message_types_by_name['TaskSnapshotDirectoryRequest']
+_TASKSNAPSHOTDIRECTORYRESPONSE = DESCRIPTOR.message_types_by_name['TaskSnapshotDirectoryResponse']
+TaskExecPollRequest = _reflection.GeneratedProtocolMessageType('TaskExecPollRequest', (_message.Message,), {
+  'DESCRIPTOR' : _TASKEXECPOLLREQUEST,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskExecPollRequest)
+  })
+_sym_db.RegisterMessage(TaskExecPollRequest)
+
+TaskExecPollResponse = _reflection.GeneratedProtocolMessageType('TaskExecPollResponse', (_message.Message,), {
+  'DESCRIPTOR' : _TASKEXECPOLLRESPONSE,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskExecPollResponse)
+  })
+_sym_db.RegisterMessage(TaskExecPollResponse)
+
+TaskExecStartRequest = _reflection.GeneratedProtocolMessageType('TaskExecStartRequest', (_message.Message,), {
+  'DESCRIPTOR' : _TASKEXECSTARTREQUEST,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskExecStartRequest)
+  })
+_sym_db.RegisterMessage(TaskExecStartRequest)
+
+TaskExecStartResponse = _reflection.GeneratedProtocolMessageType('TaskExecStartResponse', (_message.Message,), {
+  'DESCRIPTOR' : _TASKEXECSTARTRESPONSE,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskExecStartResponse)
+  })
+_sym_db.RegisterMessage(TaskExecStartResponse)
+
+TaskExecStdinWriteRequest = _reflection.GeneratedProtocolMessageType('TaskExecStdinWriteRequest', (_message.Message,), {
+  'DESCRIPTOR' : _TASKEXECSTDINWRITEREQUEST,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskExecStdinWriteRequest)
+  })
+_sym_db.RegisterMessage(TaskExecStdinWriteRequest)
+
+TaskExecStdinWriteResponse = _reflection.GeneratedProtocolMessageType('TaskExecStdinWriteResponse', (_message.Message,), {
+  'DESCRIPTOR' : _TASKEXECSTDINWRITERESPONSE,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskExecStdinWriteResponse)
+  })
+_sym_db.RegisterMessage(TaskExecStdinWriteResponse)
+
+TaskExecStdioReadRequest = _reflection.GeneratedProtocolMessageType('TaskExecStdioReadRequest', (_message.Message,), {
+  'DESCRIPTOR' : _TASKEXECSTDIOREADREQUEST,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskExecStdioReadRequest)
+  })
+_sym_db.RegisterMessage(TaskExecStdioReadRequest)
+
+TaskExecStdioReadResponse = _reflection.GeneratedProtocolMessageType('TaskExecStdioReadResponse', (_message.Message,), {
+  'DESCRIPTOR' : _TASKEXECSTDIOREADRESPONSE,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskExecStdioReadResponse)
+  })
+_sym_db.RegisterMessage(TaskExecStdioReadResponse)
+
+TaskExecWaitRequest = _reflection.GeneratedProtocolMessageType('TaskExecWaitRequest', (_message.Message,), {
+  'DESCRIPTOR' : _TASKEXECWAITREQUEST,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskExecWaitRequest)
+  })
+_sym_db.RegisterMessage(TaskExecWaitRequest)
+
+TaskExecWaitResponse = _reflection.GeneratedProtocolMessageType('TaskExecWaitResponse', (_message.Message,), {
+  'DESCRIPTOR' : _TASKEXECWAITRESPONSE,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskExecWaitResponse)
+  })
+_sym_db.RegisterMessage(TaskExecWaitResponse)
+
+TaskMountDirectoryRequest = _reflection.GeneratedProtocolMessageType('TaskMountDirectoryRequest', (_message.Message,), {
+  'DESCRIPTOR' : _TASKMOUNTDIRECTORYREQUEST,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskMountDirectoryRequest)
+  })
+_sym_db.RegisterMessage(TaskMountDirectoryRequest)
+
+TaskSnapshotDirectoryRequest = _reflection.GeneratedProtocolMessageType('TaskSnapshotDirectoryRequest', (_message.Message,), {
+  'DESCRIPTOR' : _TASKSNAPSHOTDIRECTORYREQUEST,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskSnapshotDirectoryRequest)
+  })
+_sym_db.RegisterMessage(TaskSnapshotDirectoryRequest)
+
+TaskSnapshotDirectoryResponse = _reflection.GeneratedProtocolMessageType('TaskSnapshotDirectoryResponse', (_message.Message,), {
+  'DESCRIPTOR' : _TASKSNAPSHOTDIRECTORYRESPONSE,
+  '__module__' : 'modal_proto.task_command_router_pb2'
+  # @@protoc_insertion_point(class_scope:modal.task_command_router.TaskSnapshotDirectoryResponse)
+  })
+_sym_db.RegisterMessage(TaskSnapshotDirectoryResponse)
+
+_TASKCOMMANDROUTER = DESCRIPTOR.services_by_name['TaskCommandRouter']
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'Z$github.com/modal-labs/modal/go/proto'
+  _TASKEXECSTDERRCONFIG._serialized_start=1342
+  _TASKEXECSTDERRCONFIG._serialized_end=1471
+  _TASKEXECSTDIOFILEDESCRIPTOR._serialized_start=1473
+  _TASKEXECSTDIOFILEDESCRIPTOR._serialized_end=1590
+  _TASKEXECSTDOUTCONFIG._serialized_start=1592
+  _TASKEXECSTDOUTCONFIG._serialized_end=1685
+  _TASKEXECPOLLREQUEST._serialized_start=120
+  _TASKEXECPOLLREQUEST._serialized_end=175
+  _TASKEXECPOLLRESPONSE._serialized_start=177
+  _TASKEXECPOLLRESPONSE._serialized_end=248
+  _TASKEXECSTARTREQUEST._serialized_start=251
+  _TASKEXECSTARTREQUEST._serialized_end=653
+  _TASKEXECSTARTRESPONSE._serialized_start=655
+  _TASKEXECSTARTRESPONSE._serialized_end=678
+  _TASKEXECSTDINWRITEREQUEST._serialized_start=680
+  _TASKEXECSTDINWRITEREQUEST._serialized_end=784
+  _TASKEXECSTDINWRITERESPONSE._serialized_start=786
+  _TASKEXECSTDINWRITERESPONSE._serialized_end=814
+  _TASKEXECSTDIOREADREQUEST._serialized_start=817
+  _TASKEXECSTDIOREADREQUEST._serialized_end=974
+  _TASKEXECSTDIOREADRESPONSE._serialized_start=976
+  _TASKEXECSTDIOREADRESPONSE._serialized_end=1017
+  _TASKEXECWAITREQUEST._serialized_start=1019
+  _TASKEXECWAITREQUEST._serialized_end=1074
+  _TASKEXECWAITRESPONSE._serialized_start=1076
+  _TASKEXECWAITRESPONSE._serialized_end=1147
+  _TASKMOUNTDIRECTORYREQUEST._serialized_start=1149
+  _TASKMOUNTDIRECTORYREQUEST._serialized_end=1225
+  _TASKSNAPSHOTDIRECTORYREQUEST._serialized_start=1227
+  _TASKSNAPSHOTDIRECTORYREQUEST._serialized_end=1288
+  _TASKSNAPSHOTDIRECTORYRESPONSE._serialized_start=1290
+  _TASKSNAPSHOTDIRECTORYRESPONSE._serialized_end=1339
+  _TASKCOMMANDROUTER._serialized_start=1688
+  _TASKCOMMANDROUTER._serialized_end=2553
+# @@protoc_insertion_point(module_scope)

{modal-1.2.5.dev28 → modal-1.2.6}/modal_proto/task_command_router_pb2.pyi

@@ -331,3 +331,54 @@ class TaskExecWaitResponse(google.protobuf.message.Message):
     def WhichOneof(self, oneof_group: typing_extensions.Literal["exit_status", b"exit_status"]) -> typing_extensions.Literal["code", "signal"] | None: ...
 
 global___TaskExecWaitResponse = TaskExecWaitResponse
+
+class TaskMountDirectoryRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    PATH_FIELD_NUMBER: builtins.int
+    IMAGE_ID_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    path: builtins.bytes
+    image_id: builtins.str
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+        path: builtins.bytes = ...,
+        image_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["image_id", b"image_id", "path", b"path", "task_id", b"task_id"]) -> None: ...
+
+global___TaskMountDirectoryRequest = TaskMountDirectoryRequest
+
+class TaskSnapshotDirectoryRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    PATH_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    path: builtins.bytes
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+        path: builtins.bytes = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["path", b"path", "task_id", b"task_id"]) -> None: ...
+
+global___TaskSnapshotDirectoryRequest = TaskSnapshotDirectoryRequest
+
+class TaskSnapshotDirectoryResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    IMAGE_ID_FIELD_NUMBER: builtins.int
+    image_id: builtins.str
+    def __init__(
+        self,
+        *,
+        image_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["image_id", b"image_id"]) -> None: ...
+
+global___TaskSnapshotDirectoryResponse = TaskSnapshotDirectoryResponse