modal 1.5.4.dev0__tar.gz → 1.5.4.dev1__tar.gz

{modal-1.5.4.dev0 → modal-1.5.4.dev1}/PKG-INFO

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

{modal-1.5.4.dev0 → modal-1.5.4.dev1}/modal/_environments.py

@@ -4,7 +4,7 @@ import builtins
 from collections.abc import Iterable, Mapping
 from dataclasses import dataclass
 from datetime import datetime, timezone
-from typing import Literal
+from typing import Any, Literal
 
 from google.protobuf.empty_pb2 import Empty
 from google.protobuf.message import Message
@@ -39,6 +39,7 @@ class _EnvironmentManager:
         name: str,  # Name to use for the new Environment
         *,
         restricted: bool = False,  # If True, enable RBAC restrictions on the Environment
+        experimental_options: dict[str, Any] | None = None,  # Experimental options for Environment creation
         client: _Client | None = None,  # Optional client with Modal credentials
     ) -> None:
         """Create a new Environment.
@@ -51,7 +52,10 @@ class _EnvironmentManager:
         """
         check_environment_name(name)
         client = await _Client.from_env() if client is None else client
-        await client.stub.EnvironmentCreate(api_pb2.EnvironmentCreateRequest(name=name, is_managed=restricted))
+        request = api_pb2.EnvironmentCreateRequest(name=name, is_managed=restricted)
+        if (experimental_options or {}).get("is_public", False):
+            request.environment_type = api_pb2.ENVIRONMENT_TYPE_PUBLIC
+        await client.stub.EnvironmentCreate(request)
 
     async def list(
         self,

{modal-1.5.4.dev0 → modal-1.5.4.dev1}/modal/_image.py

@@ -117,34 +117,51 @@ def _validate_image_tag(tag: str) -> None:
     check_object_name(tag, "Image tag")
 
 
-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(":")
+def _parse_image_name(partial_name: str) -> tuple[str | None, str]:
+    partial_name_no_env, sep, tag = partial_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 "/" in partial_name_no_env:
+        environment, image_name = partial_name_no_env.rsplit("/", 1)
+        if not environment:
+            raise InvalidError("Invalid Image name: environment prefix before '/' must be non-empty.")
         if not image_name:
-            raise InvalidError("Invalid Image name: name after '/' must be non-empty.")
+            raise InvalidError("Invalid Image name: name after last '/' must be non-empty.")
+    else:
+        environment = None
+        image_name = partial_name_no_env
 
     _validate_image_name(image_name)
     _validate_image_tag(tag)
 
-    full_name = f"{prefix}/{image_name}" if prefix else image_name
-    return prefix, f"{full_name}:{tag}"
+    return environment, f"{image_name}:{tag}"
+
+
+def _upgrade_image_name(
+    partial_name: str,
+    *,
+    explicit_environment_name: str | None = None,
+    fallback_environment_name: str | None = None,
+) -> str:
+    """Ensures an Image name has a tag and an environment name when possible
+
+    * If no tag is present, add :latest
+    * If an explicit environment name is passed, add it as a prefix and error
+      if there is already a prefix
+    * If a fallback environment name is passed, add it only when there is no prefix
+
+    """
+    prefix_environment, name_tag = _parse_image_name(partial_name)
+    if prefix_environment:
+        if explicit_environment_name is not None:
+            raise InvalidError("environment_name cannot be used when an Image name includes an environment prefix.")
+        return f"{prefix_environment}/{name_tag}"
+
+    environment_name = explicit_environment_name if explicit_environment_name is not None else fallback_environment_name
+    if environment_name:
+        return f"{environment_name}/{name_tag}"
+    return name_tag  # no explicit environment - use just the unprefixed name:tag
 
 
 def _validate_python_version(
@@ -2932,8 +2949,8 @@ class _Image(_Object, type_prefix="im"):
     ) -> "_Image":
         """Reference a named Image that was previously published with `.publish()`.
 
-        Names can contain an optional `:tag` part - if no tag part is included `":latest"` is used,
-        matching Docker conventions.
+        Names can contain an optional `:tag` part. If no tag part is included, `":latest"` is used, matching
+        Docker conventions.
 
         ```python notest
         image = modal.Image.from_name("my-image")     # references my-image:latest
@@ -2944,20 +2961,16 @@ class _Image(_Object, type_prefix="im"):
             ...
         ```
         """
-        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 '/'.")
+        _upgrade_image_name(name, explicit_environment_name=environment_name)  # validate
 
         async def _load(self: _Image, resolver: Resolver, load_context: LoadContext, existing_object_id: str | None):
             req = api_pb2.ImageGetByTagRequest(
-                tag=tag,
-                environment_name="" if namespace_prefix else load_context.environment_name,
+                tag=_upgrade_image_name(name, fallback_environment_name=load_context.environment_name),
             )
             response = await load_context.client.stub.ImageGetByTag(req)
             self._hydrate(response.image_id, load_context.client, None)
 
-        rep = _Image._repr(tag, environment_name)
+        rep = _Image._repr(name, environment_name)
         return _Image._from_loader(
             _load,
             rep,
@@ -2971,13 +2984,14 @@ class _Image(_Object, type_prefix="im"):
         name: str,
         *,
         environment_name: str | None = None,
+        experimental_options: dict[str, Any] | None = None,
         client: _Client | None = None,
     ) -> None:
         """Publish this image under the given name
 
         The Image must already be created (typically by calling `image.build()` or `sandbox.snapshot_filesystem()`).
 
-        Image names can contain an explicit tag designation (using the `name:tag`). If no tag is included in the name,
+        Image names can contain an explicit tag designation using `name:tag`. If no tag is included in the name,
         `":latest"` is used, matching Docker conventions. To publish multiple tags, call `.publish()` once per tag.
 
         ```python notest
@@ -2987,25 +3001,20 @@ class _Image(_Object, type_prefix="im"):
         image.publish("my-image-with-numpy:v1")
         ```
         """
-        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()
+        tag = _upgrade_image_name(
+            name,
+            explicit_environment_name=environment_name,
+            fallback_environment_name=config.get("environment"),
+        )
 
         await _client.stub.ImagePublish(
             api_pb2.ImagePublishRequest(
                 image_id=self._object_id,
-                environment_name=resolved_env,
-                allow_public=False,
+                allow_public=bool((experimental_options or {}).get("is_public", False)),
                 tag=tag,
             )
         )

{modal-1.5.4.dev0 → modal-1.5.4.dev1}/modal/cli/environment.py

@@ -72,10 +72,11 @@ def list_(json: bool = False):
 @environment_cli.command("create", help="Create a new environment in the current workspace.", no_args_is_help=True)
 @click.argument("name")
 @click.option("--restricted", is_flag=True, default=False, help="Enable RBAC restrictions on the new environment")
-def create(name: str, restricted: bool = False):
+@click.option("--public", is_flag=True, default=False, hidden=True)
+def create(name: str, restricted: bool = False, public: bool = False):
     check_environment_name(name)
-    Environment.objects.create(name, restricted=restricted)
-    prefix = "Restricted " if restricted else ""
+    Environment.objects.create(name, restricted=restricted, experimental_options={"is_public": public})
+    prefix = "Public " if public else "Restricted " if restricted else ""
     rich.print(f"[green]✓[/green] {prefix}Environment created: {name}")
 
 

{modal-1.5.4.dev0 → modal-1.5.4.dev1}/modal/cli/image.py

@@ -10,12 +10,12 @@ from typing import Optional
 import click
 
 from modal._environments import ensure_env
-from modal._image import _parse_named_image_ref
+from modal._image import _upgrade_image_name
 from modal._utils.async_utils import synchronizer
 from modal._utils.time_utils import timestamp_to_localized_str
 from modal.cli.utils import display_table, env_option
 from modal.client import _Client
-from modal.exception import InvalidError
+from modal.config import config
 from modal.output import OutputManager
 from modal_proto import api_pb2
 
@@ -91,14 +91,12 @@ async def _iter_tag_pages(
 
 async def _fetch_history_page(
     client: _Client,
-    env: str,
-    tag: str,
+    name_tag: str,
     page_token: str,
 ) -> api_pb2.ImageTagRevisionsResponse:
     return await client.stub.ImageTagRevisions(
         api_pb2.ImageTagRevisionsRequest(
-            tag=tag,
-            environment_name=env,
+            tag=name_tag,
             max_objects=IMAGE_NAMES_HISTORY_PAGE_SIZE,
             page_token=page_token,
         )
@@ -106,7 +104,7 @@ async def _fetch_history_page(
 
 
 async def _iter_history_pages(
-    client: _Client, env: str, tag: str, first_page: api_pb2.ImageTagRevisionsResponse
+    client: _Client, name_tag: str, first_page: api_pb2.ImageTagRevisionsResponse
 ) -> AsyncIterator[api_pb2.ImageTagRevisionsResponse]:
     response = first_page
 
@@ -116,7 +114,7 @@ async def _iter_history_pages(
         if not response.next_page_token:
             return
         await asyncio.sleep(IMAGE_NAMES_HISTORY_PAGE_DELAY_SECONDS)
-        response = await _fetch_history_page(client, env, tag, response.next_page_token)
+        response = await _fetch_history_page(client, name_tag, response.next_page_token)
 
 
 @image_names_cli.command("list", help="List named Images.")
@@ -160,36 +158,26 @@ async def list_(
     "history", help="Show publishing history for a named Image.", hidden=True, no_args_is_help=True
 )
 @click.argument("name")
-@env_option
 @click.option("--json", is_flag=True, default=False)
 @synchronizer.create_blocking
 async def history(
     name: str,
-    env: Optional[str] = None,
     json: bool = False,
 ):
-    explicit_env = env  # None when --env is not passed by the user
-    env = ensure_env(env)
-    try:
-        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 = ""
-
+    upgraded_image_name = _upgrade_image_name(
+        name,
+        fallback_environment_name=config.get("environment"),
+    )
     client = await _Client.from_env()
 
-    first_page = await _fetch_history_page(client, env, name_tag, "")
+    first_page = await _fetch_history_page(client, upgraded_image_name, "")
     if not first_page.items:
         raise click.ClickException(f"No publishing history found for named image {first_page.tag!r}.")
 
     if json:
         click.echo("[", nl=False)
         count = 0
-        async for response in _iter_history_pages(client, env, name_tag, first_page):
+        async for response in _iter_history_pages(client, upgraded_image_name, first_page):
             for item in response.items:
                 if count:
                     click.echo(",", nl=False)
@@ -200,7 +188,7 @@ async def history(
 
     count = 0
     show_title = True
-    async for response in _iter_history_pages(client, env, name_tag, first_page):
+    async for response in _iter_history_pages(client, upgraded_image_name, first_page):
         if response.items:
             title = f"History for {response.tag}" if show_title else ""
             display_table(

{modal-1.5.4.dev0 → modal-1.5.4.dev1}/modal/client.pyi

@@ -37,7 +37,7 @@ class _Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.5.4.dev0",
+        version: str = "1.5.4.dev1",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.
@@ -207,7 +207,7 @@ class Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "1.5.4.dev0",
+        version: str = "1.5.4.dev1",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.

{modal-1.5.4.dev0 → modal-1.5.4.dev1}/modal/environments.pyi

@@ -18,7 +18,13 @@ class EnvironmentManager:
 
     class __create_spec(typing_extensions.Protocol):
         def __call__(
-            self, /, name: str, *, restricted: bool = False, client: typing.Optional[modal.client.Client] = None
+            self,
+            /,
+            name: str,
+            *,
+            restricted: bool = False,
+            experimental_options: typing.Optional[dict[str, typing.Any]] = None,
+            client: typing.Optional[modal.client.Client] = None,
         ) -> None:
             """Create a new Environment.
 
@@ -31,7 +37,13 @@ class EnvironmentManager:
             ...
 
         async def aio(
-            self, /, name: str, *, restricted: bool = False, client: typing.Optional[modal.client.Client] = None
+            self,
+            /,
+            name: str,
+            *,
+            restricted: bool = False,
+            experimental_options: typing.Optional[dict[str, typing.Any]] = None,
+            client: typing.Optional[modal.client.Client] = None,
         ) -> None:
             """Create a new Environment.
 

{modal-1.5.4.dev0 → modal-1.5.4.dev1}/modal/functions.pyi

@@ -516,7 +516,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.
 
@@ -541,7 +541,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]:
@@ -591,7 +591,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.
 
@@ -624,7 +624,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: ...
@@ -632,7 +632,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.
 
@@ -665,7 +665,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.4.dev0 → modal-1.5.4.dev1}/modal/image.pyi

@@ -1362,8 +1362,8 @@ class Image(modal.object.Object):
     ) -> Image:
         """Reference a named Image that was previously published with `.publish()`.
 
-        Names can contain an optional `:tag` part - if no tag part is included `":latest"` is used,
-        matching Docker conventions.
+        Names can contain an optional `:tag` part. If no tag part is included, `":latest"` is used, matching
+        Docker conventions.
 
         ```python notest
         image = modal.Image.from_name("my-image")     # references my-image:latest
@@ -1383,13 +1383,14 @@ class Image(modal.object.Object):
             name: str,
             *,
             environment_name: typing.Optional[str] = None,
+            experimental_options: typing.Optional[dict[str, typing.Any]] = None,
             client: typing.Optional[modal.client.Client] = None,
         ) -> None:
             """Publish this image under the given name
 
             The Image must already be created (typically by calling `image.build()` or `sandbox.snapshot_filesystem()`).
 
-            Image names can contain an explicit tag designation (using the `name:tag`). If no tag is included in the name,
+            Image names can contain an explicit tag designation using `name:tag`. If no tag is included in the name,
             `":latest"` is used, matching Docker conventions. To publish multiple tags, call `.publish()` once per tag.
 
             ```python notest
@@ -1407,13 +1408,14 @@ class Image(modal.object.Object):
             name: str,
             *,
             environment_name: typing.Optional[str] = None,
+            experimental_options: typing.Optional[dict[str, typing.Any]] = None,
             client: typing.Optional[modal.client.Client] = None,
         ) -> None:
             """Publish this image under the given name
 
             The Image must already be created (typically by calling `image.build()` or `sandbox.snapshot_filesystem()`).
 
-            Image names can contain an explicit tag designation (using the `name:tag`). If no tag is included in the name,
+            Image names can contain an explicit tag designation using `name:tag`. If no tag is included in the name,
             `":latest"` is used, matching Docker conventions. To publish multiple tags, call `.publish()` once per tag.
 
             ```python notest

{modal-1.5.4.dev0 → modal-1.5.4.dev1}/modal.egg-info/PKG-INFO

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