modal 0.74.14__py3-none-any.whl → 0.74.16__py3-none-any.whl

modal/volume.pyi

@@ -1,6 +1,8 @@
+import _io
 import asyncio.locks
 import collections.abc
 import enum
+import google.protobuf.message
 import modal._object
 import modal._utils.blob_utils
 import modal.client
@@ -36,6 +38,7 @@ class FileEntry:
 
 class _Volume(modal._object._Object):
     _lock: typing.Optional[asyncio.locks.Lock]
+    _metadata: typing.Optional[modal_proto.api_pb2.VolumeMetadata]
 
     async def _get_lock(self): ...
     @staticmethod
@@ -46,6 +49,8 @@ class _Volume(modal._object._Object):
         create_if_missing: bool = False,
         version: typing.Optional[int] = None,
     ) -> _Volume: ...
+    def _hydrate_metadata(self, metadata: typing.Optional[google.protobuf.message.Message]): ...
+    def _get_metadata(self) -> typing.Optional[google.protobuf.message.Message]: ...
     @classmethod
     def ephemeral(
         cls: type[_Volume],
@@ -80,7 +85,7 @@ class _Volume(modal._object._Object):
     async def read_file_into_fileobj(self, path: str, fileobj: typing.IO[bytes]) -> int: ...
     async def remove_file(self, path: str, recursive: bool = False) -> None: ...
     async def copy_files(self, src_paths: collections.abc.Sequence[str], dst_path: str) -> None: ...
-    async def batch_upload(self, force: bool = False) -> _VolumeUploadContextManager: ...
+    async def batch_upload(self, force: bool = False) -> _AbstractVolumeUploadContextManager: ...
     async def _instance_delete(self): ...
     @staticmethod
     async def delete(
@@ -95,44 +100,11 @@ class _Volume(modal._object._Object):
         environment_name: typing.Optional[str] = None,
     ): ...
 
-class _VolumeUploadContextManager:
-    _volume_id: str
-    _client: modal.client._Client
-    _force: bool
-    progress_cb: collections.abc.Callable[..., typing.Any]
-    _upload_generators: list[
-        collections.abc.Generator[collections.abc.Callable[[], modal._utils.blob_utils.FileUploadSpec], None, None]
-    ]
-
-    def __init__(
-        self,
-        volume_id: str,
-        client: modal.client._Client,
-        progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
-        force: bool = False,
-    ): ...
-    async def __aenter__(self): ...
-    async def __aexit__(self, exc_type, exc_val, exc_tb): ...
-    def put_file(
-        self,
-        local_file: typing.Union[pathlib.Path, str, typing.BinaryIO],
-        remote_path: typing.Union[pathlib.PurePosixPath, str],
-        mode: typing.Optional[int] = None,
-    ): ...
-    def put_directory(
-        self,
-        local_path: typing.Union[pathlib.Path, str],
-        remote_path: typing.Union[pathlib.PurePosixPath, str],
-        recursive: bool = True,
-    ): ...
-    async def _upload_file(
-        self, file_spec: modal._utils.blob_utils.FileUploadSpec
-    ) -> modal_proto.api_pb2.MountFile: ...
-
 SUPERSELF = typing.TypeVar("SUPERSELF", covariant=True)
 
 class Volume(modal.object.Object):
     _lock: typing.Optional[asyncio.locks.Lock]
+    _metadata: typing.Optional[modal_proto.api_pb2.VolumeMetadata]
 
     def __init__(self, *args, **kwargs): ...
 
@@ -150,6 +122,8 @@ class Volume(modal.object.Object):
         create_if_missing: bool = False,
         version: typing.Optional[int] = None,
     ) -> Volume: ...
+    def _hydrate_metadata(self, metadata: typing.Optional[google.protobuf.message.Message]): ...
+    def _get_metadata(self) -> typing.Optional[google.protobuf.message.Message]: ...
     @classmethod
     def ephemeral(
         cls: type[Volume],
@@ -256,8 +230,8 @@ class Volume(modal.object.Object):
     copy_files: __copy_files_spec[typing_extensions.Self]
 
     class __batch_upload_spec(typing_extensions.Protocol[SUPERSELF]):
-        def __call__(self, force: bool = False) -> VolumeUploadContextManager: ...
-        async def aio(self, force: bool = False) -> VolumeUploadContextManager: ...
+        def __call__(self, force: bool = False) -> AbstractVolumeUploadContextManager: ...
+        async def aio(self, force: bool = False) -> AbstractVolumeUploadContextManager: ...
 
     batch_upload: __batch_upload_spec[typing_extensions.Self]
 
@@ -303,7 +277,92 @@ class Volume(modal.object.Object):
 
     rename: __rename_spec
 
-class VolumeUploadContextManager:
+class _AbstractVolumeUploadContextManager:
+    async def __aenter__(self): ...
+    async def __aexit__(self, exc_type, exc_val, exc_tb): ...
+    def put_file(
+        self,
+        local_file: typing.Union[pathlib.Path, str, typing.BinaryIO, _io.BytesIO],
+        remote_path: typing.Union[pathlib.PurePosixPath, str],
+        mode: typing.Optional[int] = None,
+    ): ...
+    def put_directory(
+        self,
+        local_path: typing.Union[pathlib.Path, str],
+        remote_path: typing.Union[pathlib.PurePosixPath, str],
+        recursive: bool = True,
+    ): ...
+    @staticmethod
+    def resolve(
+        version: int,
+        object_id: str,
+        client,
+        progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
+        force: bool = False,
+    ) -> _AbstractVolumeUploadContextManager: ...
+
+class AbstractVolumeUploadContextManager:
+    def __init__(self, /, *args, **kwargs): ...
+    def __enter__(self): ...
+    async def __aenter__(self): ...
+    def __exit__(self, exc_type, exc_val, exc_tb): ...
+    async def __aexit__(self, exc_type, exc_val, exc_tb): ...
+    def put_file(
+        self,
+        local_file: typing.Union[pathlib.Path, str, typing.BinaryIO, _io.BytesIO],
+        remote_path: typing.Union[pathlib.PurePosixPath, str],
+        mode: typing.Optional[int] = None,
+    ): ...
+    def put_directory(
+        self,
+        local_path: typing.Union[pathlib.Path, str],
+        remote_path: typing.Union[pathlib.PurePosixPath, str],
+        recursive: bool = True,
+    ): ...
+    @staticmethod
+    def resolve(
+        version: int,
+        object_id: str,
+        client,
+        progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
+        force: bool = False,
+    ) -> AbstractVolumeUploadContextManager: ...
+
+class _VolumeUploadContextManager(_AbstractVolumeUploadContextManager):
+    _volume_id: str
+    _client: modal.client._Client
+    _force: bool
+    progress_cb: collections.abc.Callable[..., typing.Any]
+    _upload_generators: list[
+        collections.abc.Generator[collections.abc.Callable[[], modal._utils.blob_utils.FileUploadSpec], None, None]
+    ]
+
+    def __init__(
+        self,
+        volume_id: str,
+        client: modal.client._Client,
+        progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
+        force: bool = False,
+    ): ...
+    async def __aenter__(self): ...
+    async def __aexit__(self, exc_type, exc_val, exc_tb): ...
+    def put_file(
+        self,
+        local_file: typing.Union[pathlib.Path, str, typing.BinaryIO, _io.BytesIO],
+        remote_path: typing.Union[pathlib.PurePosixPath, str],
+        mode: typing.Optional[int] = None,
+    ): ...
+    def put_directory(
+        self,
+        local_path: typing.Union[pathlib.Path, str],
+        remote_path: typing.Union[pathlib.PurePosixPath, str],
+        recursive: bool = True,
+    ): ...
+    async def _upload_file(
+        self, file_spec: modal._utils.blob_utils.FileUploadSpec
+    ) -> modal_proto.api_pb2.MountFile: ...
+
+class VolumeUploadContextManager(AbstractVolumeUploadContextManager):
     _volume_id: str
     _client: modal.client.Client
     _force: bool
@@ -325,7 +384,7 @@ class VolumeUploadContextManager:
     async def __aexit__(self, exc_type, exc_val, exc_tb): ...
     def put_file(
         self,
-        local_file: typing.Union[pathlib.Path, str, typing.BinaryIO],
+        local_file: typing.Union[pathlib.Path, str, typing.BinaryIO, _io.BytesIO],
         remote_path: typing.Union[pathlib.PurePosixPath, str],
         mode: typing.Optional[int] = None,
     ): ...
@@ -342,4 +401,85 @@ class VolumeUploadContextManager:
 
     _upload_file: ___upload_file_spec[typing_extensions.Self]
 
+class _VolumeUploadContextManager2(_AbstractVolumeUploadContextManager):
+    _volume_id: str
+    _client: modal.client._Client
+    _force: bool
+    _progress_cb: collections.abc.Callable[..., typing.Any]
+    _uploader_generators: list[
+        collections.abc.Generator[
+            collections.abc.Callable[[], typing.Awaitable[modal._utils.blob_utils.FileUploadSpec2]]
+        ]
+    ]
+
+    def __init__(
+        self,
+        volume_id: str,
+        client: modal.client._Client,
+        progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
+        force: bool = False,
+    ): ...
+    async def __aenter__(self): ...
+    async def __aexit__(self, exc_type, exc_val, exc_tb): ...
+    def put_file(
+        self,
+        local_file: typing.Union[pathlib.Path, str, typing.BinaryIO, _io.BytesIO],
+        remote_path: typing.Union[pathlib.PurePosixPath, str],
+        mode: typing.Optional[int] = None,
+    ): ...
+    def put_directory(
+        self,
+        local_path: typing.Union[pathlib.Path, str],
+        remote_path: typing.Union[pathlib.PurePosixPath, str],
+        recursive: bool = True,
+    ): ...
+    async def _put_file_specs(self, file_specs: list[modal._utils.blob_utils.FileUploadSpec2]): ...
+
+class VolumeUploadContextManager2(AbstractVolumeUploadContextManager):
+    _volume_id: str
+    _client: modal.client.Client
+    _force: bool
+    _progress_cb: collections.abc.Callable[..., typing.Any]
+    _uploader_generators: list[
+        collections.abc.Generator[
+            collections.abc.Callable[[], typing.Awaitable[modal._utils.blob_utils.FileUploadSpec2]]
+        ]
+    ]
+
+    def __init__(
+        self,
+        volume_id: str,
+        client: modal.client.Client,
+        progress_cb: typing.Optional[collections.abc.Callable[..., typing.Any]] = None,
+        force: bool = False,
+    ): ...
+    def __enter__(self): ...
+    async def __aenter__(self): ...
+    def __exit__(self, exc_type, exc_val, exc_tb): ...
+    async def __aexit__(self, exc_type, exc_val, exc_tb): ...
+    def put_file(
+        self,
+        local_file: typing.Union[pathlib.Path, str, typing.BinaryIO, _io.BytesIO],
+        remote_path: typing.Union[pathlib.PurePosixPath, str],
+        mode: typing.Optional[int] = None,
+    ): ...
+    def put_directory(
+        self,
+        local_path: typing.Union[pathlib.Path, str],
+        remote_path: typing.Union[pathlib.PurePosixPath, str],
+        recursive: bool = True,
+    ): ...
+
+    class ___put_file_specs_spec(typing_extensions.Protocol[SUPERSELF]):
+        def __call__(self, file_specs: list[modal._utils.blob_utils.FileUploadSpec2]): ...
+        async def aio(self, file_specs: list[modal._utils.blob_utils.FileUploadSpec2]): ...
+
+    _put_file_specs: ___put_file_specs_spec[typing_extensions.Self]
+
+async def _put_missing_blocks(
+    file_specs: list[modal._utils.blob_utils.FileUploadSpec2],
+    missing_blocks: list,
+    put_responses: dict[bytes, bytes],
+    progress_cb: collections.abc.Callable[..., typing.Any],
+): ...
 def _open_files_error_annotation(mount_path: str) -> typing.Optional[str]: ...

{modal-0.74.14.dist-info → modal-0.74.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: modal
-Version: 0.74.14
+Version: 0.74.16
 Summary: Python client library for Modal
 Author-email: Modal Labs <support@modal.com>
 License: Apache-2.0

{modal-0.74.14.dist-info → modal-0.74.16.dist-info}/RECORD

@@ -22,7 +22,7 @@ modal/app.py,sha256=1GSPMMYRkG17FQwnA_PleK9YQrPQWXlN_4mpZ5VRA4Q,48468
 modal/app.pyi,sha256=bpC9uN_B4d_UtyXVuhGpXC2RM-IcsytS0ndu-GoKrHQ,27276
 modal/call_graph.py,sha256=1g2DGcMIJvRy-xKicuf63IVE98gJSnQsr8R_NVMptNc,2581
 modal/client.py,sha256=U-YKSw0n7J1ZLREt9cbEJCtmHe5YoPKFxl0xlkan2yc,15565
-modal/client.pyi,sha256=K1ooS9wvZZk-ldqnYjZXswq7NjJ23MXo6DZkX3OJtnA,7593
+modal/client.pyi,sha256=MXDCG1c77_KDP7R66sApXRKz9ZnhLsZMjSETzBqGDzA,7593
 modal/cloud_bucket_mount.py,sha256=YOe9nnvSr4ZbeCn587d7_VhE9IioZYRvF9VYQTQux08,5914
 modal/cloud_bucket_mount.pyi,sha256=30T3K1a89l6wzmEJ_J9iWv9SknoGqaZDx59Xs-ZQcmk,1607
 modal/cls.py,sha256=GvaNl8R5UsH7Vg88WEOyerdjvZEPK7xxi3nqHlyOW_c,33497
@@ -78,8 +78,8 @@ modal/snapshot.pyi,sha256=Ypd4NKsjOTnnnqXyTGGLKq5lkocRrUURYjY5Pi67_qA,670
 modal/stream_type.py,sha256=A6320qoAAWhEfwOCZfGtymQTu5AfLfJXXgARqooTPvY,417
 modal/token_flow.py,sha256=APNPQhuqy2avu6IY26U6OiN7erRhpo03Ztt1V60_wts,6776
 modal/token_flow.pyi,sha256=0XV3d-9CGQL3qjPdw3RgwIFVqqxo8Z-u044_mkgAM3o,2064
-modal/volume.py,sha256=JAWeDvoAG95tMBv-fYIERyHsJPS_X_xGpxRRmYtb6j0,30096
-modal/volume.pyi,sha256=kTsXarphjZILXci84LQy7EyC84eXUs5-7D62IM5q3eE,12491
+modal/volume.py,sha256=3c5_aJNJtgpsFRZWBjc0jwn8Zs0jo9V6UDmh6ifrbdA,40145
+modal/volume.pyi,sha256=juOVWGlgz7IeOY4M7jBhbeNRPA9xdGUwvA3AzlZUscQ,17958
 modal/_runtime/__init__.py,sha256=MIEP8jhXUeGq_eCjYFcqN5b1bxBM4fdk0VESpjWR0fc,28
 modal/_runtime/asgi.py,sha256=KNarxvZI9z8fnmZl2vbkWTjnoLXs9kqOahkrbsTLkyc,22429
 modal/_runtime/container_io_manager.py,sha256=0yNO3HTVIM4f338rxJavD8nrRN7KhDpjz1jLux71MRY,43842
@@ -92,7 +92,7 @@ modal/_runtime/user_code_imports.py,sha256=kAv37Pl1TmGKduv0Kozum0xNTD42bDLloSIsT
 modal/_utils/__init__.py,sha256=waLjl5c6IPDhSsdWAm9Bji4e2PVxamYABKAze6CHVXY,28
 modal/_utils/app_utils.py,sha256=88BT4TPLWfYAQwKTHcyzNQRHg8n9B-QE2UyJs96iV-0,108
 modal/_utils/async_utils.py,sha256=b2TJyKY1Hq7df7M-fo3qlFM95mGdo3dCuqRPPcV5hsE,27445
-modal/_utils/blob_utils.py,sha256=jWJovk4g-YNG3CvkvglOds4a6D1M0Tcal_59v7y9VsM,14591
+modal/_utils/blob_utils.py,sha256=WhWaSFLcffVDchMyfKys-j0EFVbm9l7rebzeoa7Z2jM,18214
 modal/_utils/bytes_io_segment_payload.py,sha256=uunxVJS4PE1LojF_UpURMzVK9GuvmYWRqQo_bxEj5TU,3385
 modal/_utils/deprecation.py,sha256=EXP1beU4pmEqEzWMLw6E3kUfNfpmNA_VOp6i0EHi93g,4856
 modal/_utils/docker_utils.py,sha256=h1uETghR40mp_y3fSWuZAfbIASH1HMzuphJHghAL6DU,3722
@@ -132,7 +132,7 @@ modal/cli/run.py,sha256=NX2wWwj8HD6XUhnZRF808Qy9eeouv8KnvyOP57HqIXI,23637
 modal/cli/secret.py,sha256=WB_c-LE9-eDqleLpJxsJ9rZw62Eeza8ZFQFR10vNMEk,4197
 modal/cli/token.py,sha256=mxSgOWakXG6N71hQb1ko61XAR9ZGkTMZD-Txn7gmTac,1924
 modal/cli/utils.py,sha256=hZmjyzcPjDnQSkLvycZD2LhGdcsfdZshs_rOU78EpvI,3717
-modal/cli/volume.py,sha256=c2IuVNO2yJVaXmZkRh3xwQmznlRTgFoJr_BIzzqtVv0,10251
+modal/cli/volume.py,sha256=_PJ5Vn_prkLk_x1Lksx4kZySlKWqIn36T2Edd1-h7Mg,10497
 modal/cli/programs/__init__.py,sha256=svYKtV8HDwDCN86zbdWqyq5T8sMdGDj0PVlzc2tIxDM,28
 modal/cli/programs/run_jupyter.py,sha256=YVvJYu927A4ji72d6i27CKfyZ_uDWteeittARtJnf7E,2775
 modal/cli/programs/vscode.py,sha256=kfvhZQ4bJwtVm3MgC1V7AlygZOlKT1a33alr_uwrewA,3473
@@ -145,7 +145,7 @@ modal/requirements/2024.10.txt,sha256=qD-5cVIVM9wXesJ6JC89Ew-3m2KjEElUz3jaw_MddR
 modal/requirements/PREVIEW.txt,sha256=qD-5cVIVM9wXesJ6JC89Ew-3m2KjEElUz3jaw_MddRo,296
 modal/requirements/README.md,sha256=9tK76KP0Uph7O0M5oUgsSwEZDj5y-dcUPsnpR0Sc-Ik,854
 modal/requirements/base-images.json,sha256=57vMSqzMbLBxw5tFWSaMiIkkVEps4JfX5PAtXGnkS4U,740
-modal-0.74.14.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
+modal-0.74.16.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
 modal_docs/__init__.py,sha256=svYKtV8HDwDCN86zbdWqyq5T8sMdGDj0PVlzc2tIxDM,28
 modal_docs/gen_cli_docs.py,sha256=c1yfBS_x--gL5bs0N4ihMwqwX8l3IBWSkBAKNNIi6bQ,3801
 modal_docs/gen_reference_docs.py,sha256=cvTgltucqYLLIX84QxAwf51Z5Vc2n6cLxS8VcrxNCAo,6401
@@ -153,13 +153,13 @@ modal_docs/mdmd/__init__.py,sha256=svYKtV8HDwDCN86zbdWqyq5T8sMdGDj0PVlzc2tIxDM,2
 modal_docs/mdmd/mdmd.py,sha256=Irx49MCCTlBOP4FBdLR--JrpA3-WhsVeriq0LGgsRic,6232
 modal_docs/mdmd/signatures.py,sha256=XJaZrK7Mdepk5fdX51A8uENiLFNil85Ud0d4MH8H5f0,3218
 modal_proto/__init__.py,sha256=MIEP8jhXUeGq_eCjYFcqN5b1bxBM4fdk0VESpjWR0fc,28
-modal_proto/api.proto,sha256=kovn2o64lk57QaRxqWPgFMuiR1ceZrrXw_U3B0DNuPQ,93026
-modal_proto/api_grpc.py,sha256=DlrVx10B6HLy4LjWrgK7HJ7kUNDKqEvm6mUqh-6EVvs,112161
-modal_proto/api_pb2.py,sha256=4GDk37vbGyrbdupsLVOojyEIFfqWyYptaYO9pLzM1ec,328101
-modal_proto/api_pb2.pyi,sha256=isSiMw7B3hyz-OeVS9xK6Jnk3TaEVea_aAAx4T7aW_g,446747
-modal_proto/api_pb2_grpc.py,sha256=NQPpAa3Co-zFNB6L6BQoZjf-RaOlWFFPX8chQyGHQEo,242401
-modal_proto/api_pb2_grpc.pyi,sha256=zK2Ms2Xws02cC9uy5U5A5sPULXx4T6A9o9BONqFcipg,56456
-modal_proto/modal_api_grpc.py,sha256=K_GsrHIw2CayKh_nyuYAblQW_tOijUNW267Hs9vWUWU,14948
+modal_proto/api.proto,sha256=eD4A8icYJFeunpUiDqra14EmUnosttBtDnrrfdxGeBw,93783
+modal_proto/api_grpc.py,sha256=LZ_rylBFJ79c5-dYOcyebPjRGrCKNO3j2K7giw9P9-g,113653
+modal_proto/api_pb2.py,sha256=NEV1ZmTOfpBuCgJ4F6eSZKQkBHXPaHowdJtCs0OCek0,330916
+modal_proto/api_pb2.pyi,sha256=JP8s6JHac1Cz2kpJaBNk6K_bPoel1j50uJYco6lV7Co,450054
+modal_proto/api_pb2_grpc.py,sha256=QaApxS-KExaYpQU0kdnoxXPrERZk_b4lRZKY_V4mbq0,245789
+modal_proto/api_pb2_grpc.pyi,sha256=6obh-mFStQnsXcUHzTI_SXuED9mhwcKnfIt_urSZkJw,57552
+modal_proto/modal_api_grpc.py,sha256=r7qGJdW69v5pXca2ySm6gqTGYG_pFPWBno_4XGMkbpU,15136
 modal_proto/modal_options_grpc.py,sha256=qJ1cuwA54oRqrdTyPTbvfhFZYd9HhJKK5UCwt523r3Y,120
 modal_proto/options.proto,sha256=zp9h5r61ivsp0XwEWwNBsVqNTbRA1VSY_UtN7sEcHtE,549
 modal_proto/options_grpc.py,sha256=M18X3d-8F_cNYSVM3I25dUTO5rZ0rd-vCCfynfh13Nc,125
@@ -170,9 +170,9 @@ modal_proto/options_pb2_grpc.pyi,sha256=CImmhxHsYnF09iENPoe8S4J-n93jtgUYD2JPAc0y
 modal_proto/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 modal_version/__init__.py,sha256=m94xZNWIjH8oUtJk4l9xfovzDJede2o7X-q0MHVECtM,470
 modal_version/__main__.py,sha256=2FO0yYQQwDTh6udt1h-cBnGd1c4ZyHnHSI4BksxzVac,105
-modal_version/_version_generated.py,sha256=uuUINVrRoOHnG0345kZM29EhW_7y0-WJxEmNLI4sO_Q,149
-modal-0.74.14.dist-info/METADATA,sha256=o3yU8T6X_kHWR581mpmsux-kk2_QtQGrGqNL7Kct9jU,2474
-modal-0.74.14.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-modal-0.74.14.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
-modal-0.74.14.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
-modal-0.74.14.dist-info/RECORD,,
+modal_version/_version_generated.py,sha256=iw6S4gIMMk_kxXdUThejt0bL01XM4XObW5lyo2KKxMA,149
+modal-0.74.16.dist-info/METADATA,sha256=SOCGPNZKh82g8EnIIGaDKd-PefSR4ucoYk5mS7JCyoE,2474
+modal-0.74.16.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+modal-0.74.16.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
+modal-0.74.16.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
+modal-0.74.16.dist-info/RECORD,,

modal_proto/api.proto

@@ -574,6 +574,27 @@ message Asgi {
   }
 }
 
+message AttemptAwaitRequest {
+  string attempt_token = 1;
+  double requested_at = 2; // Used for waypoints.
+  float timeout_secs = 3;
+}
+
+message AttemptAwaitResponse {
+  optional FunctionGetOutputsItem output = 1;
+}
+
+message AttemptStartRequest {
+  string function_id = 1;
+  string parent_input_id = 2;
+  FunctionPutInputsItem input = 3;
+}
+
+message AttemptStartResponse {
+  string attempt_token = 1;
+  FunctionRetryPolicy retry_policy = 2;
+}
+
 message AutoscalerSettings {
   // A collection of user-configurable settings for Function autoscaling
   // These are used for static configuration and for dynamic autoscaler updates
@@ -3085,6 +3106,12 @@ service ModalClient {
   rpc AppSetObjects(AppSetObjectsRequest) returns (google.protobuf.Empty);
   rpc AppStop(AppStopRequest) returns (google.protobuf.Empty);
 
+  // Input Plane
+  // These RPCs are experimental, not deployed to production, and can be changed / removed
+  // without needing to worry about backwards compatibility.
+  rpc AttemptAwait(AttemptAwaitRequest) returns (AttemptAwaitResponse);
+  rpc AttemptStart(AttemptStartRequest) returns (AttemptStartResponse);
+
   // Blobs
   rpc BlobCreate(BlobCreateRequest) returns (BlobCreateResponse);
   rpc BlobGet(BlobGetRequest) returns (BlobGetResponse);

modal_proto/api_grpc.py

@@ -82,6 +82,14 @@ class ModalClientBase(abc.ABC):
     async def AppStop(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.AppStopRequest, google.protobuf.empty_pb2.Empty]') -> None:
         pass
 
+    @abc.abstractmethod
+    async def AttemptAwait(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.AttemptAwaitRequest, modal_proto.api_pb2.AttemptAwaitResponse]') -> None:
+        pass
+
+    @abc.abstractmethod
+    async def AttemptStart(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.AttemptStartRequest, modal_proto.api_pb2.AttemptStartResponse]') -> None:
+        pass
+
     @abc.abstractmethod
     async def BlobCreate(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.BlobCreateRequest, modal_proto.api_pb2.BlobCreateResponse]') -> None:
         pass
@@ -700,6 +708,18 @@ class ModalClientBase(abc.ABC):
                 modal_proto.api_pb2.AppStopRequest,
                 google.protobuf.empty_pb2.Empty,
             ),
+            '/modal.client.ModalClient/AttemptAwait': grpclib.const.Handler(
+                self.AttemptAwait,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.api_pb2.AttemptAwaitRequest,
+                modal_proto.api_pb2.AttemptAwaitResponse,
+            ),
+            '/modal.client.ModalClient/AttemptStart': grpclib.const.Handler(
+                self.AttemptStart,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.api_pb2.AttemptStartRequest,
+                modal_proto.api_pb2.AttemptStartResponse,
+            ),
             '/modal.client.ModalClient/BlobCreate': grpclib.const.Handler(
                 self.BlobCreate,
                 grpclib.const.Cardinality.UNARY_UNARY,
@@ -1582,6 +1602,18 @@ class ModalClientStub:
             modal_proto.api_pb2.AppStopRequest,
             google.protobuf.empty_pb2.Empty,
         )
+        self.AttemptAwait = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.client.ModalClient/AttemptAwait',
+            modal_proto.api_pb2.AttemptAwaitRequest,
+            modal_proto.api_pb2.AttemptAwaitResponse,
+        )
+        self.AttemptStart = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.client.ModalClient/AttemptStart',
+            modal_proto.api_pb2.AttemptStartRequest,
+            modal_proto.api_pb2.AttemptStartResponse,
+        )
         self.BlobCreate = grpclib.client.UnaryUnaryMethod(
             channel,
             '/modal.client.ModalClient/BlobCreate',