modal 0.72.24__py3-none-any.whl → 0.72.26__py3-none-any.whl

modal_proto/api_pb2.pyi

@@ -1898,6 +1898,7 @@ class CheckpointInfo(google.protobuf.message.Message):
     SIZE_FIELD_NUMBER: builtins.int
     CHECKSUM_IS_FILE_INDEX_FIELD_NUMBER: builtins.int
     ORIGINAL_TASK_ID_FIELD_NUMBER: builtins.int
+    SNAPSHOT_VERSION_FIELD_NUMBER: builtins.int
     checksum: builtins.str
     status: global___CheckpointStatus.ValueType
     checkpoint_id: builtins.str
@@ -1905,6 +1906,8 @@ class CheckpointInfo(google.protobuf.message.Message):
     size: builtins.int
     checksum_is_file_index: builtins.bool
     original_task_id: builtins.str
+    snapshot_version: builtins.int
+    """snapshot version used to take the checkpoint"""
     def __init__(
         self,
         *,
@@ -1915,8 +1918,9 @@ class CheckpointInfo(google.protobuf.message.Message):
         size: builtins.int = ...,
         checksum_is_file_index: builtins.bool = ...,
         original_task_id: builtins.str = ...,
+        snapshot_version: builtins.int = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["checkpoint_id", b"checkpoint_id", "checksum", b"checksum", "checksum_is_file_index", b"checksum_is_file_index", "original_task_id", b"original_task_id", "runtime_fingerprint", b"runtime_fingerprint", "size", b"size", "status", b"status"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["checkpoint_id", b"checkpoint_id", "checksum", b"checksum", "checksum_is_file_index", b"checksum_is_file_index", "original_task_id", b"original_task_id", "runtime_fingerprint", b"runtime_fingerprint", "size", b"size", "snapshot_version", b"snapshot_version", "status", b"status"]) -> None: ...
 
 global___CheckpointInfo = CheckpointInfo
 
@@ -7036,6 +7040,8 @@ class Sandbox(google.protobuf.message.Message):
     I6PN_ENABLED_FIELD_NUMBER: builtins.int
     NETWORK_ACCESS_FIELD_NUMBER: builtins.int
     PROXY_ID_FIELD_NUMBER: builtins.int
+    ENABLE_MEMORY_SNAPSHOT_FIELD_NUMBER: builtins.int
+    SNAPSHOT_VERSION_FIELD_NUMBER: builtins.int
     @property
     def entrypoint_args(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
     @property
@@ -7072,6 +7078,12 @@ class Sandbox(google.protobuf.message.Message):
     def network_access(self) -> global___NetworkAccess:
         """Network access configuration beyond simple allow/block."""
     proxy_id: builtins.str
+    enable_memory_snapshot: builtins.bool
+    """Enable memory snapshots."""
+    snapshot_version: builtins.int
+    """Used to pin gVisor version for memory-snapshottable sandboxes.
+    This field is set by the server, not the client.
+    """
     def __init__(
         self,
         *,
@@ -7096,14 +7108,18 @@ class Sandbox(google.protobuf.message.Message):
         i6pn_enabled: builtins.bool = ...,
         network_access: global___NetworkAccess | None = ...,
         proxy_id: builtins.str | None = ...,
+        enable_memory_snapshot: builtins.bool = ...,
+        snapshot_version: builtins.int | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "_workdir", b"_workdir", "network_access", b"network_access", "open_ports", b"open_ports", "open_ports_oneof", b"open_ports_oneof", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "resources", b"resources", "scheduler_placement", b"scheduler_placement", "workdir", b"workdir"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "_workdir", b"_workdir", "block_network", b"block_network", "cloud_bucket_mounts", b"cloud_bucket_mounts", "cloud_provider", b"cloud_provider", "entrypoint_args", b"entrypoint_args", "i6pn_enabled", b"i6pn_enabled", "image_id", b"image_id", "mount_ids", b"mount_ids", "network_access", b"network_access", "nfs_mounts", b"nfs_mounts", "open_ports", b"open_ports", "open_ports_oneof", b"open_ports_oneof", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "resources", b"resources", "runtime_debug", b"runtime_debug", "s3_mounts", b"s3_mounts", "scheduler_placement", b"scheduler_placement", "secret_ids", b"secret_ids", "timeout_secs", b"timeout_secs", "volume_mounts", b"volume_mounts", "workdir", b"workdir", "worker_id", b"worker_id"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "_snapshot_version", b"_snapshot_version", "_workdir", b"_workdir", "network_access", b"network_access", "open_ports", b"open_ports", "open_ports_oneof", b"open_ports_oneof", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "resources", b"resources", "scheduler_placement", b"scheduler_placement", "snapshot_version", b"snapshot_version", "workdir", b"workdir"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_proxy_id", b"_proxy_id", "_scheduler_placement", b"_scheduler_placement", "_snapshot_version", b"_snapshot_version", "_workdir", b"_workdir", "block_network", b"block_network", "cloud_bucket_mounts", b"cloud_bucket_mounts", "cloud_provider", b"cloud_provider", "enable_memory_snapshot", b"enable_memory_snapshot", "entrypoint_args", b"entrypoint_args", "i6pn_enabled", b"i6pn_enabled", "image_id", b"image_id", "mount_ids", b"mount_ids", "network_access", b"network_access", "nfs_mounts", b"nfs_mounts", "open_ports", b"open_ports", "open_ports_oneof", b"open_ports_oneof", "proxy_id", b"proxy_id", "pty_info", b"pty_info", "resources", b"resources", "runtime_debug", b"runtime_debug", "s3_mounts", b"s3_mounts", "scheduler_placement", b"scheduler_placement", "secret_ids", b"secret_ids", "snapshot_version", b"snapshot_version", "timeout_secs", b"timeout_secs", "volume_mounts", b"volume_mounts", "workdir", b"workdir", "worker_id", b"worker_id"]) -> None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_proxy_id", b"_proxy_id"]) -> typing_extensions.Literal["proxy_id"] | None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_scheduler_placement", b"_scheduler_placement"]) -> typing_extensions.Literal["scheduler_placement"] | None: ...
     @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_snapshot_version", b"_snapshot_version"]) -> typing_extensions.Literal["snapshot_version"] | None: ...
+    @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_workdir", b"_workdir"]) -> typing_extensions.Literal["workdir"] | None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["open_ports_oneof", b"open_ports_oneof"]) -> typing_extensions.Literal["open_ports"] | None: ...
@@ -7335,6 +7351,40 @@ class SandboxListResponse(google.protobuf.message.Message):
 
 global___SandboxListResponse = SandboxListResponse
 
+class SandboxRestoreRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    APP_ID_FIELD_NUMBER: builtins.int
+    SNAPSHOT_ID_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_NAME_FIELD_NUMBER: builtins.int
+    app_id: builtins.str
+    snapshot_id: builtins.str
+    environment_name: builtins.str
+    def __init__(
+        self,
+        *,
+        app_id: builtins.str = ...,
+        snapshot_id: builtins.str = ...,
+        environment_name: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["app_id", b"app_id", "environment_name", b"environment_name", "snapshot_id", b"snapshot_id"]) -> None: ...
+
+global___SandboxRestoreRequest = SandboxRestoreRequest
+
+class SandboxRestoreResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SANDBOX_ID_FIELD_NUMBER: builtins.int
+    sandbox_id: builtins.str
+    def __init__(
+        self,
+        *,
+        sandbox_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["sandbox_id", b"sandbox_id"]) -> None: ...
+
+global___SandboxRestoreResponse = SandboxRestoreResponse
+
 class SandboxSnapshotFsRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -7376,6 +7426,67 @@ class SandboxSnapshotFsResponse(google.protobuf.message.Message):
 
 global___SandboxSnapshotFsResponse = SandboxSnapshotFsResponse
 
+class SandboxSnapshotRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SANDBOX_ID_FIELD_NUMBER: builtins.int
+    sandbox_id: builtins.str
+    def __init__(
+        self,
+        *,
+        sandbox_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["sandbox_id", b"sandbox_id"]) -> None: ...
+
+global___SandboxSnapshotRequest = SandboxSnapshotRequest
+
+class SandboxSnapshotResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SNAPSHOT_ID_FIELD_NUMBER: builtins.int
+    snapshot_id: builtins.str
+    def __init__(
+        self,
+        *,
+        snapshot_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["snapshot_id", b"snapshot_id"]) -> None: ...
+
+global___SandboxSnapshotResponse = SandboxSnapshotResponse
+
+class SandboxSnapshotWaitRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SNAPSHOT_ID_FIELD_NUMBER: builtins.int
+    TIMEOUT_FIELD_NUMBER: builtins.int
+    snapshot_id: builtins.str
+    timeout: builtins.float
+    def __init__(
+        self,
+        *,
+        snapshot_id: builtins.str = ...,
+        timeout: builtins.float = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["snapshot_id", b"snapshot_id", "timeout", b"timeout"]) -> None: ...
+
+global___SandboxSnapshotWaitRequest = SandboxSnapshotWaitRequest
+
+class SandboxSnapshotWaitResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    RESULT_FIELD_NUMBER: builtins.int
+    @property
+    def result(self) -> global___GenericResult: ...
+    def __init__(
+        self,
+        *,
+        result: global___GenericResult | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["result", b"result"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["result", b"result"]) -> None: ...
+
+global___SandboxSnapshotWaitResponse = SandboxSnapshotWaitResponse
+
 class SandboxStdinWriteRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

modal_proto/api_pb2_grpc.py

@@ -505,11 +505,26 @@ class ModalClientStub(object):
                 request_serializer=modal__proto_dot_api__pb2.SandboxListRequest.SerializeToString,
                 response_deserializer=modal__proto_dot_api__pb2.SandboxListResponse.FromString,
                 )
+        self.SandboxRestore = channel.unary_unary(
+                '/modal.client.ModalClient/SandboxRestore',
+                request_serializer=modal__proto_dot_api__pb2.SandboxRestoreRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_api__pb2.SandboxRestoreResponse.FromString,
+                )
+        self.SandboxSnapshot = channel.unary_unary(
+                '/modal.client.ModalClient/SandboxSnapshot',
+                request_serializer=modal__proto_dot_api__pb2.SandboxSnapshotRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_api__pb2.SandboxSnapshotResponse.FromString,
+                )
         self.SandboxSnapshotFs = channel.unary_unary(
                 '/modal.client.ModalClient/SandboxSnapshotFs',
                 request_serializer=modal__proto_dot_api__pb2.SandboxSnapshotFsRequest.SerializeToString,
                 response_deserializer=modal__proto_dot_api__pb2.SandboxSnapshotFsResponse.FromString,
                 )
+        self.SandboxSnapshotWait = channel.unary_unary(
+                '/modal.client.ModalClient/SandboxSnapshotWait',
+                request_serializer=modal__proto_dot_api__pb2.SandboxSnapshotWaitRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_api__pb2.SandboxSnapshotWaitResponse.FromString,
+                )
         self.SandboxStdinWrite = channel.unary_unary(
                 '/modal.client.ModalClient/SandboxStdinWrite',
                 request_serializer=modal__proto_dot_api__pb2.SandboxStdinWriteRequest.SerializeToString,
@@ -1307,12 +1322,30 @@ class ModalClientServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def SandboxRestore(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def SandboxSnapshot(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def SandboxSnapshotFs(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def SandboxSnapshotWait(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def SandboxStdinWrite(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -2035,11 +2068,26 @@ def add_ModalClientServicer_to_server(servicer, server):
                     request_deserializer=modal__proto_dot_api__pb2.SandboxListRequest.FromString,
                     response_serializer=modal__proto_dot_api__pb2.SandboxListResponse.SerializeToString,
             ),
+            'SandboxRestore': grpc.unary_unary_rpc_method_handler(
+                    servicer.SandboxRestore,
+                    request_deserializer=modal__proto_dot_api__pb2.SandboxRestoreRequest.FromString,
+                    response_serializer=modal__proto_dot_api__pb2.SandboxRestoreResponse.SerializeToString,
+            ),
+            'SandboxSnapshot': grpc.unary_unary_rpc_method_handler(
+                    servicer.SandboxSnapshot,
+                    request_deserializer=modal__proto_dot_api__pb2.SandboxSnapshotRequest.FromString,
+                    response_serializer=modal__proto_dot_api__pb2.SandboxSnapshotResponse.SerializeToString,
+            ),
             'SandboxSnapshotFs': grpc.unary_unary_rpc_method_handler(
                     servicer.SandboxSnapshotFs,
                     request_deserializer=modal__proto_dot_api__pb2.SandboxSnapshotFsRequest.FromString,
                     response_serializer=modal__proto_dot_api__pb2.SandboxSnapshotFsResponse.SerializeToString,
             ),
+            'SandboxSnapshotWait': grpc.unary_unary_rpc_method_handler(
+                    servicer.SandboxSnapshotWait,
+                    request_deserializer=modal__proto_dot_api__pb2.SandboxSnapshotWaitRequest.FromString,
+                    response_serializer=modal__proto_dot_api__pb2.SandboxSnapshotWaitResponse.SerializeToString,
+            ),
             'SandboxStdinWrite': grpc.unary_unary_rpc_method_handler(
                     servicer.SandboxStdinWrite,
                     request_deserializer=modal__proto_dot_api__pb2.SandboxStdinWriteRequest.FromString,
@@ -3901,6 +3949,40 @@ class ModalClient(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def SandboxRestore(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.client.ModalClient/SandboxRestore',
+            modal__proto_dot_api__pb2.SandboxRestoreRequest.SerializeToString,
+            modal__proto_dot_api__pb2.SandboxRestoreResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def SandboxSnapshot(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.client.ModalClient/SandboxSnapshot',
+            modal__proto_dot_api__pb2.SandboxSnapshotRequest.SerializeToString,
+            modal__proto_dot_api__pb2.SandboxSnapshotResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def SandboxSnapshotFs(request,
             target,
@@ -3918,6 +4000,23 @@ class ModalClient(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def SandboxSnapshotWait(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.client.ModalClient/SandboxSnapshotWait',
+            modal__proto_dot_api__pb2.SandboxSnapshotWaitRequest.SerializeToString,
+            modal__proto_dot_api__pb2.SandboxSnapshotWaitResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def SandboxStdinWrite(request,
             target,

modal_proto/api_pb2_grpc.pyi

@@ -421,10 +421,22 @@ class ModalClientStub:
         modal_proto.api_pb2.SandboxListRequest,
         modal_proto.api_pb2.SandboxListResponse,
     ]
+    SandboxRestore: grpc.UnaryUnaryMultiCallable[
+        modal_proto.api_pb2.SandboxRestoreRequest,
+        modal_proto.api_pb2.SandboxRestoreResponse,
+    ]
+    SandboxSnapshot: grpc.UnaryUnaryMultiCallable[
+        modal_proto.api_pb2.SandboxSnapshotRequest,
+        modal_proto.api_pb2.SandboxSnapshotResponse,
+    ]
     SandboxSnapshotFs: grpc.UnaryUnaryMultiCallable[
         modal_proto.api_pb2.SandboxSnapshotFsRequest,
         modal_proto.api_pb2.SandboxSnapshotFsResponse,
     ]
+    SandboxSnapshotWait: grpc.UnaryUnaryMultiCallable[
+        modal_proto.api_pb2.SandboxSnapshotWaitRequest,
+        modal_proto.api_pb2.SandboxSnapshotWaitResponse,
+    ]
     SandboxStdinWrite: grpc.UnaryUnaryMultiCallable[
         modal_proto.api_pb2.SandboxStdinWriteRequest,
         modal_proto.api_pb2.SandboxStdinWriteResponse,
@@ -1190,12 +1202,30 @@ class ModalClientServicer(metaclass=abc.ABCMeta):
         context: grpc.ServicerContext,
     ) -> modal_proto.api_pb2.SandboxListResponse: ...
     @abc.abstractmethod
+    def SandboxRestore(
+        self,
+        request: modal_proto.api_pb2.SandboxRestoreRequest,
+        context: grpc.ServicerContext,
+    ) -> modal_proto.api_pb2.SandboxRestoreResponse: ...
+    @abc.abstractmethod
+    def SandboxSnapshot(
+        self,
+        request: modal_proto.api_pb2.SandboxSnapshotRequest,
+        context: grpc.ServicerContext,
+    ) -> modal_proto.api_pb2.SandboxSnapshotResponse: ...
+    @abc.abstractmethod
     def SandboxSnapshotFs(
         self,
         request: modal_proto.api_pb2.SandboxSnapshotFsRequest,
         context: grpc.ServicerContext,
     ) -> modal_proto.api_pb2.SandboxSnapshotFsResponse: ...
     @abc.abstractmethod
+    def SandboxSnapshotWait(
+        self,
+        request: modal_proto.api_pb2.SandboxSnapshotWaitRequest,
+        context: grpc.ServicerContext,
+    ) -> modal_proto.api_pb2.SandboxSnapshotWaitResponse: ...
+    @abc.abstractmethod
     def SandboxStdinWrite(
         self,
         request: modal_proto.api_pb2.SandboxStdinWriteRequest,

modal_proto/modal_api_grpc.py

@@ -110,7 +110,10 @@ class ModalClientModal:
         self.SandboxGetTaskId = modal.client.UnaryUnaryWrapper(grpclib_stub.SandboxGetTaskId, client)
         self.SandboxGetTunnels = modal.client.UnaryUnaryWrapper(grpclib_stub.SandboxGetTunnels, client)
         self.SandboxList = modal.client.UnaryUnaryWrapper(grpclib_stub.SandboxList, client)
+        self.SandboxRestore = modal.client.UnaryUnaryWrapper(grpclib_stub.SandboxRestore, client)
+        self.SandboxSnapshot = modal.client.UnaryUnaryWrapper(grpclib_stub.SandboxSnapshot, client)
         self.SandboxSnapshotFs = modal.client.UnaryUnaryWrapper(grpclib_stub.SandboxSnapshotFs, client)
+        self.SandboxSnapshotWait = modal.client.UnaryUnaryWrapper(grpclib_stub.SandboxSnapshotWait, client)
         self.SandboxStdinWrite = modal.client.UnaryUnaryWrapper(grpclib_stub.SandboxStdinWrite, client)
         self.SandboxTagsSet = modal.client.UnaryUnaryWrapper(grpclib_stub.SandboxTagsSet, client)
         self.SandboxTerminate = modal.client.UnaryUnaryWrapper(grpclib_stub.SandboxTerminate, client)

modal_version/_version_generated.py

@@ -1,4 +1,4 @@
 # Copyright Modal Labs 2025
 
 # Note: Reset this value to -1 whenever you make a minor `0.X` release of the client.
-build_number = 24  # git: 8a549a2
+build_number = 26  # git: 4ec7a05