modAL 1.5.4.dev9__py3-none-any.whl → 1.5.4.dev11__py3-none-any.whl

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

@@ -292,9 +292,10 @@ class TaskCommandRouterClient:
     ) -> "TaskCommandRouterClient":
         """Initialize a TaskCommandRouterClient for a V2 sandbox.
 
-        `access` is the router access returned by SandboxCreateV2, which lets a
-        freshly created 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.
+        `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

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

modal/sandbox.py

@@ -2288,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-1.5.4.dev9.dist-info → modal-1.5.4.dev11.dist-info}/METADATA

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

{modal-1.5.4.dev9.dist-info → modal-1.5.4.dev11.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=XKsjCTpM6Q265diR31Ob1HVD6pkl_4rJ9_AXAQQ2NxA,17997
+modal/client.pyi,sha256=QFNaYNlNXJCAo7pTbX4AgqHCxU5VOx_qpUcuCBHYI3E,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,7 +76,7 @@ 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=BLpMLDRF39a3FmY_Eusg_ejvW6yCsjHJY0hNsWPRko4,125868
+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
@@ -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=q0MyeG5Q6diOSkN89FIg_uODQ1Ix2C4IrOvBYh2fOq0,46046
+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.dev9.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
+modal-1.5.4.dev11.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=J8CYJap2EuMDb1ct4beDGBxfN5d6uyJHTbdVySIZVrE,183310
-modal_proto/api_pb2.py,sha256=DU9_PKaAnqpEhWPwNtMi7V4feZ6niWhycTnBN1uevgU,500582
-modal_proto/api_pb2.pyi,sha256=glpbAVDbXk0M7j4C4dfTsgnpsmN4z1tS_Mn9fHPcoU8,692212
-modal_proto/api_pb2_grpc.py,sha256=nTPxiGWnzwoI5CDwOzQBs-YNm2MbbR4hY-0k-1O4q54,393438
-modal_proto/api_pb2_grpc.pyi,sha256=e2XqayYGQCGmq3oy2Oy4nT29vZZ3wZYSJeZS2OZ-IEM,91991
-modal_proto/modal_api_grpc.py,sha256=brzTrJA6gH0678l8r4BF7Uj6XNcw-4PDorV0AYfs4QA,28921
+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=nwgaaUSs19Z9kxmj-Qv_8W5K88StfjfUwGbnX6TZZto,120
+modal_version/__init__.py,sha256=f197D89MvvrXc5P_dA9HTULCibAz73m15IoOoWlbd3A,121
 modal_version/__main__.py,sha256=nLUbubxKx7GXNzJIkTB7M7_u8080LyLM0IS0rzFkC_c,119
-modal-1.5.4.dev9.dist-info/METADATA,sha256=fehlLhzMf8xnGjFBE5lyi_od-Aykmk_lnntqWkpn1Rw,2512
-modal-1.5.4.dev9.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-modal-1.5.4.dev9.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
-modal-1.5.4.dev9.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
-modal-1.5.4.dev9.dist-info/RECORD,,
+modal-1.5.4.dev11.dist-info/METADATA,sha256=Zebs41aU_dhmL3bh94SKRb1jqPY9pnIwnlZiXsJkf88,2513
+modal-1.5.4.dev11.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+modal-1.5.4.dev11.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
+modal-1.5.4.dev11.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
+modal-1.5.4.dev11.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
@@ -1117,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,
@@ -2533,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',