modal 1.1.1.dev17__py3-none-any.whl → 1.1.1.dev19__py3-none-any.whl

modal_proto/api_pb2.pyi

@@ -649,6 +649,25 @@ SYSTEM_ERROR_CODE_NOSPC: SystemErrorCode.ValueType  # 28
 """ENOSPC: No space left on device"""
 global___SystemErrorCode = SystemErrorCode
 
+class _TaskSnapshotBehavior:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _TaskSnapshotBehaviorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TaskSnapshotBehavior.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    TASK_SNAPSHOT_BEHAVIOR_UNSPECIFIED: _TaskSnapshotBehavior.ValueType  # 0
+    TASK_SNAPSHOT_BEHAVIOR_SNAPSHOT: _TaskSnapshotBehavior.ValueType  # 1
+    TASK_SNAPSHOT_BEHAVIOR_RESTORE: _TaskSnapshotBehavior.ValueType  # 2
+    TASK_SNAPSHOT_BEHAVIOR_NONE: _TaskSnapshotBehavior.ValueType  # 3
+
+class TaskSnapshotBehavior(_TaskSnapshotBehavior, metaclass=_TaskSnapshotBehaviorEnumTypeWrapper): ...
+
+TASK_SNAPSHOT_BEHAVIOR_UNSPECIFIED: TaskSnapshotBehavior.ValueType  # 0
+TASK_SNAPSHOT_BEHAVIOR_SNAPSHOT: TaskSnapshotBehavior.ValueType  # 1
+TASK_SNAPSHOT_BEHAVIOR_RESTORE: TaskSnapshotBehavior.ValueType  # 2
+TASK_SNAPSHOT_BEHAVIOR_NONE: TaskSnapshotBehavior.ValueType  # 3
+global___TaskSnapshotBehavior = TaskSnapshotBehavior
+
 class _TaskState:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
@@ -6770,6 +6789,42 @@ class MapAwaitResponse(google.protobuf.message.Message):
 
 global___MapAwaitResponse = MapAwaitResponse
 
+class MapCheckInputsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    LAST_ENTRY_ID_FIELD_NUMBER: builtins.int
+    TIMEOUT_FIELD_NUMBER: builtins.int
+    ATTEMPT_TOKENS_FIELD_NUMBER: builtins.int
+    last_entry_id: builtins.str
+    timeout: builtins.float
+    @property
+    def attempt_tokens(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+    def __init__(
+        self,
+        *,
+        last_entry_id: builtins.str = ...,
+        timeout: builtins.float = ...,
+        attempt_tokens: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["attempt_tokens", b"attempt_tokens", "last_entry_id", b"last_entry_id", "timeout", b"timeout"]) -> None: ...
+
+global___MapCheckInputsRequest = MapCheckInputsRequest
+
+class MapCheckInputsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    LOST_FIELD_NUMBER: builtins.int
+    @property
+    def lost(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: ...
+    def __init__(
+        self,
+        *,
+        lost: collections.abc.Iterable[builtins.bool] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["lost", b"lost"]) -> None: ...
+
+global___MapCheckInputsResponse = MapCheckInputsResponse
+
 class MapStartOrContinueItem(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -9619,6 +9674,7 @@ class TaskInfo(google.protobuf.message.Message):
     ENQUEUED_AT_FIELD_NUMBER: builtins.int
     GPU_TYPE_FIELD_NUMBER: builtins.int
     SANDBOX_ID_FIELD_NUMBER: builtins.int
+    SNAPSHOT_BEHAVIOR_FIELD_NUMBER: builtins.int
     id: builtins.str
     started_at: builtins.float
     finished_at: builtins.float
@@ -9627,6 +9683,7 @@ class TaskInfo(google.protobuf.message.Message):
     enqueued_at: builtins.float
     gpu_type: builtins.str
     sandbox_id: builtins.str
+    snapshot_behavior: global___TaskSnapshotBehavior.ValueType
     def __init__(
         self,
         *,
@@ -9637,9 +9694,10 @@ class TaskInfo(google.protobuf.message.Message):
         enqueued_at: builtins.float = ...,
         gpu_type: builtins.str = ...,
         sandbox_id: builtins.str = ...,
+        snapshot_behavior: global___TaskSnapshotBehavior.ValueType = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["result", b"result"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["enqueued_at", b"enqueued_at", "finished_at", b"finished_at", "gpu_type", b"gpu_type", "id", b"id", "result", b"result", "sandbox_id", b"sandbox_id", "started_at", b"started_at"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["enqueued_at", b"enqueued_at", "finished_at", b"finished_at", "gpu_type", b"gpu_type", "id", b"id", "result", b"result", "sandbox_id", b"sandbox_id", "snapshot_behavior", b"snapshot_behavior", "started_at", b"started_at"]) -> None: ...
 
 global___TaskInfo = TaskInfo
 

modal_proto/api_pb2_grpc.py

@@ -450,6 +450,11 @@ class ModalClientStub(object):
                 request_serializer=modal__proto_dot_api__pb2.MapAwaitRequest.SerializeToString,
                 response_deserializer=modal__proto_dot_api__pb2.MapAwaitResponse.FromString,
                 )
+        self.MapCheckInputs = channel.unary_unary(
+                '/modal.client.ModalClient/MapCheckInputs',
+                request_serializer=modal__proto_dot_api__pb2.MapCheckInputsRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_api__pb2.MapCheckInputsResponse.FromString,
+                )
         self.MapStartOrContinue = channel.unary_unary(
                 '/modal.client.ModalClient/MapStartOrContinue',
                 request_serializer=modal__proto_dot_api__pb2.MapStartOrContinueRequest.SerializeToString,
@@ -1367,6 +1372,12 @@ class ModalClientServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def MapCheckInputs(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 MapStartOrContinue(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -2262,6 +2273,11 @@ def add_ModalClientServicer_to_server(servicer, server):
                     request_deserializer=modal__proto_dot_api__pb2.MapAwaitRequest.FromString,
                     response_serializer=modal__proto_dot_api__pb2.MapAwaitResponse.SerializeToString,
             ),
+            'MapCheckInputs': grpc.unary_unary_rpc_method_handler(
+                    servicer.MapCheckInputs,
+                    request_deserializer=modal__proto_dot_api__pb2.MapCheckInputsRequest.FromString,
+                    response_serializer=modal__proto_dot_api__pb2.MapCheckInputsResponse.SerializeToString,
+            ),
             'MapStartOrContinue': grpc.unary_unary_rpc_method_handler(
                     servicer.MapStartOrContinue,
                     request_deserializer=modal__proto_dot_api__pb2.MapStartOrContinueRequest.FromString,
@@ -4121,6 +4137,23 @@ class ModalClient(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def MapCheckInputs(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/MapCheckInputs',
+            modal__proto_dot_api__pb2.MapCheckInputsRequest.SerializeToString,
+            modal__proto_dot_api__pb2.MapCheckInputsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def MapStartOrContinue(request,
             target,

modal_proto/api_pb2_grpc.pyi

@@ -379,6 +379,10 @@ class ModalClientStub:
         modal_proto.api_pb2.MapAwaitResponse,
     ]
     """Input Plane Map"""
+    MapCheckInputs: grpc.UnaryUnaryMultiCallable[
+        modal_proto.api_pb2.MapCheckInputsRequest,
+        modal_proto.api_pb2.MapCheckInputsResponse,
+    ]
     MapStartOrContinue: grpc.UnaryUnaryMultiCallable[
         modal_proto.api_pb2.MapStartOrContinueRequest,
         modal_proto.api_pb2.MapStartOrContinueResponse,
@@ -1234,6 +1238,12 @@ class ModalClientServicer(metaclass=abc.ABCMeta):
     ) -> modal_proto.api_pb2.MapAwaitResponse:
         """Input Plane Map"""
     @abc.abstractmethod
+    def MapCheckInputs(
+        self,
+        request: modal_proto.api_pb2.MapCheckInputsRequest,
+        context: grpc.ServicerContext,
+    ) -> modal_proto.api_pb2.MapCheckInputsResponse: ...
+    @abc.abstractmethod
     def MapStartOrContinue(
         self,
         request: modal_proto.api_pb2.MapStartOrContinueRequest,

modal_proto/modal_api_grpc.py

@@ -105,6 +105,7 @@ class ModalClientModal:
         self.ImageGetOrCreate = modal.client.UnaryUnaryWrapper(grpclib_stub.ImageGetOrCreate, client, server_url)
         self.ImageJoinStreaming = modal.client.UnaryStreamWrapper(grpclib_stub.ImageJoinStreaming, client, server_url)
         self.MapAwait = modal.client.UnaryUnaryWrapper(grpclib_stub.MapAwait, client, server_url)
+        self.MapCheckInputs = modal.client.UnaryUnaryWrapper(grpclib_stub.MapCheckInputs, client, server_url)
         self.MapStartOrContinue = modal.client.UnaryUnaryWrapper(grpclib_stub.MapStartOrContinue, client, server_url)
         self.MountGetOrCreate = modal.client.UnaryUnaryWrapper(grpclib_stub.MountGetOrCreate, client, server_url)
         self.MountPutFile = modal.client.UnaryUnaryWrapper(grpclib_stub.MountPutFile, client, server_url)

modal_version/__init__.py

@@ -1,4 +1,4 @@
 # Copyright Modal Labs 2025
 """Supplies the current version of the modal client library."""
 
-__version__ = "1.1.1.dev17"
+__version__ = "1.1.1.dev19"