modAL 1.5.4.dev8__py3-none-any.whl → 1.5.4.dev10__py3-none-any.whl

modal/_output/rich.py

@@ -37,6 +37,7 @@ from rich.progress import (
 )
 from rich.spinner import Spinner
 from rich.status import Status
+from rich.table import Column
 from rich.text import Text
 from rich.tree import Tree
 
@@ -476,7 +477,10 @@ class RichOutputManager(OutputManager):
         and adds it to the current render group."""
         if not self._function_queueing_progress:
             self._function_queueing_progress = Progress(
-                TextColumn("[progress.description]{task.description}"),
+                TextColumn(
+                    "[progress.description]{task.description}",
+                    table_column=Column(no_wrap=False, overflow="fold"),
+                ),
                 FunctionQueuingColumn(),
                 console=self._console,
                 transient=True,
@@ -581,11 +585,17 @@ class RichOutputManager(OutputManager):
         """Handle queueing updates, ignoring completion updates for functions that have no queue progress bar."""
         task_key = (function_id, api_pb2.FUNCTION_QUEUED)
         task_description = description or f"'{function_id}' function waiting on worker"
+        task_description = escape(task_description)
         task_desc = f"[yellow]{task_description}. Time in queue:"
         if task_key in self._task_progress_items:
             progress_task_id = self._task_progress_items[task_key]
             try:
-                self._function_queueing_progress_bar.update(progress_task_id, completed=completed, total=total)
+                self._function_queueing_progress_bar.update(
+                    progress_task_id,
+                    description=task_desc if description else None,
+                    completed=completed,
+                    total=total,
+                )
                 if completed == total:
                     del self._task_progress_items[task_key]
                     self._function_queueing_progress_bar.remove_task(progress_task_id)

modal/_utils/auth_token_manager.py

@@ -2,6 +2,7 @@
 import asyncio
 import base64
 import json
+import random
 import time
 from typing import Any
 
@@ -17,8 +18,14 @@ AUTH_DENIED_EXCEPTIONS = (AuthError, PermissionDeniedError)
 class _AuthTokenManager:
     """Handles fetching and refreshing of the input plane auth token."""
 
-    # Start refreshing this many seconds before the token expires
-    REFRESH_WINDOW = 5 * 60
+    # Fraction of the token's lifetime to use before refreshing.
+    REFRESH_FRACTION = 0.5
+    # Width of the random band added on top of REFRESH_FRACTION, so that clients issued
+    # same-lifetime tokens at the same moment (a fan-out of containers starting together)
+    # spread their refreshes out instead of refreshing in lockstep every cycle. The band
+    # only ever delays a refresh, so the margin before expiry stays at no less than
+    # 1 - REFRESH_FRACTION - REFRESH_JITTER of the lifetime.
+    REFRESH_JITTER = 0.1
     # Bound each AuthTokenGet attempt so a refresh can't hang.
     AUTH_TOKEN_TIMEOUT = 5.0
     AUTH_TOKEN_RETRY_TOTAL_TIMEOUT = 3 * AUTH_TOKEN_TIMEOUT
@@ -33,9 +40,10 @@ class _AuthTokenManager:
         self._stub = stub
         self._token = ""
         self._expiry = 0.0
+        self._refresh_at = 0.0
         self._retry_after = 0.0
         self._backoff = self.FAILURE_BACKOFF_BASE
-        self._lock: asyncio.Lock | None = None
+        self._refresh_task: asyncio.Task | None = None
 
     async def get_token(self) -> str:
         """
@@ -45,21 +53,20 @@ class _AuthTokenManager:
         concurrently by multiple coroutines, all requests will block until the token has been fetched. But only one
         coroutine will actually make a request to the control plane to fetch the new token. This ensures we do not hit
         the control plane with more requests than needed.
-        3. Has a valid cached token, but it is going to expire in the next 5 minutes. In this case we fetch a new token
-        and cache it. If `get_token` is called concurrently, only one request will fetch the new token, and the others
-        will be given the old (but still valid) token - i.e. they will not block.
+        3. Has a valid cached token that has used up its refresh fraction of its lifetime. In this case we fetch a new
+        token and cache it. If `get_token` is called concurrently, only one request will fetch the new token, and the
+        others will be given the old (but still valid) token - i.e. they will not block.
         """
         if not self._token or self._is_expired():
             # We either have no token or it is expired - block everyone until we get a new token
             await self._try_refresh_token()
         elif self._should_refresh():
-            # The token hasn't expired yet, but will soon, so it needs a refresh.
-            lock = await self._get_lock()
-            if lock.locked():
-                # The lock is taken, so someone else is refreshing. Continue to use the old token.
+            # The token is still valid, but has used up its refresh fraction of its lifetime.
+            if self._refresh_task is not None:
+                # A refresh is in flight, so someone else is refreshing. Continue to use the old token.
                 return self._token
             else:
-                # The lock is not taken, so we need to fetch a new token.
+                # Nobody is refreshing, so we need to fetch a new token.
                 await self._try_refresh_token()
 
         return self._token
@@ -76,58 +83,63 @@ class _AuthTokenManager:
 
     async def _refresh_token(self):
         """
-        Fetch a new token from the control plane. If called concurrently, only one coroutine will make a request for a
-        new token. The others will block on a lock, until the first coroutine has fetched the new token.
+        Fetch a new token from the control plane. Concurrent callers share a single request: the first caller starts it
+        and the others await the same task, so a burst of callers results in one RPC whether it succeeds or fails. The
+        task is owned by the event loop rather than by the caller that started it, so it runs to completion even if that
+        caller goes away.
         """
-        lock = await self._get_lock()
-        async with lock:
-            # Double check inside lock - maybe another coroutine refreshed already. This happens the first time we fetch
-            # the token. The first coroutine will fetch the token, while the others block on the lock, waiting for the
-            # new token. Once we have a new token, the other coroutines will unblock and return from here.
+        if self._refresh_task is None:
+            # Maybe another coroutine refreshed already, or we're cooling down after a failure.
             if self._token and not self._should_refresh():
                 return
-            try:
-                resp: api_pb2.AuthTokenGetResponse = await self._stub.AuthTokenGet(
-                    api_pb2.AuthTokenGetRequest(),
-                    # No cached token to fall back on, so a failure is user-visible: retry transient errors.
-                    # Otherwise one attempt, and _retry_after handles the cooldown.
-                    retry=Retry(
-                        attempt_timeout=self.AUTH_TOKEN_TIMEOUT,
-                        total_timeout=self.AUTH_TOKEN_RETRY_TOTAL_TIMEOUT,
-                        max_retries=0 if self._token else DEFAULT_MAX_RETRIES,
-                    ),
+            # Note: the task is created in async code so it binds to the synchronicity event loop.
+            self._refresh_task = asyncio.create_task(self._fetch_token())
+            self._refresh_task.add_done_callback(self._refresh_task_done)
+        # Shielded so a cancelled caller doesn't cancel the refresh for everyone else.
+        await asyncio.shield(self._refresh_task)
+
+    def _refresh_task_done(self, task: "asyncio.Task"):
+        if self._refresh_task is task:
+            self._refresh_task = None
+        if not task.cancelled():
+            # Retrieve the exception, if any, so a failure isn't logged as unhandled when no caller is waiting.
+            task.exception()
+
+    async def _fetch_token(self):
+        """Make the AuthTokenGet request and cache its token, or arm the failure backoff."""
+        try:
+            resp: api_pb2.AuthTokenGetResponse = await self._stub.AuthTokenGet(
+                api_pb2.AuthTokenGetRequest(),
+                # No cached token to fall back on, so a failure is user-visible: retry transient errors.
+                # Otherwise one attempt, and _retry_after handles the cooldown.
+                retry=Retry(
+                    attempt_timeout=self.AUTH_TOKEN_TIMEOUT,
+                    total_timeout=self.AUTH_TOKEN_RETRY_TOTAL_TIMEOUT,
+                    max_retries=0 if self._token else DEFAULT_MAX_RETRIES,
+                ),
+            )
+            if not resp.token:
+                # Not expected
+                raise ExecutionError(
+                    "Internal error: Did not receive auth token from server. Please contact Modal support."
                 )
-                if not resp.token:
-                    # Not expected
-                    raise ExecutionError(
-                        "Internal error: Did not receive auth token from server. Please contact Modal support."
-                    )
-                if exp := self._decode_jwt(resp.token).get("exp"):
-                    expiry = float(exp)
-                else:
-                    # This should never happen.
-                    logger.warning("x-modal-auth-token does not contain exp field")
-                    expiry = time.time() + self.DEFAULT_EXPIRY_OFFSET
-                self._token = resp.token
-                self._expiry = expiry
-                self._retry_after = 0.0
-                self._backoff = self.FAILURE_BACKOFF_BASE
-            except AUTH_DENIED_EXCEPTIONS:
-                raise
-            except Exception as e:
-                # Back off (exponentially on consecutive failures) so we don't hammer a struggling server.
-                self._retry_after = time.time() + self._backoff
-                self._backoff = min(self._backoff * 2, self.FAILURE_BACKOFF_MAX)
-                raise
-
-    async def _get_lock(self) -> asyncio.Lock:
-        # Note: this function runs no async code but is marked as async to ensure it's
-        # being run inside the synchronicity event loop and binds the lock to the
-        # correct event loop on Python 3.9 which eagerly assigns event loops on
-        # constructions of locks
-        if self._lock is None:
-            self._lock = asyncio.Lock()
-        return self._lock
+            if exp := self._decode_jwt(resp.token).get("exp"):
+                expiry = float(exp)
+            else:
+                # This should never happen.
+                logger.warning("x-modal-auth-token does not contain exp field")
+                expiry = time.time() + self.DEFAULT_EXPIRY_OFFSET
+            self._token = resp.token
+            self._set_expiry(expiry)
+            self._retry_after = 0.0
+            self._backoff = self.FAILURE_BACKOFF_BASE
+        except AUTH_DENIED_EXCEPTIONS:
+            raise
+        except Exception as e:
+            # Back off (exponentially on consecutive failures) so we don't hammer a struggling server.
+            self._retry_after = time.time() + self._backoff
+            self._backoff = min(self._backoff * 2, self.FAILURE_BACKOFF_MAX)
+            raise
 
     @staticmethod
     def _decode_jwt(token: str) -> dict[str, Any]:
@@ -150,8 +162,20 @@ class _AuthTokenManager:
         # Fetch only when the token is stale/expiring and we're not in a post-failure backoff.
         return self._needs_refresh() and not self._in_backoff()
 
+    def _set_expiry(self, expiry: float) -> None:
+        """Record a new expiry and schedule the proactive refresh a jittered fraction of the way through.
+
+        The delay is measured from the remaining lifetime as this client sees it, so a client whose clock is
+        skewed relative to the issuer still keeps a margin proportional to the lifetime it observes.
+        """
+        now = time.time()
+        self._expiry = expiry
+        remaining = max(expiry - now, 0.0)
+        fraction = self.REFRESH_FRACTION + random.random() * self.REFRESH_JITTER
+        self._refresh_at = now + remaining * fraction
+
     def _needs_refresh(self):
-        return time.time() >= (self._expiry - self.REFRESH_WINDOW)
+        return time.time() >= self._refresh_at
 
     def _is_expired(self):
         return time.time() >= self._expiry

modal/_utils/task_command_router_client.py

@@ -288,11 +288,22 @@ class TaskCommandRouterClient:
         server_client,
         sandbox_id: str,
         task_id: str,
+        access: api_pb2.CommandRouterAccess | None = None,
     ) -> "TaskCommandRouterClient":
-        """Initialize a TaskCommandRouterClient for a V2 sandbox."""
-        resp = await fetch_command_router_access_v2(server_client, sandbox_id=sandbox_id)
+        """Initialize a TaskCommandRouterClient for a V2 sandbox.
+
+        `access` is the router access returned by SandboxCreateV2/SandboxRestoreV2,
+        which lets a freshly scheduled sandbox connect without a round-trip. When it
+        is absent — a re-attached sandbox, or the server could not mint a token — we
+        ask for it.
+        """
+        if access is not None:
+            url, jwt = access.url, access.jwt
+        else:
+            resp = await fetch_command_router_access_v2(server_client, sandbox_id=sandbox_id)
+            url, jwt = resp.url, resp.jwt
         logger.debug(f"Using command router access for sandbox {sandbox_id}")
-        return await cls._connect(server_client, task_id, resp.url, resp.jwt, sandbox_id=sandbox_id)
+        return await cls._connect(server_client, task_id, url, jwt, sandbox_id=sandbox_id)
 
     @classmethod
     async def init_v2_by_task_id(

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.dev8",
+        version: str = "1.5.4.dev10",
     ):
         """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.dev8",
+        version: str = "1.5.4.dev10",
     ):
         """mdmd:hidden
         The Modal client object is not intended to be instantiated directly by users.

modal/sandbox.py

@@ -333,6 +333,7 @@ class _Sandbox(_Object, type_prefix="sb"):
     _tunnels: dict[int, Tunnel] | None
     _enable_snapshot: bool
     _command_router_client: TaskCommandRouterClient | None
+    _init_command_router_access: api_pb2.CommandRouterAccess | None
     _attached: bool
     _filesystem: _SandboxFilesystem | None
     _is_v2: bool = False
@@ -1034,6 +1035,10 @@ class _Sandbox(_Object, type_prefix="sb"):
             self._hydrate(sandbox_id, load_context.client, create_resp.metadata)
             self._is_v2 = True
             self._task_id = create_resp.task_id
+            # An unset proto message reads as an empty one rather than None, so the
+            # presence check is what distinguishes "no access" from "empty access".
+            if create_resp.HasField("command_router_access"):
+                self._init_command_router_access = create_resp.command_router_access
             self._hydrate_metadata_v2()
             # Unencrypted tunnel endpoints are not known at create time. Only
             # cache a complete set so tunnels() fetches the rest otherwise.
@@ -1121,6 +1126,7 @@ class _Sandbox(_Object, type_prefix="sb"):
         self._tunnels = None
         self._enable_snapshot = False
         self._command_router_client = None
+        self._init_command_router_access = None
         self._filesystem = None
         self._is_v2 = _is_v2_sandbox_id(self.object_id)
         if self._is_v2:
@@ -1912,8 +1918,12 @@ class _Sandbox(_Object, type_prefix="sb"):
         if self._command_router_client is None:
             try:
                 if self._is_v2:
+                    # Consume the seeded access, so if connecting with it fails the
+                    # retry falls back to the authoritative RPC instead of
+                    # re-trying the same credentials forever.
+                    access, self._init_command_router_access = self._init_command_router_access, None
                     self._command_router_client = await TaskCommandRouterClient.init_v2_by_sandbox_id(
-                        self._client, self.object_id, task_id
+                        self._client, self.object_id, task_id, access
                     )
                 else:
                     self._command_router_client = await TaskCommandRouterClient.init(self._client, task_id)
@@ -2278,6 +2288,9 @@ class _Sandbox(_Object, type_prefix="sb"):
 
         sandbox = await _Sandbox.from_id(restore_resp.sandbox_id, client)
         sandbox._task_id = restore_resp.task_id
+        # Seeded after from_id, whose hydration resets the field.
+        if restore_resp.HasField("command_router_access"):
+            sandbox._init_command_router_access = restore_resp.command_router_access
         return sandbox
 
     @property

modal/sandbox.pyi

@@ -147,6 +147,7 @@ class _Sandbox(modal._object._Object):
     _tunnels: typing.Optional[dict[int, modal._tunnel.Tunnel]]
     _enable_snapshot: bool
     _command_router_client: typing.Optional[modal._utils.task_command_router_client.TaskCommandRouterClient]
+    _init_command_router_access: typing.Optional[modal_proto.api_pb2.CommandRouterAccess]
     _attached: bool
     _filesystem: typing.Optional[modal.sandbox_fs._SandboxFilesystem]
     _is_v2: bool
@@ -1454,6 +1455,7 @@ class Sandbox(modal.object.Object):
     _tunnels: typing.Optional[dict[int, modal._tunnel.Tunnel]]
     _enable_snapshot: bool
     _command_router_client: typing.Optional[modal._utils.task_command_router_client.TaskCommandRouterClient]
+    _init_command_router_access: typing.Optional[modal_proto.api_pb2.CommandRouterAccess]
     _attached: bool
     _filesystem: typing.Optional[modal.sandbox_fs.SandboxFilesystem]
     _is_v2: bool

{modal-1.5.4.dev8.dist-info → modal-1.5.4.dev10.dist-info}/METADATA

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

{modal-1.5.4.dev8.dist-info → modal-1.5.4.dev10.dist-info}/RECORD

@@ -34,7 +34,7 @@ modal/billing.py,sha256=E1-P_P8bJFLp-bTaggYd4BaaViImPB5UdtXiY8VeAZQ,390
 modal/billing.pyi,sha256=4m7VLYNS8IiJ2I3CJx0awvrOM6BILWRVLfdeqtAYr8M,7189
 modal/call_graph.py,sha256=XGALIDn1bLchWo_goyX23bcPjzCZcfqKUo8INIfDapo,1775
 modal/client.py,sha256=NjpAL0fIKabEaN0uXJPEoI1OTjFtXSp4tSck-M-WF3k,16360
-modal/client.pyi,sha256=9l4Zsaj2DRBywAFp42Px57ByNlXL4XVkWFiUaNpMV1A,17997
+modal/client.pyi,sha256=ibgsT1a5xZKVUot-Zpn_E1GOVWcWn2eQYvXRL1ZuSdM,17999
 modal/cloud_bucket_mount.py,sha256=NrdNV9r9aL3QFrlt-Dhtd7_K3f8v1igrHf2FRwgJ1CI,6827
 modal/cloud_bucket_mount.pyi,sha256=0E8d53pMh2KsFa-rMh_i6LxlzW4cMV8drVmmkPHSqZk,8149
 modal/cls.py,sha256=TxRxHdY4L6AZ6I16DNPgDryAkGtlFadD-Mm0ovtfN58,41206
@@ -76,8 +76,8 @@ modal/retries.py,sha256=tSGB5bGM5xZZUDXMkuelIEoQJY0vOukOssZhwc6tXAo,5014
 modal/runner.py,sha256=oqKyahvUEDpbGcNkxoCtKqFUShKWhoYEPecn7pk-fc4,31090
 modal/runner.pyi,sha256=HjkJBkv3GPIe9occmuQUBRgVuw82vrTL0urM8zV4TaY,9760
 modal/running_app.py,sha256=xSR1VVlhxEFytmjx2x697_0GpD25Vc9crhp6r117_QY,1145
-modal/sandbox.py,sha256=3UdESA3fDGlDOCP6K0QEl7UCmo0t204ahMrMiyHZyTE,125106
-modal/sandbox.pyi,sha256=GO6ADS5GLluBVTvsm78vyVlK7Oyve0aMu1nIxgRbmbQ,146442
+modal/sandbox.py,sha256=p-QgzNALkRb0q_qRc-K28HpgAe1LRyomc3URd3QsRBY,126078
+modal/sandbox.pyi,sha256=PXlInwHCuZYNXkRO0AKMpX3yKTAmhliN-ajiXltcsb0,146622
 modal/sandbox_fs.py,sha256=GSuPlrLSdQ7fG7c_TGMd-VWs_s6RqQxMvLivdvC6s4s,27002
 modal/sandbox_fs.pyi,sha256=TJWx8h7x3yRC7KUxeF7q62_0a7w1IHufnJL7AYEojAg,45285
 modal/schedule.py,sha256=J0rpNAVcoQjCCzobYFQXHXv1BBPLYATUwGS5prCHp4c,3242
@@ -101,7 +101,7 @@ modal/workspace.pyi,sha256=aB3LCYnbzD8eyaBMsEfMeMiSHOUdpTdio4rK6nsjHUU,24215
 modal/_output/__init__.py,sha256=xtGN6SPv2zvByrJ32JPao0yJKpxyCIZ-2JMTScrisuQ,28
 modal/_output/manager.py,sha256=lL9fK9_cSJHdVIOcgzjCZEJ90fgNB7_mO7ih_4sk1hI,18097
 modal/_output/pty.py,sha256=YIrNGH9kxJr8TFbIMwPnK3Wvh2rECyGYebx4TOqNWhk,10296
-modal/_output/rich.py,sha256=6QsXXD-LK0sxd5wiLQS4-sy-pgtBKvAV92flzAmAwKo,31864
+modal/_output/rich.py,sha256=K6dUEqeeyytqUR-78eiS1GNmvuFW7HuKhws_41pjCRQ,32205
 modal/_output/status.py,sha256=c0rdvjoYKtIG7XJqq42TlcKawkYqZWtod38RmRY5IWA,4492
 modal/_runtime/__init__.py,sha256=MIEP8jhXUeGq_eCjYFcqN5b1bxBM4fdk0VESpjWR0fc,28
 modal/_runtime/asgi.py,sha256=eDNPsmV6SXl0CwUqgy7yrFDaPdlx4NiWTScXpCTlBFQ,22805
@@ -118,7 +118,7 @@ modal/_runtime/user_code_imports.py,sha256=5MgewJzAeZqop47FQyB-lEpHBfC50ApHgSb86
 modal/_utils/__init__.py,sha256=waLjl5c6IPDhSsdWAm9Bji4e2PVxamYABKAze6CHVXY,28
 modal/_utils/app_utils.py,sha256=88BT4TPLWfYAQwKTHcyzNQRHg8n9B-QE2UyJs96iV-0,108
 modal/_utils/async_utils.py,sha256=2pBzV_ErMH055TGxT0re2szYUN6k8X94RkLwIp_JbZw,46393
-modal/_utils/auth_token_manager.py,sha256=_XAYZsMDgtRqVeyNBjcr88pMnZbg49n8D8wxDhqLltg,7335
+modal/_utils/auth_token_manager.py,sha256=qhnbTv6EO5_BxvUUlyCrNt8sQaUIZ77CdSrzBH9sLaU,8563
 modal/_utils/blob_utils.py,sha256=KyqXWF82PdfA6pvEu99QwGfSqyHQxPUvjyi760GXkdY,26242
 modal/_utils/browser_utils.py,sha256=Yqq0qU4Id6oaDOdeT1JEmbCN7vj4baNJJraJhRWU8ZI,1085
 modal/_utils/bytes_io_segment_payload.py,sha256=2OrIrjrsJR38a9D7JCITLqGvIPaiWViTQhxtcZlFn_M,4328
@@ -142,7 +142,7 @@ modal/_utils/proxy_support.py,sha256=hLA_XojYxzPysOiLqyUbB-SgMLmn_X8WtXTk92xh7g4
 modal/_utils/rand_pb_testing.py,sha256=kwAKIros_g6XPXluiJAktQ79YPkD9SGopKGl83GQ1l0,4361
 modal/_utils/sandbox_fs_utils.py,sha256=rsfo1uDbVSoiY21XERDzsmp-DALcNmlPfJ6jn_2p5T0,15680
 modal/_utils/shell_utils.py,sha256=_2yHrlx4Crx5JWQXECLPQ51kclu8H-U3N7jOkBN7q5U,3390
-modal/_utils/task_command_router_client.py,sha256=QTIkgXqSFHi6sO2rhRyAk1TT5XeCISbq7nY93lWW1NY,45599
+modal/_utils/task_command_router_client.py,sha256=QJxxzceZDrr9BEDVs524tZ17tgO3wEVDw-4gF8-P96U,46073
 modal/_utils/time_utils.py,sha256=JaFyW8y4iimzhrGT5KlQmlXH4oMZrZOGCtUBZAV-gfU,10965
 modal/_vendor/__init__.py,sha256=MIEP8jhXUeGq_eCjYFcqN5b1bxBM4fdk0VESpjWR0fc,28
 modal/_vendor/a2wsgi_wsgi.py,sha256=RVLIYVJA8a8o96QxEXYT5tuog2zev2S24XOuAVTXZAs,21872
@@ -198,7 +198,7 @@ modal/experimental/flash.py,sha256=GAJ1c-EGL8M3_loAkqr5gdUfySMHY3U_oc6RywfZ8-0,3
 modal/experimental/flash.pyi,sha256=lpTqJFZRMGWscw0xpYZVdBMnAYjj16asXE4_64A6itQ,17774
 modal/experimental/ipython.py,sha256=kJ6o9wkAY_MiSb7paewI7f1MJAeqdi8TX5pBdhr_mUI,2925
 modal/skills/modal/SKILL.md,sha256=WzLjYJ1utjc_ZOf6qZTPwB6wxxwRYB5PibdvIf_FLy0,5316
-modal-1.5.4.dev8.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
+modal-1.5.4.dev10.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
 modal_docs/__init__.py,sha256=svYKtV8HDwDCN86zbdWqyq5T8sMdGDj0PVlzc2tIxDM,28
 modal_docs/gen_cli_docs.py,sha256=9-b8vl2EsLWD9zJqpCK8IUi5h44Bjn3i3rSdedZV4bA,6415
 modal_docs/gen_cli_docs_main.py,sha256=UPGr7O1DYTIRLwfINe2rbdh89J3cPVTU1OWL-SjxnOQ,198
@@ -209,23 +209,23 @@ modal_docs/mdmd/__init__.py,sha256=svYKtV8HDwDCN86zbdWqyq5T8sMdGDj0PVlzc2tIxDM,2
 modal_docs/mdmd/mdmd.py,sha256=J49V8ak-hnF7SyBJDc-p3a_aEZpaAImkz2Rf74mDvA4,25070
 modal_docs/mdmd/signatures.py,sha256=OASroKle763Kfumn4Ek4PABJXEGP56JMCTug44QoBx4,11327
 modal_docs/mdmd/types.py,sha256=tv7fvziPr-MeDzOp4e2pK5h3d1DXTUOO74rPqDoBn6g,736
-modal_proto/__init__.py,sha256=tFdSv9eaHBvYcwhgq-XqIlYDNDw57clrvO91Fb-oUV4,130
-modal_proto/api_grpc.py,sha256=zFY9HEa330mx7P8-b4bQexf5HkEbOqdnymeYO0SN7Po,182439
-modal_proto/api_pb2.py,sha256=fCCrs6Bc-kFT0FJ0Alja_9Qa-3jrVhwBHgvBdUQRSiA,498800
-modal_proto/api_pb2.pyi,sha256=ykSac1gwUcpCLGxOAPZKXDI493PHxwSLNUBjQRJM8bQ,687971
-modal_proto/api_pb2_grpc.py,sha256=CU5p-AoluXydNZsQva4WPolS2caboTbxD-jINzWJqdE,391633
-modal_proto/api_pb2_grpc.pyi,sha256=pWvTEJqn6kW_sFQ6lzw_G08E_T63XMuA-I-vO4avDus,91568
-modal_proto/modal_api_grpc.py,sha256=j0WxD1iNwLzvm2bS4KAc2OTN4QnXZ9_672oGA227P40,28771
+modal_proto/__init__.py,sha256=Hkou4lcD5tArEkW_9v9KZhLY7fl0fA9-O28beKU55pg,396
+modal_proto/api_grpc.py,sha256=T7G1w30VHACC466T9WgEuAUKzk1T55K4xhh9yrx30jE,184199
+modal_proto/api_pb2.py,sha256=Cv-bAV11PEAgWXYeQHcLBgmjvgjDvwR1MrrdT7hNXr0,503011
+modal_proto/api_pb2.pyi,sha256=RrSAYQpmuc2pCz3sF3zbTcmC7R1oOppakVUdwm-IKOs,695248
+modal_proto/api_pb2_grpc.py,sha256=iVzwMoa9O8vZKvPX4gCFq3f5QYTMpHFsLa-mEc-Zwis,395203
+modal_proto/api_pb2_grpc.pyi,sha256=Clzhm_sI17aGhz31Rtl2MpCdfeM_8PlGu2O4aH92fbk,92749
+modal_proto/modal_api_grpc.py,sha256=36yBgbLRHAKnchTZlRHVQyYWbFTTc_xrx6j8JHz1Ozo,29059
 modal_proto/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 modal_proto/task_command_router_grpc.py,sha256=UurJEgLeAc2CCdEsCOFE9DgIQy1ra6bXHXJLNBK81Mg,21284
 modal_proto/task_command_router_pb2.py,sha256=4cyDT_uPGi54blsO663bnSzUe7aBFqi9F4iM8N1JPaU,41130
 modal_proto/task_command_router_pb2.pyi,sha256=5xfhnwtTDTmJo1Nv0tPBJrYzb0aeilY2Lm92wWT_6yw,46211
 modal_proto/task_command_router_pb2_grpc.py,sha256=zuFkJgKcswB2ftjnyEgs70dp61VMA7gK0h0uzR3gYFw,42563
 modal_proto/task_command_router_pb2_grpc.pyi,sha256=UjAJFqz9DpOFXP04fVm5lB2eAyTtjOg3oKjfWei8yLU,14262
-modal_version/__init__.py,sha256=KhzhvJODlTPfdzPYDv79P-JU9OqeyywBlvigSELIBg4,120
+modal_version/__init__.py,sha256=m4zi0JLZ3JoNiMRuYVI2kPLAjBX0EfNFfw7dstre8IU,121
 modal_version/__main__.py,sha256=nLUbubxKx7GXNzJIkTB7M7_u8080LyLM0IS0rzFkC_c,119
-modal-1.5.4.dev8.dist-info/METADATA,sha256=oOydWAb-YD50rDlYo1Op1Tu3jzMF74jSXYayUax-nEw,2512
-modal-1.5.4.dev8.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-modal-1.5.4.dev8.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
-modal-1.5.4.dev8.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
-modal-1.5.4.dev8.dist-info/RECORD,,
+modal-1.5.4.dev10.dist-info/METADATA,sha256=4c3UGSDeWIAxuzAsfIuIYr7TtmJxI0aTnsOd4uOkzYw,2513
+modal-1.5.4.dev10.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+modal-1.5.4.dev10.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
+modal-1.5.4.dev10.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
+modal-1.5.4.dev10.dist-info/RECORD,,

modal_proto/__init__.py

@@ -1,4 +1,11 @@
 # Copyright Modal Labs 2024
+"""Modal gRPC types for Python.
+
+Warning: direct usage of Modal's gRPC API is discouraged,
+and no support or compatibility guarantees are provided.
+Message fields or entire RPCs may be changed or removed
+without notice. We recommend using official SDKs instead.
+"""
 from pkgutil import extend_path
 
 __path__ = extend_path(__path__, __name__)  # type: ignore[has-type]

modal_proto/api_grpc.py

@@ -123,6 +123,10 @@ class ModalClientBase(abc.ABC):
     async def AuthTokenGet(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.AuthTokenGetRequest, modal_proto.api_pb2.AuthTokenGetResponse]') -> None:
         pass
 
+    @abc.abstractmethod
+    async def AutoscalerOverrideHistory(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.AutoscalerOverrideHistoryRequest, modal_proto.api_pb2.AutoscalerOverrideHistoryResponse]') -> None:
+        pass
+
     @abc.abstractmethod
     async def BlobCreate(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.BlobCreateRequest, modal_proto.api_pb2.BlobCreateResponse]') -> None:
         pass
@@ -327,6 +331,10 @@ class ModalClientBase(abc.ABC):
     async def EnvironmentSetBudget(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.EnvironmentSetBudgetRequest, google.protobuf.empty_pb2.Empty]') -> None:
         pass
 
+    @abc.abstractmethod
+    async def EnvironmentSetDefaultMemberRole(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.EnvironmentSetDefaultMemberRoleRequest, google.protobuf.empty_pb2.Empty]') -> None:
+        pass
+
     @abc.abstractmethod
     async def EnvironmentSetManaged(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.EnvironmentSetManagedRequest, google.protobuf.empty_pb2.Empty]') -> None:
         pass
@@ -1113,6 +1121,12 @@ class ModalClientBase(abc.ABC):
                 modal_proto.api_pb2.AuthTokenGetRequest,
                 modal_proto.api_pb2.AuthTokenGetResponse,
             ),
+            '/modal.client.ModalClient/AutoscalerOverrideHistory': grpclib.const.Handler(
+                self.AutoscalerOverrideHistory,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.api_pb2.AutoscalerOverrideHistoryRequest,
+                modal_proto.api_pb2.AutoscalerOverrideHistoryResponse,
+            ),
             '/modal.client.ModalClient/BlobCreate': grpclib.const.Handler(
                 self.BlobCreate,
                 grpclib.const.Cardinality.UNARY_UNARY,
@@ -1419,6 +1433,12 @@ class ModalClientBase(abc.ABC):
                 modal_proto.api_pb2.EnvironmentSetBudgetRequest,
                 google.protobuf.empty_pb2.Empty,
             ),
+            '/modal.client.ModalClient/EnvironmentSetDefaultMemberRole': grpclib.const.Handler(
+                self.EnvironmentSetDefaultMemberRole,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.api_pb2.EnvironmentSetDefaultMemberRoleRequest,
+                google.protobuf.empty_pb2.Empty,
+            ),
             '/modal.client.ModalClient/EnvironmentSetManaged': grpclib.const.Handler(
                 self.EnvironmentSetManaged,
                 grpclib.const.Cardinality.UNARY_UNARY,
@@ -2523,6 +2543,12 @@ class ModalClientStub:
             modal_proto.api_pb2.AuthTokenGetRequest,
             modal_proto.api_pb2.AuthTokenGetResponse,
         )
+        self.AutoscalerOverrideHistory = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.client.ModalClient/AutoscalerOverrideHistory',
+            modal_proto.api_pb2.AutoscalerOverrideHistoryRequest,
+            modal_proto.api_pb2.AutoscalerOverrideHistoryResponse,
+        )
         self.BlobCreate = grpclib.client.UnaryUnaryMethod(
             channel,
             '/modal.client.ModalClient/BlobCreate',
@@ -2829,6 +2855,12 @@ class ModalClientStub:
             modal_proto.api_pb2.EnvironmentSetBudgetRequest,
             google.protobuf.empty_pb2.Empty,
         )
+        self.EnvironmentSetDefaultMemberRole = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.client.ModalClient/EnvironmentSetDefaultMemberRole',
+            modal_proto.api_pb2.EnvironmentSetDefaultMemberRoleRequest,
+            google.protobuf.empty_pb2.Empty,
+        )
         self.EnvironmentSetManaged = grpclib.client.UnaryUnaryMethod(
             channel,
             '/modal.client.ModalClient/EnvironmentSetManaged',