modal 1.1.5.dev66__py3-none-any.whl → 1.3.1.dev8__py3-none-any.whl

modal_proto/{sandbox_router_pb2.pyi → task_command_router_pb2.pyi}

@@ -19,72 +19,116 @@ else:
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
-class _SandboxExecStderrConfig:
+class _TaskExecStderrConfig:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _SandboxExecStderrConfigEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SandboxExecStderrConfig.ValueType], builtins.type):  # noqa: F821
+class _TaskExecStderrConfigEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TaskExecStderrConfig.ValueType], builtins.type):  # noqa: F821
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    SANDBOX_EXEC_STDERR_CONFIG_DEVNULL: _SandboxExecStderrConfig.ValueType  # 0
+    TASK_EXEC_STDERR_CONFIG_DEVNULL: _TaskExecStderrConfig.ValueType  # 0
     """The output will be discarded."""
-    SANDBOX_EXEC_STDERR_CONFIG_PIPE: _SandboxExecStderrConfig.ValueType  # 1
+    TASK_EXEC_STDERR_CONFIG_PIPE: _TaskExecStderrConfig.ValueType  # 1
     """The output will be streamed to the client."""
-    SANDBOX_EXEC_STDERR_CONFIG_STDOUT: _SandboxExecStderrConfig.ValueType  # 2
+    TASK_EXEC_STDERR_CONFIG_STDOUT: _TaskExecStderrConfig.ValueType  # 2
     """A special value that can be used to indicate that the stderr stream should
     be merged with the stdout stream.
     """
 
-class SandboxExecStderrConfig(_SandboxExecStderrConfig, metaclass=_SandboxExecStderrConfigEnumTypeWrapper): ...
+class TaskExecStderrConfig(_TaskExecStderrConfig, metaclass=_TaskExecStderrConfigEnumTypeWrapper): ...
 
-SANDBOX_EXEC_STDERR_CONFIG_DEVNULL: SandboxExecStderrConfig.ValueType  # 0
+TASK_EXEC_STDERR_CONFIG_DEVNULL: TaskExecStderrConfig.ValueType  # 0
 """The output will be discarded."""
-SANDBOX_EXEC_STDERR_CONFIG_PIPE: SandboxExecStderrConfig.ValueType  # 1
+TASK_EXEC_STDERR_CONFIG_PIPE: TaskExecStderrConfig.ValueType  # 1
 """The output will be streamed to the client."""
-SANDBOX_EXEC_STDERR_CONFIG_STDOUT: SandboxExecStderrConfig.ValueType  # 2
+TASK_EXEC_STDERR_CONFIG_STDOUT: TaskExecStderrConfig.ValueType  # 2
 """A special value that can be used to indicate that the stderr stream should
 be merged with the stdout stream.
 """
-global___SandboxExecStderrConfig = SandboxExecStderrConfig
+global___TaskExecStderrConfig = TaskExecStderrConfig
 
-class _SandboxExecStdioFileDescriptor:
+class _TaskExecStdioFileDescriptor:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _SandboxExecStdioFileDescriptorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SandboxExecStdioFileDescriptor.ValueType], builtins.type):  # noqa: F821
+class _TaskExecStdioFileDescriptorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TaskExecStdioFileDescriptor.ValueType], builtins.type):  # noqa: F821
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    SANDBOX_EXEC_STDIO_FILE_DESCRIPTOR_STDOUT: _SandboxExecStdioFileDescriptor.ValueType  # 0
+    TASK_EXEC_STDIO_FILE_DESCRIPTOR_STDOUT: _TaskExecStdioFileDescriptor.ValueType  # 0
     """Read from stdout."""
-    SANDBOX_EXEC_STDIO_FILE_DESCRIPTOR_STDERR: _SandboxExecStdioFileDescriptor.ValueType  # 1
+    TASK_EXEC_STDIO_FILE_DESCRIPTOR_STDERR: _TaskExecStdioFileDescriptor.ValueType  # 1
     """Read from stderr."""
 
-class SandboxExecStdioFileDescriptor(_SandboxExecStdioFileDescriptor, metaclass=_SandboxExecStdioFileDescriptorEnumTypeWrapper): ...
+class TaskExecStdioFileDescriptor(_TaskExecStdioFileDescriptor, metaclass=_TaskExecStdioFileDescriptorEnumTypeWrapper): ...
 
-SANDBOX_EXEC_STDIO_FILE_DESCRIPTOR_STDOUT: SandboxExecStdioFileDescriptor.ValueType  # 0
+TASK_EXEC_STDIO_FILE_DESCRIPTOR_STDOUT: TaskExecStdioFileDescriptor.ValueType  # 0
 """Read from stdout."""
-SANDBOX_EXEC_STDIO_FILE_DESCRIPTOR_STDERR: SandboxExecStdioFileDescriptor.ValueType  # 1
+TASK_EXEC_STDIO_FILE_DESCRIPTOR_STDERR: TaskExecStdioFileDescriptor.ValueType  # 1
 """Read from stderr."""
-global___SandboxExecStdioFileDescriptor = SandboxExecStdioFileDescriptor
+global___TaskExecStdioFileDescriptor = TaskExecStdioFileDescriptor
 
-class _SandboxExecStdoutConfig:
+class _TaskExecStdoutConfig:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _SandboxExecStdoutConfigEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SandboxExecStdoutConfig.ValueType], builtins.type):  # noqa: F821
+class _TaskExecStdoutConfigEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TaskExecStdoutConfig.ValueType], builtins.type):  # noqa: F821
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    SANDBOX_EXEC_STDOUT_CONFIG_DEVNULL: _SandboxExecStdoutConfig.ValueType  # 0
+    TASK_EXEC_STDOUT_CONFIG_DEVNULL: _TaskExecStdoutConfig.ValueType  # 0
     """The output will be discarded."""
-    SANDBOX_EXEC_STDOUT_CONFIG_PIPE: _SandboxExecStdoutConfig.ValueType  # 1
+    TASK_EXEC_STDOUT_CONFIG_PIPE: _TaskExecStdoutConfig.ValueType  # 1
     """The output will be streamed to the client."""
 
-class SandboxExecStdoutConfig(_SandboxExecStdoutConfig, metaclass=_SandboxExecStdoutConfigEnumTypeWrapper): ...
+class TaskExecStdoutConfig(_TaskExecStdoutConfig, metaclass=_TaskExecStdoutConfigEnumTypeWrapper): ...
 
-SANDBOX_EXEC_STDOUT_CONFIG_DEVNULL: SandboxExecStdoutConfig.ValueType  # 0
+TASK_EXEC_STDOUT_CONFIG_DEVNULL: TaskExecStdoutConfig.ValueType  # 0
 """The output will be discarded."""
-SANDBOX_EXEC_STDOUT_CONFIG_PIPE: SandboxExecStdoutConfig.ValueType  # 1
+TASK_EXEC_STDOUT_CONFIG_PIPE: TaskExecStdoutConfig.ValueType  # 1
 """The output will be streamed to the client."""
-global___SandboxExecStdoutConfig = SandboxExecStdoutConfig
+global___TaskExecStdoutConfig = TaskExecStdoutConfig
 
-class SandboxExecStartRequest(google.protobuf.message.Message):
+class TaskExecPollRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    EXEC_ID_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    """The ID of the task running the exec'd command."""
+    exec_id: builtins.str
+    """The execution ID of the command to wait on."""
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+        exec_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["exec_id", b"exec_id", "task_id", b"task_id"]) -> None: ...
+
+global___TaskExecPollRequest = TaskExecPollRequest
+
+class TaskExecPollResponse(google.protobuf.message.Message):
+    """The response to a TaskExecPollRequest. If the exec'd command has not
+    completed, exit_status will be unset.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CODE_FIELD_NUMBER: builtins.int
+    SIGNAL_FIELD_NUMBER: builtins.int
+    code: builtins.int
+    """The exit code of the command."""
+    signal: builtins.int
+    """The signal that terminated the command."""
+    def __init__(
+        self,
+        *,
+        code: builtins.int = ...,
+        signal: builtins.int = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["code", b"code", "exit_status", b"exit_status", "signal", b"signal"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["code", b"code", "exit_status", b"exit_status", "signal", b"signal"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["exit_status", b"exit_status"]) -> typing_extensions.Literal["code", "signal"] | None: ...
+
+global___TaskExecPollResponse = TaskExecPollResponse
+
+class TaskExecStartRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     TASK_ID_FIELD_NUMBER: builtins.int
@@ -98,21 +142,17 @@ class SandboxExecStartRequest(google.protobuf.message.Message):
     PTY_INFO_FIELD_NUMBER: builtins.int
     RUNTIME_DEBUG_FIELD_NUMBER: builtins.int
     task_id: builtins.str
-    """The task ID of the sandbox to execute the command in."""
+    """The ID of the task to execute the command in."""
     exec_id: builtins.str
     """Execution ID. This ID will be used to identify the execution for other
     requests and ensure exec commands are idempotent.
-
-    TODO(saltzm): Could instead have a separate idempotency key from the exec_id
-    like present day, and have the server generate the exec_id and return it in
-    the ExecStartResponse.
     """
     @property
     def command_args(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
         """Command arguments to execute."""
-    stdout_config: global___SandboxExecStdoutConfig.ValueType
+    stdout_config: global___TaskExecStdoutConfig.ValueType
     """Configures how the stdout of the command will be handled."""
-    stderr_config: global___SandboxExecStderrConfig.ValueType
+    stderr_config: global___TaskExecStderrConfig.ValueType
     """Configures how the stderr of the command will be handled."""
     timeout_secs: builtins.int
     """Timeout in seconds for the exec'd command to exit. If the command does not
@@ -123,7 +163,7 @@ class SandboxExecStartRequest(google.protobuf.message.Message):
     """Working directory for the command."""
     @property
     def secret_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
-        """Secret IDs to mount into the sandbox."""
+        """Secret IDs to mount into the task."""
     @property
     def pty_info(self) -> modal_proto.api_pb2.PTYInfo:
         """PTY info for the command."""
@@ -137,8 +177,8 @@ class SandboxExecStartRequest(google.protobuf.message.Message):
         task_id: builtins.str = ...,
         exec_id: builtins.str = ...,
         command_args: collections.abc.Iterable[builtins.str] | None = ...,
-        stdout_config: global___SandboxExecStdoutConfig.ValueType = ...,
-        stderr_config: global___SandboxExecStderrConfig.ValueType = ...,
+        stdout_config: global___TaskExecStdoutConfig.ValueType = ...,
+        stderr_config: global___TaskExecStderrConfig.ValueType = ...,
         timeout_secs: builtins.int | None = ...,
         workdir: builtins.str | None = ...,
         secret_ids: collections.abc.Iterable[builtins.str] | None = ...,
@@ -154,18 +194,18 @@ class SandboxExecStartRequest(google.protobuf.message.Message):
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_workdir", b"_workdir"]) -> typing_extensions.Literal["workdir"] | None: ...
 
-global___SandboxExecStartRequest = SandboxExecStartRequest
+global___TaskExecStartRequest = TaskExecStartRequest
 
-class SandboxExecStartResponse(google.protobuf.message.Message):
+class TaskExecStartResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     def __init__(
         self,
     ) -> None: ...
 
-global___SandboxExecStartResponse = SandboxExecStartResponse
+global___TaskExecStartResponse = TaskExecStartResponse
 
-class SandboxExecStdinWriteRequest(google.protobuf.message.Message):
+class TaskExecStdinWriteRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     TASK_ID_FIELD_NUMBER: builtins.int
@@ -174,7 +214,7 @@ class SandboxExecStdinWriteRequest(google.protobuf.message.Message):
     DATA_FIELD_NUMBER: builtins.int
     EOF_FIELD_NUMBER: builtins.int
     task_id: builtins.str
-    """The task ID of the sandbox running the exec'd command."""
+    """The ID of the task running the exec'd command."""
     exec_id: builtins.str
     """The execution ID of the command to write to."""
     offset: builtins.int
@@ -197,18 +237,18 @@ class SandboxExecStdinWriteRequest(google.protobuf.message.Message):
     ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["data", b"data", "eof", b"eof", "exec_id", b"exec_id", "offset", b"offset", "task_id", b"task_id"]) -> None: ...
 
-global___SandboxExecStdinWriteRequest = SandboxExecStdinWriteRequest
+global___TaskExecStdinWriteRequest = TaskExecStdinWriteRequest
 
-class SandboxExecStdinWriteResponse(google.protobuf.message.Message):
+class TaskExecStdinWriteResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     def __init__(
         self,
     ) -> None: ...
 
-global___SandboxExecStdinWriteResponse = SandboxExecStdinWriteResponse
+global___TaskExecStdinWriteResponse = TaskExecStdinWriteResponse
 
-class SandboxExecStdioReadRequest(google.protobuf.message.Message):
+class TaskExecStdioReadRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     TASK_ID_FIELD_NUMBER: builtins.int
@@ -216,14 +256,14 @@ class SandboxExecStdioReadRequest(google.protobuf.message.Message):
     OFFSET_FIELD_NUMBER: builtins.int
     FILE_DESCRIPTOR_FIELD_NUMBER: builtins.int
     task_id: builtins.str
-    """The task ID of the sandbox running the exec'd command."""
+    """The ID of the task running the exec'd command."""
     exec_id: builtins.str
     """The execution ID of the command to read from."""
     offset: builtins.int
     """The offset to start reading from. This is used to resume reading from the
     last read position if the connection is closed and reopened.
     """
-    file_descriptor: global___SandboxExecStdioFileDescriptor.ValueType
+    file_descriptor: global___TaskExecStdioFileDescriptor.ValueType
     """Which file descriptor to read from."""
     def __init__(
         self,
@@ -231,13 +271,13 @@ class SandboxExecStdioReadRequest(google.protobuf.message.Message):
         task_id: builtins.str = ...,
         exec_id: builtins.str = ...,
         offset: builtins.int = ...,
-        file_descriptor: global___SandboxExecStdioFileDescriptor.ValueType = ...,
+        file_descriptor: global___TaskExecStdioFileDescriptor.ValueType = ...,
     ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["exec_id", b"exec_id", "file_descriptor", b"file_descriptor", "offset", b"offset", "task_id", b"task_id"]) -> None: ...
 
-global___SandboxExecStdioReadRequest = SandboxExecStdioReadRequest
+global___TaskExecStdioReadRequest = TaskExecStdioReadRequest
 
-class SandboxExecStdioReadResponse(google.protobuf.message.Message):
+class TaskExecStdioReadResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     DATA_FIELD_NUMBER: builtins.int
@@ -250,15 +290,15 @@ class SandboxExecStdioReadResponse(google.protobuf.message.Message):
     ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["data", b"data"]) -> None: ...
 
-global___SandboxExecStdioReadResponse = SandboxExecStdioReadResponse
+global___TaskExecStdioReadResponse = TaskExecStdioReadResponse
 
-class SandboxExecWaitRequest(google.protobuf.message.Message):
+class TaskExecWaitRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     TASK_ID_FIELD_NUMBER: builtins.int
     EXEC_ID_FIELD_NUMBER: builtins.int
     task_id: builtins.str
-    """The task ID of the sandbox running the exec'd command."""
+    """The ID of the task running the exec'd command."""
     exec_id: builtins.str
     """The execution ID of the command to wait on."""
     def __init__(
@@ -269,9 +309,9 @@ class SandboxExecWaitRequest(google.protobuf.message.Message):
     ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["exec_id", b"exec_id", "task_id", b"task_id"]) -> None: ...
 
-global___SandboxExecWaitRequest = SandboxExecWaitRequest
+global___TaskExecWaitRequest = TaskExecWaitRequest
 
-class SandboxExecWaitResponse(google.protobuf.message.Message):
+class TaskExecWaitResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     CODE_FIELD_NUMBER: builtins.int
@@ -290,4 +330,55 @@ class SandboxExecWaitResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["code", b"code", "exit_status", b"exit_status", "signal", b"signal"]) -> None: ...
     def WhichOneof(self, oneof_group: typing_extensions.Literal["exit_status", b"exit_status"]) -> typing_extensions.Literal["code", "signal"] | None: ...
 
-global___SandboxExecWaitResponse = SandboxExecWaitResponse
+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

modal_proto/task_command_router_pb2_grpc.py

@@ -0,0 +1,272 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
+from modal_proto import task_command_router_pb2 as modal__proto_dot_task__command__router__pb2
+
+
+class TaskCommandRouterStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.TaskExecPoll = channel.unary_unary(
+                '/modal.task_command_router.TaskCommandRouter/TaskExecPoll',
+                request_serializer=modal__proto_dot_task__command__router__pb2.TaskExecPollRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_task__command__router__pb2.TaskExecPollResponse.FromString,
+                )
+        self.TaskExecStart = channel.unary_unary(
+                '/modal.task_command_router.TaskCommandRouter/TaskExecStart',
+                request_serializer=modal__proto_dot_task__command__router__pb2.TaskExecStartRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_task__command__router__pb2.TaskExecStartResponse.FromString,
+                )
+        self.TaskExecStdinWrite = channel.unary_unary(
+                '/modal.task_command_router.TaskCommandRouter/TaskExecStdinWrite',
+                request_serializer=modal__proto_dot_task__command__router__pb2.TaskExecStdinWriteRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_task__command__router__pb2.TaskExecStdinWriteResponse.FromString,
+                )
+        self.TaskExecStdioRead = channel.unary_stream(
+                '/modal.task_command_router.TaskCommandRouter/TaskExecStdioRead',
+                request_serializer=modal__proto_dot_task__command__router__pb2.TaskExecStdioReadRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_task__command__router__pb2.TaskExecStdioReadResponse.FromString,
+                )
+        self.TaskExecWait = channel.unary_unary(
+                '/modal.task_command_router.TaskCommandRouter/TaskExecWait',
+                request_serializer=modal__proto_dot_task__command__router__pb2.TaskExecWaitRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_task__command__router__pb2.TaskExecWaitResponse.FromString,
+                )
+        self.TaskMountDirectory = channel.unary_unary(
+                '/modal.task_command_router.TaskCommandRouter/TaskMountDirectory',
+                request_serializer=modal__proto_dot_task__command__router__pb2.TaskMountDirectoryRequest.SerializeToString,
+                response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+                )
+        self.TaskSnapshotDirectory = channel.unary_unary(
+                '/modal.task_command_router.TaskCommandRouter/TaskSnapshotDirectory',
+                request_serializer=modal__proto_dot_task__command__router__pb2.TaskSnapshotDirectoryRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_task__command__router__pb2.TaskSnapshotDirectoryResponse.FromString,
+                )
+
+
+class TaskCommandRouterServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def TaskExecPoll(self, request, context):
+        """Poll for the exit status of an exec'd command.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def TaskExecStart(self, request, context):
+        """Execute a command in the task.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def TaskExecStdinWrite(self, request, context):
+        """Write to the stdin stream of an exec'd command.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def TaskExecStdioRead(self, request, context):
+        """Get a stream of output from the stdout or stderr stream of an exec'd command.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def TaskExecWait(self, request, context):
+        """Wait for an exec'd command to exit and return the exit code.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def TaskMountDirectory(self, request, context):
+        """Mount an image at a directory in the container.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def TaskSnapshotDirectory(self, request, context):
+        """Snapshot a directory with a mounted image, including any local changes, into a new image.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_TaskCommandRouterServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'TaskExecPoll': grpc.unary_unary_rpc_method_handler(
+                    servicer.TaskExecPoll,
+                    request_deserializer=modal__proto_dot_task__command__router__pb2.TaskExecPollRequest.FromString,
+                    response_serializer=modal__proto_dot_task__command__router__pb2.TaskExecPollResponse.SerializeToString,
+            ),
+            'TaskExecStart': grpc.unary_unary_rpc_method_handler(
+                    servicer.TaskExecStart,
+                    request_deserializer=modal__proto_dot_task__command__router__pb2.TaskExecStartRequest.FromString,
+                    response_serializer=modal__proto_dot_task__command__router__pb2.TaskExecStartResponse.SerializeToString,
+            ),
+            'TaskExecStdinWrite': grpc.unary_unary_rpc_method_handler(
+                    servicer.TaskExecStdinWrite,
+                    request_deserializer=modal__proto_dot_task__command__router__pb2.TaskExecStdinWriteRequest.FromString,
+                    response_serializer=modal__proto_dot_task__command__router__pb2.TaskExecStdinWriteResponse.SerializeToString,
+            ),
+            'TaskExecStdioRead': grpc.unary_stream_rpc_method_handler(
+                    servicer.TaskExecStdioRead,
+                    request_deserializer=modal__proto_dot_task__command__router__pb2.TaskExecStdioReadRequest.FromString,
+                    response_serializer=modal__proto_dot_task__command__router__pb2.TaskExecStdioReadResponse.SerializeToString,
+            ),
+            'TaskExecWait': grpc.unary_unary_rpc_method_handler(
+                    servicer.TaskExecWait,
+                    request_deserializer=modal__proto_dot_task__command__router__pb2.TaskExecWaitRequest.FromString,
+                    response_serializer=modal__proto_dot_task__command__router__pb2.TaskExecWaitResponse.SerializeToString,
+            ),
+            'TaskMountDirectory': grpc.unary_unary_rpc_method_handler(
+                    servicer.TaskMountDirectory,
+                    request_deserializer=modal__proto_dot_task__command__router__pb2.TaskMountDirectoryRequest.FromString,
+                    response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
+            ),
+            'TaskSnapshotDirectory': grpc.unary_unary_rpc_method_handler(
+                    servicer.TaskSnapshotDirectory,
+                    request_deserializer=modal__proto_dot_task__command__router__pb2.TaskSnapshotDirectoryRequest.FromString,
+                    response_serializer=modal__proto_dot_task__command__router__pb2.TaskSnapshotDirectoryResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'modal.task_command_router.TaskCommandRouter', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class TaskCommandRouter(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def TaskExecPoll(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/TaskExecPoll',
+            modal__proto_dot_task__command__router__pb2.TaskExecPollRequest.SerializeToString,
+            modal__proto_dot_task__command__router__pb2.TaskExecPollResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def TaskExecStart(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/TaskExecStart',
+            modal__proto_dot_task__command__router__pb2.TaskExecStartRequest.SerializeToString,
+            modal__proto_dot_task__command__router__pb2.TaskExecStartResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def TaskExecStdinWrite(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/TaskExecStdinWrite',
+            modal__proto_dot_task__command__router__pb2.TaskExecStdinWriteRequest.SerializeToString,
+            modal__proto_dot_task__command__router__pb2.TaskExecStdinWriteResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def TaskExecStdioRead(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/TaskExecStdioRead',
+            modal__proto_dot_task__command__router__pb2.TaskExecStdioReadRequest.SerializeToString,
+            modal__proto_dot_task__command__router__pb2.TaskExecStdioReadResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def TaskExecWait(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/TaskExecWait',
+            modal__proto_dot_task__command__router__pb2.TaskExecWaitRequest.SerializeToString,
+            modal__proto_dot_task__command__router__pb2.TaskExecWaitResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def TaskMountDirectory(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/TaskMountDirectory',
+            modal__proto_dot_task__command__router__pb2.TaskMountDirectoryRequest.SerializeToString,
+            google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def TaskSnapshotDirectory(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/TaskSnapshotDirectory',
+            modal__proto_dot_task__command__router__pb2.TaskSnapshotDirectoryRequest.SerializeToString,
+            modal__proto_dot_task__command__router__pb2.TaskSnapshotDirectoryResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)