modal 1.5.2.dev19__tar.gz → 1.5.2.dev20__tar.gz

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/PKG-INFO

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

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/_environments.py

@@ -103,9 +103,6 @@ class _EnvironmentManager:
         await client.stub.EnvironmentDelete(api_pb2.EnvironmentDeleteRequest(name=name))
 
 
-MemberRole = Literal["viewer", "contributor"]
-
-
 def _role_to_proto(role: str) -> "api_pb2.EnvironmentRole.ValueType":
     match role:
         case "viewer":
@@ -116,7 +113,7 @@ def _role_to_proto(role: str) -> "api_pb2.EnvironmentRole.ValueType":
             raise InvalidError(f"Invalid Environment role: {role!r} (expected 'viewer' or 'contributor')")
 
 
-def _role_from_proto(proto_value: int) -> MemberRole:
+def _role_from_proto(proto_value: int) -> str:
     match proto_value:
         case api_pb2.ENVIRONMENT_ROLE_VIEWER:
             return "viewer"
@@ -137,7 +134,7 @@ class _EnvironmentMembersManager:
         """mdmd:hidden"""
         self._environment = environment
 
-    async def list(self) -> dict[Literal["users", "service_users"], dict[str, MemberRole]]:
+    async def list(self) -> dict[Literal["users", "service_users"], dict[str, str]]:
         """Return the members of a restricted Environment with their roles.
 
         **Examples:**
@@ -155,8 +152,8 @@ class _EnvironmentMembersManager:
         req = api_pb2.EnvironmentGetManagedRequest(environment_id=self._environment.object_id)
         resp = await self._environment.client.stub.EnvironmentGetManaged(req)
 
-        users: dict[str, MemberRole] = {}
-        service_users: dict[str, MemberRole] = {}
+        users: dict[str, str] = {}
+        service_users: dict[str, str] = {}
         for principal in resp.principal_roles:
             role = _role_from_proto(principal.role)
             if principal.user_id:
@@ -169,8 +166,8 @@ class _EnvironmentMembersManager:
     async def update(
         self,
         *,
-        users: Mapping[str, MemberRole] | None = None,
-        service_users: Mapping[str, MemberRole] | None = None,
+        users: Mapping[str, str] | None = None,
+        service_users: Mapping[str, str] | None = None,
     ) -> None:
         """Add or modify roles for members of a restricted Environment.
 

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/_image.py

@@ -117,13 +117,34 @@ def _validate_image_tag(tag: str) -> None:
     check_object_name(tag, "Image tag")
 
 
-def _parse_named_image_ref(name: str) -> str:
+def _parse_named_image_ref(name: str) -> tuple[str, str]:
+    """Parse an image reference, returning (namespace_prefix, name_tag).
+
+    If the name contains a '/', the part before the last '/' is extracted as
+    a namespace prefix (intended for environment/name or workspace/env/name
+    syntax). The actual image name (after the last '/') is validated as a
+    standard object name.
+
+    Returns a tuple of (prefix, "full_name:tag") where prefix is empty string
+    if no '/' is present.
+    """
     image_name, sep, tag = name.partition(":")
     if not sep:
         tag = "latest"
+
+    prefix = ""
+    if "/" in image_name:
+        prefix, image_name = image_name.rsplit("/", 1)
+        if not prefix:
+            raise InvalidError("Invalid Image name: '/' prefix must be non-empty.")
+        if not image_name:
+            raise InvalidError("Invalid Image name: name after '/' must be non-empty.")
+
     _validate_image_name(image_name)
     _validate_image_tag(tag)
-    return f"{image_name}:{tag}"
+
+    full_name = f"{prefix}/{image_name}" if prefix else image_name
+    return prefix, f"{full_name}:{tag}"
 
 
 def _validate_python_version(
@@ -2923,12 +2944,15 @@ class _Image(_Object, type_prefix="im"):
             ...
         ```
         """
-        tag = _parse_named_image_ref(name)
+        namespace_prefix, tag = _parse_named_image_ref(name)
+
+        if namespace_prefix and environment_name:
+            raise InvalidError("Cannot specify 'environment_name' when the image name contains a '/'.")
 
         async def _load(self: _Image, resolver: Resolver, load_context: LoadContext, existing_object_id: str | None):
             req = api_pb2.ImageGetByTagRequest(
                 tag=tag,
-                environment_name=load_context.environment_name,
+                environment_name="" if namespace_prefix else load_context.environment_name,
             )
             response = await load_context.client.stub.ImageGetByTag(req)
             self._hydrate(response.image_id, load_context.client, None)
@@ -2963,15 +2987,20 @@ class _Image(_Object, type_prefix="im"):
         image.publish("my-image-with-numpy:v1")
         ```
         """
-        tag = _parse_named_image_ref(name)
+        namespace_prefix, tag = _parse_named_image_ref(name)
+
+        if namespace_prefix:
+            if environment_name:
+                raise InvalidError("Cannot specify 'environment_name' when the image name contains a '/'.")
+            resolved_env = ""
+        else:
+            resolved_env = environment_name or config.get("environment") or ""
 
         if self._object_id is None:
             raise InvalidError("Cannot publish an image that has not been created yet. Call `.build()` first.")
 
         _client = client or await _Client.from_env()
 
-        resolved_env = environment_name or config.get("environment") or ""
-
         await _client.stub.ImagePublish(
             api_pb2.ImagePublishRequest(
                 image_id=self._object_id,

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/_workspace.py

@@ -13,13 +13,19 @@ from ._object import _Object
 from ._resolver import Resolver
 from ._utils.time_utils import timestamp_to_localized_dt
 from .client import _Client
-from .types import BillingReportItem, MemberRole, ProxyTokenInfo, TokenData, WorkspaceMemberInfo, WorkspaceSettings
+from .types import (
+    BillingReportItem,
+    ProxyTokenInfo,
+    TokenData,
+    WorkspaceMemberInfo,
+    WorkspaceSettings,
+)
 
 
-def _member_role_from_proto(proto_value: int) -> MemberRole:
+def _member_role_from_proto(proto_value: int) -> str:
     match proto_value:
         case api_pb2.MEMBER_ROLE_USER:
-            return "user"
+            return "member"
         case api_pb2.MEMBER_ROLE_MANAGER:
             return "manager"
         case api_pb2.MEMBER_ROLE_OWNER:

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/cli/container.py

@@ -202,20 +202,20 @@ async def logs(
         task_info_resp = await client.stub.TaskGetInfo(api_pb2.TaskGetInfoRequest(task_id=task_id))
         app_id = task_info_resp.app_id
 
-        if not task_info_resp.info.started_at:
-            # Unlikely race or Modal backend issue, don't treat as a usage exception
+        if not task_info_resp.info.enqueued_at:
             return
-        container_started_dt = datetime.fromtimestamp(task_info_resp.info.started_at, timezone.utc)
+
+        container_enqueued_dt = datetime.fromtimestamp(task_info_resp.info.enqueued_at, timezone.utc)
 
         now = datetime.now(timezone.utc)
         if all_logs:
-            since_dt = container_started_dt
+            since_dt = container_enqueued_dt
             if task_info_resp.info.finished_at:
                 until_dt = datetime.fromtimestamp(task_info_resp.info.finished_at, timezone.utc)
             else:
                 until_dt = now
         else:
-            since_dt = _parse_time_arg(since, default=container_started_dt)
+            since_dt = _parse_time_arg(since, default=container_enqueued_dt)
             if task_info_resp.info.finished_at:
                 default_until_dt = datetime.fromtimestamp(task_info_resp.info.finished_at, timezone.utc)
             else:

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/cli/environment.py

@@ -7,7 +7,7 @@ from click import UsageError
 from rich.text import Text
 
 from modal import environments
-from modal._environments import MemberRole, _Environment
+from modal._environments import _Environment
 from modal._utils.async_utils import synchronizer
 from modal._utils.name_utils import check_environment_name
 from modal._utils.time_utils import format_interval
@@ -158,7 +158,7 @@ def members_list(environment: str, json: bool = False):
     "--role", type=click.Choice(["contributor", "viewer"]), required=True, help="Role to assign to the member"
 )
 @service_user_option
-def members_update(environment: str, member: str, role: MemberRole, service_user: bool = False):
+def members_update(environment: str, member: str, role: str, service_user: bool = False):
     env = Environment.from_name(environment)
     if service_user:
         env.members.update(service_users={member: role})

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/cli/image.py

@@ -168,11 +168,18 @@ async def history(
     env: Optional[str] = None,
     json: bool = False,
 ):
+    explicit_env = env  # None when --env is not passed by the user
     env = ensure_env(env)
     try:
-        name_tag = _parse_named_image_ref(name)
+        namespace_prefix, name_tag = _parse_named_image_ref(name)
     except InvalidError as exc:
         raise click.UsageError(str(exc)) from exc
+
+    if namespace_prefix:
+        if explicit_env is not None:
+            raise click.UsageError("Cannot specify '--env' when the image name contains a '/'.")
+        env = ""
+
     client = await _Client.from_env()
 
     first_page = await _fetch_history_page(client, env, name_tag, "")

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/client.pyi

@@ -35,7 +35,7 @@ class _Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.5.2.dev19",
+        version: str = "1.5.2.dev20",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.
@@ -205,7 +205,7 @@ class Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.5.2.dev19",
+        version: str = "1.5.2.dev20",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/environments.pyi

@@ -112,9 +112,7 @@ class EnvironmentMembersManager:
         ...
 
     class __list_spec(typing_extensions.Protocol):
-        def __call__(
-            self, /
-        ) -> dict[typing.Literal["users", "service_users"], dict[str, typing.Literal["viewer", "contributor"]]]:
+        def __call__(self, /) -> dict[typing.Literal["users", "service_users"], dict[str, str]]:
             """Return the members of a restricted Environment with their roles.
 
             **Examples:**
@@ -130,9 +128,7 @@ class EnvironmentMembersManager:
             """
             ...
 
-        async def aio(
-            self, /
-        ) -> dict[typing.Literal["users", "service_users"], dict[str, typing.Literal["viewer", "contributor"]]]:
+        async def aio(self, /) -> dict[typing.Literal["users", "service_users"], dict[str, str]]:
             """Return the members of a restricted Environment with their roles.
 
             **Examples:**
@@ -155,10 +151,8 @@ class EnvironmentMembersManager:
             self,
             /,
             *,
-            users: typing.Optional[collections.abc.Mapping[str, typing.Literal["viewer", "contributor"]]] = None,
-            service_users: typing.Optional[
-                collections.abc.Mapping[str, typing.Literal["viewer", "contributor"]]
-            ] = None,
+            users: typing.Optional[collections.abc.Mapping[str, str]] = None,
+            service_users: typing.Optional[collections.abc.Mapping[str, str]] = None,
         ) -> None:
             """Add or modify roles for members of a restricted Environment.
 
@@ -181,10 +175,8 @@ class EnvironmentMembersManager:
             self,
             /,
             *,
-            users: typing.Optional[collections.abc.Mapping[str, typing.Literal["viewer", "contributor"]]] = None,
-            service_users: typing.Optional[
-                collections.abc.Mapping[str, typing.Literal["viewer", "contributor"]]
-            ] = None,
+            users: typing.Optional[collections.abc.Mapping[str, str]] = None,
+            service_users: typing.Optional[collections.abc.Mapping[str, str]] = None,
         ) -> None:
             """Add or modify roles for members of a restricted Environment.
 

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/functions.pyi

@@ -491,7 +491,7 @@ class Function(
 
     _call_generator: ___call_generator_spec
 
-    class __remote_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER]):
+    class __remote_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER]):
         def __call__(self, /, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> ReturnType_INNER:
             """Calls the function remotely, executing it with the given arguments and returning the execution's result.
 
@@ -516,7 +516,7 @@ class Function(
             """
             ...
 
-    remote: __remote_spec[modal._functions.P, modal._functions.ReturnType]
+    remote: __remote_spec[modal._functions.ReturnType, modal._functions.P]
 
     class __remote_gen_spec(typing_extensions.Protocol):
         def __call__(self, /, *args, **kwargs) -> typing.Generator[typing.Any, None, None]:
@@ -566,7 +566,7 @@ class Function(
         """
         ...
 
-    class ___experimental_spawn_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER]):
+    class ___experimental_spawn_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER]):
         def __call__(self, /, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]:
             """[Experimental] Calls the function with the given arguments, without waiting for the results.
 
@@ -599,7 +599,7 @@ class Function(
             """
             ...
 
-    _experimental_spawn: ___experimental_spawn_spec[modal._functions.P, modal._functions.ReturnType]
+    _experimental_spawn: ___experimental_spawn_spec[modal._functions.ReturnType, modal._functions.P]
 
     class ___spawn_map_inner_spec(typing_extensions.Protocol[P_INNER]):
         def __call__(self, /, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> None: ...
@@ -607,7 +607,7 @@ class Function(
 
     _spawn_map_inner: ___spawn_map_inner_spec[modal._functions.P]
 
-    class __spawn_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER]):
+    class __spawn_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER]):
         def __call__(self, /, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]:
             """Calls the function with the given arguments, without waiting for the results.
 
@@ -640,7 +640,7 @@ class Function(
             """
             ...
 
-    spawn: __spawn_spec[modal._functions.P, modal._functions.ReturnType]
+    spawn: __spawn_spec[modal._functions.ReturnType, modal._functions.P]
 
     def get_raw_f(self) -> collections.abc.Callable[..., typing.Any]:
         """Return the inner Python object wrapped by this Modal Function.

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/types.py

@@ -5,7 +5,7 @@ import enum
 from dataclasses import FrozenInstanceError, dataclass
 from datetime import datetime, timezone
 from decimal import Decimal
-from typing import Any, Iterable, Literal, Optional
+from typing import Any, Iterable, Optional
 
 from modal_proto import api_pb2
 
@@ -54,9 +54,6 @@ class FileWatchEventType(enum.Enum):
     Remove = "Remove"
 
 
-MemberRole = Literal["user", "manager", "owner"]
-
-
 @dataclass
 class DictInfo:
     """Information about a Dict object."""
@@ -206,7 +203,7 @@ class WorkspaceMemberInfo:
     name: str
     email: str
     user_id: str
-    role: MemberRole
+    role: str
     joined_at: datetime
     last_active_at: Optional[datetime]  # None if the member has never been active
 

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal/workspace.pyi

@@ -72,7 +72,7 @@ class WorkspaceMemberInfo:
     name: str
     email: str
     user_id: str
-    role: typing.Literal["user", "manager", "owner"]
+    role: str
     joined_at: datetime.datetime
     last_active_at: typing.Optional[datetime.datetime]
 
@@ -81,7 +81,7 @@ class WorkspaceMemberInfo:
         name: str,
         email: str,
         user_id: str,
-        role: typing.Literal["user", "manager", "owner"],
+        role: str,
         joined_at: datetime.datetime,
         last_active_at: typing.Optional[datetime.datetime],
     ) -> None:

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal.egg-info/PKG-INFO

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

{modal-1.5.2.dev19 → modal-1.5.2.dev20}/modal_proto/api_grpc.py

@@ -483,6 +483,10 @@ class ModalClientBase(abc.ABC):
     async def MapStartOrContinue(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.MapStartOrContinueRequest, modal_proto.api_pb2.MapStartOrContinueResponse]') -> None:
         pass
 
+    @abc.abstractmethod
+    async def MountBatchedCheckExistence(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.MountBatchedCheckExistenceRequest, modal_proto.api_pb2.MountBatchedCheckExistenceResponse]') -> None:
+        pass
+
     @abc.abstractmethod
     async def MountGetOrCreate(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.MountGetOrCreateRequest, modal_proto.api_pb2.MountGetOrCreateResponse]') -> None:
         pass
@@ -1609,6 +1613,12 @@ class ModalClientBase(abc.ABC):
                 modal_proto.api_pb2.MapStartOrContinueRequest,
                 modal_proto.api_pb2.MapStartOrContinueResponse,
             ),
+            '/modal.client.ModalClient/MountBatchedCheckExistence': grpclib.const.Handler(
+                self.MountBatchedCheckExistence,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.api_pb2.MountBatchedCheckExistenceRequest,
+                modal_proto.api_pb2.MountBatchedCheckExistenceResponse,
+            ),
             '/modal.client.ModalClient/MountGetOrCreate': grpclib.const.Handler(
                 self.MountGetOrCreate,
                 grpclib.const.Cardinality.UNARY_UNARY,
@@ -2953,6 +2963,12 @@ class ModalClientStub:
             modal_proto.api_pb2.MapStartOrContinueRequest,
             modal_proto.api_pb2.MapStartOrContinueResponse,
         )
+        self.MountBatchedCheckExistence = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.client.ModalClient/MountBatchedCheckExistence',
+            modal_proto.api_pb2.MountBatchedCheckExistenceRequest,
+            modal_proto.api_pb2.MountBatchedCheckExistenceResponse,
+        )
         self.MountGetOrCreate = grpclib.client.UnaryUnaryMethod(
             channel,
             '/modal.client.ModalClient/MountGetOrCreate',