modal 1.1.1.dev9__py3-none-any.whl → 1.1.1.dev11__py3-none-any.whl

modal_proto/api_pb2.pyi

@@ -1940,7 +1940,8 @@ class AttemptStartResponse(google.protobuf.message.Message):
     RETRY_POLICY_FIELD_NUMBER: builtins.int
     attempt_token: builtins.str
     @property
-    def retry_policy(self) -> global___FunctionRetryPolicy: ...
+    def retry_policy(self) -> global___FunctionRetryPolicy:
+        """TODO(ben-okeefe) TODO(nathan): Not currently used"""
     def __init__(
         self,
         *,
@@ -5213,6 +5214,7 @@ class FunctionGetInputsItem(google.protobuf.message.Message):
     FUNCTION_CALL_ID_FIELD_NUMBER: builtins.int
     FUNCTION_CALL_INVOCATION_TYPE_FIELD_NUMBER: builtins.int
     RETRY_COUNT_FIELD_NUMBER: builtins.int
+    FUNCTION_MAP_IDX_FIELD_NUMBER: builtins.int
     input_id: builtins.str
     @property
     def input(self) -> global___FunctionInput: ...
@@ -5220,6 +5222,8 @@ class FunctionGetInputsItem(google.protobuf.message.Message):
     function_call_id: builtins.str
     function_call_invocation_type: global___FunctionCallInvocationType.ValueType
     retry_count: builtins.int
+    function_map_idx: builtins.int
+    """intercepted and only used by the worker."""
     def __init__(
         self,
         *,
@@ -5229,9 +5233,11 @@ class FunctionGetInputsItem(google.protobuf.message.Message):
         function_call_id: builtins.str = ...,
         function_call_invocation_type: global___FunctionCallInvocationType.ValueType = ...,
         retry_count: builtins.int = ...,
+        function_map_idx: builtins.int | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["input", b"input"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["function_call_id", b"function_call_id", "function_call_invocation_type", b"function_call_invocation_type", "input", b"input", "input_id", b"input_id", "kill_switch", b"kill_switch", "retry_count", b"retry_count"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_function_map_idx", b"_function_map_idx", "function_map_idx", b"function_map_idx", "input", b"input"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_function_map_idx", b"_function_map_idx", "function_call_id", b"function_call_id", "function_call_invocation_type", b"function_call_invocation_type", "function_map_idx", b"function_map_idx", "input", b"input", "input_id", b"input_id", "kill_switch", b"kill_switch", "retry_count", b"retry_count"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_function_map_idx", b"_function_map_idx"]) -> typing_extensions.Literal["function_map_idx"] | None: ...
 
 global___FunctionGetInputsItem = FunctionGetInputsItem
 
@@ -5913,6 +5919,8 @@ class FunctionPutOutputsItem(google.protobuf.message.Message):
     OUTPUT_CREATED_AT_FIELD_NUMBER: builtins.int
     DATA_FORMAT_FIELD_NUMBER: builtins.int
     RETRY_COUNT_FIELD_NUMBER: builtins.int
+    FUNCTION_CALL_ID_FIELD_NUMBER: builtins.int
+    FUNCTION_MAP_IDX_FIELD_NUMBER: builtins.int
     input_id: builtins.str
     @property
     def result(self) -> global___GenericResult: ...
@@ -5921,6 +5929,10 @@ class FunctionPutOutputsItem(google.protobuf.message.Message):
     data_format: global___DataFormat.ValueType
     """for result.data_oneof"""
     retry_count: builtins.int
+    function_call_id: builtins.str
+    """injected by the worker"""
+    function_map_idx: builtins.int
+    """injected by the worker"""
     def __init__(
         self,
         *,
@@ -5930,9 +5942,12 @@ class FunctionPutOutputsItem(google.protobuf.message.Message):
         output_created_at: builtins.float = ...,
         data_format: global___DataFormat.ValueType = ...,
         retry_count: builtins.int = ...,
+        function_call_id: builtins.str = ...,
+        function_map_idx: builtins.int | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["result", b"result"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["data_format", b"data_format", "input_id", b"input_id", "input_started_at", b"input_started_at", "output_created_at", b"output_created_at", "result", b"result", "retry_count", b"retry_count"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_function_map_idx", b"_function_map_idx", "function_map_idx", b"function_map_idx", "result", b"result"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_function_map_idx", b"_function_map_idx", "data_format", b"data_format", "function_call_id", b"function_call_id", "function_map_idx", b"function_map_idx", "input_id", b"input_id", "input_started_at", b"input_started_at", "output_created_at", b"output_created_at", "result", b"result", "retry_count", b"retry_count"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_function_map_idx", b"_function_map_idx"]) -> typing_extensions.Literal["function_map_idx"] | None: ...
 
 global___FunctionPutOutputsItem = FunctionPutOutputsItem
 
@@ -6668,6 +6683,126 @@ class InputInfo(google.protobuf.message.Message):
 
 global___InputInfo = InputInfo
 
+class MapAwaitRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FUNCTION_CALL_ID_FIELD_NUMBER: builtins.int
+    LAST_ENTRY_ID_FIELD_NUMBER: builtins.int
+    REQUESTED_AT_FIELD_NUMBER: builtins.int
+    TIMEOUT_FIELD_NUMBER: builtins.int
+    function_call_id: builtins.str
+    last_entry_id: builtins.str
+    requested_at: builtins.float
+    """Used for waypoints."""
+    timeout: builtins.float
+    def __init__(
+        self,
+        *,
+        function_call_id: builtins.str = ...,
+        last_entry_id: builtins.str = ...,
+        requested_at: builtins.float = ...,
+        timeout: builtins.float = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["function_call_id", b"function_call_id", "last_entry_id", b"last_entry_id", "requested_at", b"requested_at", "timeout", b"timeout"]) -> None: ...
+
+global___MapAwaitRequest = MapAwaitRequest
+
+class MapAwaitResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    OUTPUTS_FIELD_NUMBER: builtins.int
+    LAST_ENTRY_ID_FIELD_NUMBER: builtins.int
+    @property
+    def outputs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___FunctionGetOutputsItem]: ...
+    last_entry_id: builtins.str
+    def __init__(
+        self,
+        *,
+        outputs: collections.abc.Iterable[global___FunctionGetOutputsItem] | None = ...,
+        last_entry_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["last_entry_id", b"last_entry_id", "outputs", b"outputs"]) -> None: ...
+
+global___MapAwaitResponse = MapAwaitResponse
+
+class MapStartOrContinueItem(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    INPUT_FIELD_NUMBER: builtins.int
+    ATTEMPT_TOKEN_FIELD_NUMBER: builtins.int
+    @property
+    def input(self) -> global___FunctionPutInputsItem: ...
+    attempt_token: builtins.str
+    """None if this is a fresh input, otherwise it is the attempt token for a retry."""
+    def __init__(
+        self,
+        *,
+        input: global___FunctionPutInputsItem | None = ...,
+        attempt_token: builtins.str | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_attempt_token", b"_attempt_token", "attempt_token", b"attempt_token", "input", b"input"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_attempt_token", b"_attempt_token", "attempt_token", b"attempt_token", "input", b"input"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_attempt_token", b"_attempt_token"]) -> typing_extensions.Literal["attempt_token"] | None: ...
+
+global___MapStartOrContinueItem = MapStartOrContinueItem
+
+class MapStartOrContinueRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FUNCTION_ID_FIELD_NUMBER: builtins.int
+    PARENT_INPUT_ID_FIELD_NUMBER: builtins.int
+    FUNCTION_CALL_ID_FIELD_NUMBER: builtins.int
+    ITEMS_FIELD_NUMBER: builtins.int
+    function_id: builtins.str
+    parent_input_id: builtins.str
+    function_call_id: builtins.str
+    """This is given iff this is a map continue request."""
+    @property
+    def items(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___MapStartOrContinueItem]: ...
+    def __init__(
+        self,
+        *,
+        function_id: builtins.str = ...,
+        parent_input_id: builtins.str = ...,
+        function_call_id: builtins.str | None = ...,
+        items: collections.abc.Iterable[global___MapStartOrContinueItem] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_function_call_id", b"_function_call_id", "function_call_id", b"function_call_id"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_function_call_id", b"_function_call_id", "function_call_id", b"function_call_id", "function_id", b"function_id", "items", b"items", "parent_input_id", b"parent_input_id"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_function_call_id", b"_function_call_id"]) -> typing_extensions.Literal["function_call_id"] | None: ...
+
+global___MapStartOrContinueRequest = MapStartOrContinueRequest
+
+class MapStartOrContinueResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FUNCTION_ID_FIELD_NUMBER: builtins.int
+    FUNCTION_CALL_ID_FIELD_NUMBER: builtins.int
+    MAX_INPUTS_OUTSTANDING_FIELD_NUMBER: builtins.int
+    ATTEMPT_TOKENS_FIELD_NUMBER: builtins.int
+    RETRY_POLICY_FIELD_NUMBER: builtins.int
+    function_id: builtins.str
+    function_call_id: builtins.str
+    max_inputs_outstanding: builtins.int
+    @property
+    def attempt_tokens(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+    @property
+    def retry_policy(self) -> global___FunctionRetryPolicy:
+        """TODO(ben-okeefe): Not currently used"""
+    def __init__(
+        self,
+        *,
+        function_id: builtins.str = ...,
+        function_call_id: builtins.str = ...,
+        max_inputs_outstanding: builtins.int = ...,
+        attempt_tokens: collections.abc.Iterable[builtins.str] | None = ...,
+        retry_policy: global___FunctionRetryPolicy | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["retry_policy", b"retry_policy"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["attempt_tokens", b"attempt_tokens", "function_call_id", b"function_call_id", "function_id", b"function_id", "max_inputs_outstanding", b"max_inputs_outstanding", "retry_policy", b"retry_policy"]) -> None: ...
+
+global___MapStartOrContinueResponse = MapStartOrContinueResponse
+
 class MethodDefinition(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

modal_proto/api_pb2_grpc.py

@@ -445,6 +445,16 @@ class ModalClientStub(object):
                 request_serializer=modal__proto_dot_api__pb2.ImageJoinStreamingRequest.SerializeToString,
                 response_deserializer=modal__proto_dot_api__pb2.ImageJoinStreamingResponse.FromString,
                 )
+        self.MapAwait = channel.unary_unary(
+                '/modal.client.ModalClient/MapAwait',
+                request_serializer=modal__proto_dot_api__pb2.MapAwaitRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_api__pb2.MapAwaitResponse.FromString,
+                )
+        self.MapStartOrContinue = channel.unary_unary(
+                '/modal.client.ModalClient/MapStartOrContinue',
+                request_serializer=modal__proto_dot_api__pb2.MapStartOrContinueRequest.SerializeToString,
+                response_deserializer=modal__proto_dot_api__pb2.MapStartOrContinueResponse.FromString,
+                )
         self.MountGetOrCreate = channel.unary_unary(
                 '/modal.client.ModalClient/MountGetOrCreate',
                 request_serializer=modal__proto_dot_api__pb2.MountGetOrCreateRequest.SerializeToString,
@@ -1350,6 +1360,19 @@ class ModalClientServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def MapAwait(self, request, context):
+        """Input Plane Map
+        """
+        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)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def MountGetOrCreate(self, request, context):
         """Mounts
         """
@@ -2234,6 +2257,16 @@ def add_ModalClientServicer_to_server(servicer, server):
                     request_deserializer=modal__proto_dot_api__pb2.ImageJoinStreamingRequest.FromString,
                     response_serializer=modal__proto_dot_api__pb2.ImageJoinStreamingResponse.SerializeToString,
             ),
+            'MapAwait': grpc.unary_unary_rpc_method_handler(
+                    servicer.MapAwait,
+                    request_deserializer=modal__proto_dot_api__pb2.MapAwaitRequest.FromString,
+                    response_serializer=modal__proto_dot_api__pb2.MapAwaitResponse.SerializeToString,
+            ),
+            'MapStartOrContinue': grpc.unary_unary_rpc_method_handler(
+                    servicer.MapStartOrContinue,
+                    request_deserializer=modal__proto_dot_api__pb2.MapStartOrContinueRequest.FromString,
+                    response_serializer=modal__proto_dot_api__pb2.MapStartOrContinueResponse.SerializeToString,
+            ),
             'MountGetOrCreate': grpc.unary_unary_rpc_method_handler(
                     servicer.MountGetOrCreate,
                     request_deserializer=modal__proto_dot_api__pb2.MountGetOrCreateRequest.FromString,
@@ -4071,6 +4104,40 @@ class ModalClient(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def MapAwait(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/MapAwait',
+            modal__proto_dot_api__pb2.MapAwaitRequest.SerializeToString,
+            modal__proto_dot_api__pb2.MapAwaitResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def MapStartOrContinue(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/MapStartOrContinue',
+            modal__proto_dot_api__pb2.MapStartOrContinueRequest.SerializeToString,
+            modal__proto_dot_api__pb2.MapStartOrContinueResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def MountGetOrCreate(request,
             target,

modal_proto/api_pb2_grpc.pyi

@@ -374,6 +374,15 @@ class ModalClientStub:
         modal_proto.api_pb2.ImageJoinStreamingRequest,
         modal_proto.api_pb2.ImageJoinStreamingResponse,
     ]
+    MapAwait: grpc.UnaryUnaryMultiCallable[
+        modal_proto.api_pb2.MapAwaitRequest,
+        modal_proto.api_pb2.MapAwaitResponse,
+    ]
+    """Input Plane Map"""
+    MapStartOrContinue: grpc.UnaryUnaryMultiCallable[
+        modal_proto.api_pb2.MapStartOrContinueRequest,
+        modal_proto.api_pb2.MapStartOrContinueResponse,
+    ]
     MountGetOrCreate: grpc.UnaryUnaryMultiCallable[
         modal_proto.api_pb2.MountGetOrCreateRequest,
         modal_proto.api_pb2.MountGetOrCreateResponse,
@@ -1218,6 +1227,19 @@ class ModalClientServicer(metaclass=abc.ABCMeta):
         context: grpc.ServicerContext,
     ) -> collections.abc.Iterator[modal_proto.api_pb2.ImageJoinStreamingResponse]: ...
     @abc.abstractmethod
+    def MapAwait(
+        self,
+        request: modal_proto.api_pb2.MapAwaitRequest,
+        context: grpc.ServicerContext,
+    ) -> modal_proto.api_pb2.MapAwaitResponse:
+        """Input Plane Map"""
+    @abc.abstractmethod
+    def MapStartOrContinue(
+        self,
+        request: modal_proto.api_pb2.MapStartOrContinueRequest,
+        context: grpc.ServicerContext,
+    ) -> modal_proto.api_pb2.MapStartOrContinueResponse: ...
+    @abc.abstractmethod
     def MountGetOrCreate(
         self,
         request: modal_proto.api_pb2.MountGetOrCreateRequest,

modal_proto/modal_api_grpc.py

@@ -104,6 +104,8 @@ class ModalClientModal:
         self.ImageFromId = modal.client.UnaryUnaryWrapper(grpclib_stub.ImageFromId, client, server_url)
         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.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)
         self.NotebookKernelPublishResults = modal.client.UnaryUnaryWrapper(grpclib_stub.NotebookKernelPublishResults, 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.dev9"
+__version__ = "1.1.1.dev11"