modAL 1.2.7.dev9__tar.gz → 1.2.7.dev10__tar.gz

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/PKG-INFO

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

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/modal/_container_entrypoint.py

@@ -43,7 +43,6 @@ from ._runtime.container_io_manager import (
 )
 
 if TYPE_CHECKING:
-    import modal._object
     import modal._runtime.container_io_manager
     import modal._runtime.user_code_imports
 

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/modal/_utils/grpc_utils.py

@@ -16,8 +16,6 @@ from typing import Any, Optional, Sequence, TypeVar
 import grpclib.client
 import grpclib.config
 import grpclib.events
-import grpclib.protocol
-import grpclib.stream
 from google.protobuf.message import Message
 from google.protobuf.symbol_database import SymbolDatabase
 from grpclib import GRPCError, Status

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/modal/_utils/package_utils.py

@@ -1,5 +1,4 @@
 # Copyright Modal Labs 2022
-import importlib
 import importlib.util
 import typing
 from importlib.metadata import PackageNotFoundError, files

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/modal/_utils/task_command_router_client.py

@@ -521,3 +521,17 @@ class TaskCommandRouterClient:
                     await sleep_and_update_delay_and_num_retries_remaining(e)
                 else:
                     raise ConnectionError(str(e))
+
+    async def mount_image(self, request: sr_pb2.TaskMountDirectoryRequest):
+        with grpc_error_converter():
+            return await call_with_retries_on_transient_errors(
+                lambda: self._call_with_auth_retry(self._stub.TaskMountDirectory, request)
+            )
+
+    async def snapshot_directory(
+        self, request: sr_pb2.TaskSnapshotDirectoryRequest
+    ) -> sr_pb2.TaskSnapshotDirectoryResponse:
+        with grpc_error_converter():
+            return await call_with_retries_on_transient_errors(
+                lambda: self._call_with_auth_retry(self._stub.TaskSnapshotDirectory, request)
+            )

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/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.7.dev9",
+        version: str = "1.2.7.dev10",
     ):
         """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.7.dev9",
+        version: str = "1.2.7.dev10",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/modal/image.py

@@ -874,6 +874,7 @@ class _Image(_Object, type_prefix="im"):
 
         rep = f"Image.from_id({image_id!r})"
         obj = _Image._from_loader(_load, rep, load_context_overrides=LoadContext(client=client))
+        obj._object_id = image_id
 
         return obj
 

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/modal/sandbox.py

@@ -6,6 +6,7 @@ import time
 import uuid
 from collections.abc import AsyncGenerator, Collection, Sequence
 from dataclasses import dataclass
+from pathlib import PurePosixPath
 from typing import TYPE_CHECKING, Any, AsyncIterator, Literal, Optional, Union, overload
 
 from ._pty import get_pty_info
@@ -600,6 +601,49 @@ class _Sandbox(_Object, type_prefix="sb"):
 
         return image
 
+    async def _experimental_mount_image(self, path: Union[PurePosixPath, str], image: Optional[_Image]):
+        """Mount an Image at a path in the Sandbox filesystem."""
+
+        image_id = None
+
+        if image:
+            if not image._object_id:
+                # FIXME
+                raise InvalidError("Image has not been built.")
+            image_id = image._object_id
+        else:
+            image_id = ""  # empty string indicates mount an empty dir
+
+        task_id = await self._get_task_id()
+        if (command_router_client := await self._get_command_router_client(task_id)) is None:
+            raise InvalidError("Mounting directories requires direct Sandbox control - please contact Modal support.")
+
+        posix_path = PurePosixPath(path)
+        if not posix_path.is_absolute():
+            raise InvalidError(f"Mount path must be absolute; got: {posix_path}")
+        path_bytes = posix_path.as_posix().encode("utf8")
+
+        req = sr_pb2.TaskMountDirectoryRequest(task_id=task_id, path=path_bytes, image_id=image_id)
+        await command_router_client.mount_image(req)
+
+    async def _experimental_snapshot_directory(self, path: Union[PurePosixPath, str]) -> _Image:
+        """Snapshot local changes to a previously mounted Image, creating a new Image."""
+
+        task_id = await self._get_task_id()
+        if (command_router_client := await self._get_command_router_client(task_id)) is None:
+            raise InvalidError(
+                "Snapshotting directories requires direct Sandbox control - please contact Modal support."
+            )
+
+        posix_path = PurePosixPath(path)
+        if not posix_path.is_absolute():
+            raise InvalidError(f"Snapshot path must be absolute; got: {posix_path}")
+        path_bytes = posix_path.as_posix().encode("utf8")
+
+        req = sr_pb2.TaskSnapshotDirectoryRequest(task_id=task_id, path=path_bytes)
+        res = await command_router_client.snapshot_directory(req)
+        return _Image._new_hydrated(res.image_id, self._client, None)
+
     # Live handle methods
 
     async def wait(self, raise_on_termination: bool = True):

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/modal/sandbox.pyi

@@ -22,6 +22,7 @@ import modal.stream_type
 import modal.volume
 import modal_proto.api_pb2
 import os
+import pathlib
 import typing
 import typing_extensions
 
@@ -254,6 +255,18 @@ class _Sandbox(modal._object._Object):
         """
         ...
 
+    async def _experimental_mount_image(
+        self, path: typing.Union[pathlib.PurePosixPath, str], image: typing.Optional[modal.image._Image]
+    ):
+        """Mount an Image at a path in the Sandbox filesystem."""
+        ...
+
+    async def _experimental_snapshot_directory(
+        self, path: typing.Union[pathlib.PurePosixPath, str]
+    ) -> modal.image._Image:
+        """Snapshot local changes to a previously mounted Image, creating a new Image."""
+        ...
+
     async def wait(self, raise_on_termination: bool = True):
         """Wait for the Sandbox to finish running."""
         ...
@@ -820,6 +833,32 @@ class Sandbox(modal.object.Object):
 
     snapshot_filesystem: __snapshot_filesystem_spec[typing_extensions.Self]
 
+    class ___experimental_mount_image_spec(typing_extensions.Protocol[SUPERSELF]):
+        def __call__(
+            self, /, path: typing.Union[pathlib.PurePosixPath, str], image: typing.Optional[modal.image.Image]
+        ):
+            """Mount an Image at a path in the Sandbox filesystem."""
+            ...
+
+        async def aio(
+            self, /, path: typing.Union[pathlib.PurePosixPath, str], image: typing.Optional[modal.image.Image]
+        ):
+            """Mount an Image at a path in the Sandbox filesystem."""
+            ...
+
+    _experimental_mount_image: ___experimental_mount_image_spec[typing_extensions.Self]
+
+    class ___experimental_snapshot_directory_spec(typing_extensions.Protocol[SUPERSELF]):
+        def __call__(self, /, path: typing.Union[pathlib.PurePosixPath, str]) -> modal.image.Image:
+            """Snapshot local changes to a previously mounted Image, creating a new Image."""
+            ...
+
+        async def aio(self, /, path: typing.Union[pathlib.PurePosixPath, str]) -> modal.image.Image:
+            """Snapshot local changes to a previously mounted Image, creating a new Image."""
+            ...
+
+    _experimental_snapshot_directory: ___experimental_snapshot_directory_spec[typing_extensions.Self]
+
     class __wait_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, /, raise_on_termination: bool = True):
             """Wait for the Sandbox to finish running."""

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/modal.egg-info/PKG-INFO

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

{modal-1.2.7.dev9 → modal-1.2.7.dev10}/modal_version/__init__.py

@@ -1,4 +1,4 @@
 # Copyright Modal Labs 2025
 """Supplies the current version of the modal client library."""
 
-__version__ = "1.2.7.dev9"
+__version__ = "1.2.7.dev10"