modAL 1.2.5.dev28__tar.gz → 1.2.6.dev0__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modal
-Version: 1.2.5.dev28
+Version: 1.2.6.dev0
 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.dev0}/modal/client.pyi

@@ -32,7 +32,7 @@ class _Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.2.5.dev28",
+        version: str = "1.2.6.dev0",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.
@@ -163,7 +163,7 @@ class Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.2.5.dev28",
+        version: str = "1.2.6.dev0",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modal
-Version: 1.2.5.dev28
+Version: 1.2.6.dev0
 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.dev0}/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.dev0}/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.dev0/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.dev0}/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

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

@@ -2,6 +2,7 @@
 """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
 
 
@@ -39,6 +40,16 @@ class TaskCommandRouterStub(object):
                 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):
@@ -79,6 +90,20 @@ class TaskCommandRouterServicer(object):
         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 = {
@@ -107,6 +132,16 @@ def add_TaskCommandRouterServicer_to_server(servicer, server):
                     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)
@@ -201,3 +236,37 @@ class TaskCommandRouter(object):
             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)

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

@@ -4,6 +4,7 @@ isort:skip_file
 """
 import abc
 import collections.abc
+import google.protobuf.empty_pb2
 import grpc
 import modal_proto.task_command_router_pb2
 
@@ -34,6 +35,16 @@ class TaskCommandRouterStub:
         modal_proto.task_command_router_pb2.TaskExecWaitResponse,
     ]
     """Wait for an exec'd command to exit and return the exit code."""
+    TaskMountDirectory: grpc.UnaryUnaryMultiCallable[
+        modal_proto.task_command_router_pb2.TaskMountDirectoryRequest,
+        google.protobuf.empty_pb2.Empty,
+    ]
+    """Mount an image at a directory in the container."""
+    TaskSnapshotDirectory: grpc.UnaryUnaryMultiCallable[
+        modal_proto.task_command_router_pb2.TaskSnapshotDirectoryRequest,
+        modal_proto.task_command_router_pb2.TaskSnapshotDirectoryResponse,
+    ]
+    """Snapshot a directory with a mounted image, including any local changes, into a new image."""
 
 class TaskCommandRouterServicer(metaclass=abc.ABCMeta):
     @abc.abstractmethod
@@ -71,5 +82,19 @@ class TaskCommandRouterServicer(metaclass=abc.ABCMeta):
         context: grpc.ServicerContext,
     ) -> modal_proto.task_command_router_pb2.TaskExecWaitResponse:
         """Wait for an exec'd command to exit and return the exit code."""
+    @abc.abstractmethod
+    def TaskMountDirectory(
+        self,
+        request: modal_proto.task_command_router_pb2.TaskMountDirectoryRequest,
+        context: grpc.ServicerContext,
+    ) -> google.protobuf.empty_pb2.Empty:
+        """Mount an image at a directory in the container."""
+    @abc.abstractmethod
+    def TaskSnapshotDirectory(
+        self,
+        request: modal_proto.task_command_router_pb2.TaskSnapshotDirectoryRequest,
+        context: grpc.ServicerContext,
+    ) -> modal_proto.task_command_router_pb2.TaskSnapshotDirectoryResponse:
+        """Snapshot a directory with a mounted image, including any local changes, into a new image."""
 
 def add_TaskCommandRouterServicer_to_server(servicer: TaskCommandRouterServicer, server: grpc.Server) -> None: ...

{modal-1.2.5.dev28 → modal-1.2.6.dev0}/modal_version/__init__.py

@@ -1,4 +1,4 @@
 # Copyright Modal Labs 2025
 """Supplies the current version of the modal client library."""
 
-__version__ = "1.2.5.dev28"
+__version__ = "1.2.6.dev0"