modAL 1.4.2.dev13__py3-none-any.whl → 1.4.2.dev14__py3-none-any.whl

modal/client.pyi

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

modal/sandbox.py

@@ -1110,7 +1110,12 @@ class _Sandbox(_Object, type_prefix="sb"):
 
         req = api_pb2.SandboxGetTunnelsRequest(sandbox_id=self.object_id, timeout=timeout)
         stub = self._client.stub
-        resp = await (stub.SandboxGetTunnelsV2(req) if self._is_v2 else stub.SandboxGetTunnels(req))
+        if self._is_v2:
+            assert self._client._auth_token_manager
+            auth_token = await self._client._auth_token_manager.get_token()
+            resp = await stub.SandboxGetTunnelsV2(req, metadata=[("x-modal-auth-token", auth_token)])
+        else:
+            resp = await stub.SandboxGetTunnels(req)
 
         # If we couldn't get the tunnels in time, report the timeout.
         if resp.result.status == api_pb2.GenericResult.GENERIC_STATUS_TIMEOUT:
@@ -1179,7 +1184,12 @@ class _Sandbox(_Object, type_prefix="sb"):
         This is a no-op if the Sandbox has already finished running."""
         req = api_pb2.SandboxTerminateRequest(sandbox_id=self.object_id)
         stub = self._client.stub
-        await (stub.SandboxTerminateV2(req) if self._is_v2 else stub.SandboxTerminate(req))
+        if self._is_v2:
+            assert self._client._auth_token_manager
+            auth_token = await self._client._auth_token_manager.get_token()
+            await stub.SandboxTerminateV2(req, metadata=[("x-modal-auth-token", auth_token)])
+        else:
+            await stub.SandboxTerminate(req)
         if wait:
             await self.wait(raise_on_termination=False)
             return self.returncode
@@ -1208,7 +1218,12 @@ class _Sandbox(_Object, type_prefix="sb"):
         while not self._task_id:
             req = api_pb2.SandboxGetTaskIdRequest(sandbox_id=self.object_id)
             stub = self._client.stub
-            resp = await (stub.SandboxGetTaskIdV2(req) if self._is_v2 else stub.SandboxGetTaskId(req))
+            if self._is_v2:
+                assert self._client._auth_token_manager
+                auth_token = await self._client._auth_token_manager.get_token()
+                resp = await stub.SandboxGetTaskIdV2(req, metadata=[("x-modal-auth-token", auth_token)])
+            else:
+                resp = await stub.SandboxGetTaskId(req)
             if not resp.task_id and raise_if_task_complete and resp.HasField("task_result"):
                 msg = resp.task_result.exception or "Sandbox already finished"
                 raise Error(msg)

{modal-1.4.2.dev13.dist-info → modal-1.4.2.dev14.dist-info}/METADATA

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

{modal-1.4.2.dev13.dist-info → modal-1.4.2.dev14.dist-info}/RECORD

@@ -26,7 +26,7 @@ modal/app.pyi,sha256=mysbOCD6OjZi938NeMY40RpdpQbnI43Bb4s5P79x5OQ,53773
 modal/billing.py,sha256=B2zXa7KeMnpqi2AQINycEk-KSXVYaLTTTAQTtuf2Gfs,257
 modal/call_graph.py,sha256=1g2DGcMIJvRy-xKicuf63IVE98gJSnQsr8R_NVMptNc,2581
 modal/client.py,sha256=jjT4eLFHGcvIF6Xw4kPptIx-VmExtphRN8XJ1JA7fWI,15502
-modal/client.pyi,sha256=zoxiLuMLdCY5Hawmn572y6hFjf65qDcfRz4e5sTv9SI,15703
+modal/client.pyi,sha256=gavLYn7MPjE5gpZ2Min3eHBppoosAogpNIuFkFBAJkk,15703
 modal/cloud_bucket_mount.py,sha256=KkaJVy5vpO_u2NtDYtv1JM-M3y4J0UtWBapTDKpQRcA,6034
 modal/cloud_bucket_mount.pyi,sha256=mGDeEd1XvxGpCE6O5jRBSfIvMXvZ5AtIybea5k-0w9w,7560
 modal/cls.py,sha256=Yrl5b_tQhZMhc5avInM-i5UlGuqeog6Gs4rX2d9lLjI,40390
@@ -68,7 +68,7 @@ modal/retries.py,sha256=vPWWKP-FRolJwTBuLPXdWqHV7aJfSQFgEW7vhlBBXRQ,5067
 modal/runner.py,sha256=yIB1PE5EGdvLWPPymyT4P0KYliREnFf-MAtULl3tVN0,30062
 modal/runner.pyi,sha256=ySefjmlMQ5FsvJgJkCWiUZKvcPNfEqdB5hwxI1fGtDo,9445
 modal/running_app.py,sha256=v61mapYNV1-O-Uaho5EfJlryMLvIT9We0amUOSvSGx8,1188
-modal/sandbox.py,sha256=LayePtgoare1fNR9DnGsYyBELz75b1FP5DB8jcPAkvw,81865
+modal/sandbox.py,sha256=NTsVSS2XZNqz1A50KTnVq1XNeKVFsoH0aQJwFasF-AQ,82538
 modal/sandbox.pyi,sha256=YBf_dPM23kwYL9cBQyIsNTYw7rsEo3nyWckKemLXc8I,85634
 modal/sandbox_fs.py,sha256=pNlfii0vrCOBwhTEuT3Uzrp_f-0e6XZSx2sjtXnnPCc,15026
 modal/sandbox_fs.pyi,sha256=HEZnC31doeUdJMrdQYcAWGvbUGS2Y20do4heBN03pfg,23801
@@ -173,7 +173,7 @@ modal/experimental/__init__.py,sha256=Wb1YoAKAnNxH_0c2wA0T12I7bSu9PSO2nS_LuLska7
 modal/experimental/flash.py,sha256=ATGhVO8AleJ-TOe2PGolI5T7zfCEtj0kl_ivMPp7Rqg,33362
 modal/experimental/flash.pyi,sha256=R4h2nQl7-oiwqq_twCp0YqID_LnPjPGqxywXvxbiLIk,17011
 modal/experimental/ipython.py,sha256=TrCfmol9LGsRZMeDoeMPx3Hv3BFqQhYnmD_iH0pqdhk,2904
-modal-1.4.2.dev13.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
+modal-1.4.2.dev14.dist-info/licenses/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
 modal_docs/__init__.py,sha256=svYKtV8HDwDCN86zbdWqyq5T8sMdGDj0PVlzc2tIxDM,28
 modal_docs/gen_cli_docs.py,sha256=5sVABw4uGUqPXjcWT0Nqjkpe5KLs89C3tFnla58-D0M,3756
 modal_docs/gen_cli_docs_main.py,sha256=UPGr7O1DYTIRLwfINe2rbdh89J3cPVTU1OWL-SjxnOQ,198
@@ -183,22 +183,22 @@ modal_docs/mdmd/__init__.py,sha256=svYKtV8HDwDCN86zbdWqyq5T8sMdGDj0PVlzc2tIxDM,2
 modal_docs/mdmd/mdmd.py,sha256=z9T3lMktTq3N4r6T6MSSyw8lebbqKc3Igw8l9EdMJNM,9998
 modal_docs/mdmd/signatures.py,sha256=XJaZrK7Mdepk5fdX51A8uENiLFNil85Ud0d4MH8H5f0,3218
 modal_proto/__init__.py,sha256=MIEP8jhXUeGq_eCjYFcqN5b1bxBM4fdk0VESpjWR0fc,28
-modal_proto/api_grpc.py,sha256=XRW6PS7TAMwsI90sbW1GH8aa2wft0ZA4AeDrOMH3ULk,149205
-modal_proto/api_pb2.py,sha256=t4SUnufDsMzehOaVyGEB8gInXBGXe8TPInos--Bpf3I,414280
-modal_proto/api_pb2.pyi,sha256=4KAFG4bWAKFuI-o6itDmf_rKrlebx0hW-hqLkRQd9qU,580502
-modal_proto/api_pb2_grpc.py,sha256=Li6NnZt0kW4DIDOYecagfGaImcVkQBmGOgIgwK9GvGQ,321425
-modal_proto/api_pb2_grpc.pyi,sha256=IQEuD26-FJAAs4rvWQQbv6E_-wriUSod1OCbybn-uVE,75061
-modal_proto/modal_api_grpc.py,sha256=dFGRDQp8kmoh3Bl6e-rkiAzchAwQhTvu-xne9UViiPQ,23547
+modal_proto/api_grpc.py,sha256=H4Z21x81592KlGzgDIcgltk5WixY5vs5EyOH8Q89wvA,153140
+modal_proto/api_pb2.py,sha256=aBiF3ao8juCQ37cS0PnauYI6Y3_bSfwt3RL-6fEVb7E,423723
+modal_proto/api_pb2.pyi,sha256=nrxMqbHx9VNhq4wOv8jqgMLIGL4f0WKNj57wXxIMvF4,592885
+modal_proto/api_pb2_grpc.py,sha256=dWnojBtfuaqHvBos1FL-JMNEB7LsNHM1UGUpKXO5IFY,329867
+modal_proto/api_pb2_grpc.pyi,sha256=qDqqKIs18olWaoBjLOJ8Z6Gn8_GjpNn3Z1Pw1yG_M4I,77024
+modal_proto/modal_api_grpc.py,sha256=bBPhTehvCriGANMHxXNxKa58ew_2QbSvO88zhKCr4CE,24177
 modal_proto/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 modal_proto/task_command_router_grpc.py,sha256=wbRjiCyx9MlvkmPKM_6UVU1KmJaynUO4Eyt6tDKtmRk,12572
 modal_proto/task_command_router_pb2.py,sha256=Kadayms1M02hoAtV4hom00t143GAHxUNvohOxTIRaCg,24322
 modal_proto/task_command_router_pb2.pyi,sha256=nnMXJQ81eSbgeA0SkBOcUHXiD5hRknYi5MK7wVkPlU0,27068
 modal_proto/task_command_router_pb2_grpc.py,sha256=CW52Dj00YLRRAc2-KIz_aimGe7RuKvx0cROMs2No-b0,25002
 modal_proto/task_command_router_pb2_grpc.pyi,sha256=j4-VU3QFp8nUyx5_6F04T4IIU28KWHpAFMnY8Eg521g,8018
-modal_version/__init__.py,sha256=i3yyCvfu1TL4UrZ52wJhan-DxclsalsHrE9B4QZ9DO8,121
+modal_version/__init__.py,sha256=SGJgmgTsGVJ0OJHpDmLDQrZP1D9VFYeIxyTVhgRdLBU,121
 modal_version/__main__.py,sha256=nLUbubxKx7GXNzJIkTB7M7_u8080LyLM0IS0rzFkC_c,119
-modal-1.4.2.dev13.dist-info/METADATA,sha256=3WAulmtrP2DabERweC85FgtFoKiODaGPKRAz4FAZFoU,2365
-modal-1.4.2.dev13.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-modal-1.4.2.dev13.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
-modal-1.4.2.dev13.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
-modal-1.4.2.dev13.dist-info/RECORD,,
+modal-1.4.2.dev14.dist-info/METADATA,sha256=7hlQWy82I3NNj0vuscIfVNuOIL0Jh_UnGLBWPbpcyKQ,2365
+modal-1.4.2.dev14.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+modal-1.4.2.dev14.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
+modal-1.4.2.dev14.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
+modal-1.4.2.dev14.dist-info/RECORD,,

modal_proto/api_grpc.py

@@ -267,6 +267,10 @@ class ModalClientBase(abc.ABC):
     async def EnvironmentDelete(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.EnvironmentDeleteRequest, google.protobuf.empty_pb2.Empty]') -> None:
         pass
 
+    @abc.abstractmethod
+    async def EnvironmentGetManaged(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.EnvironmentGetManagedRequest, modal_proto.api_pb2.EnvironmentGetManagedResponse]') -> None:
+        pass
+
     @abc.abstractmethod
     async def EnvironmentGetOrCreate(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.EnvironmentGetOrCreateRequest, modal_proto.api_pb2.EnvironmentGetOrCreateResponse]') -> None:
         pass
@@ -275,6 +279,14 @@ class ModalClientBase(abc.ABC):
     async def EnvironmentList(self, stream: 'grpclib.server.Stream[google.protobuf.empty_pb2.Empty, modal_proto.api_pb2.EnvironmentListResponse]') -> None:
         pass
 
+    @abc.abstractmethod
+    async def EnvironmentRoleSet(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.EnvironmentRoleSetRequest, 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
+
     @abc.abstractmethod
     async def EnvironmentUpdate(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.EnvironmentUpdateRequest, modal_proto.api_pb2.EnvironmentListItem]') -> None:
         pass
@@ -623,6 +635,10 @@ class ModalClientBase(abc.ABC):
     async def SecretUpdate(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.SecretUpdateRequest, google.protobuf.empty_pb2.Empty]') -> None:
         pass
 
+    @abc.abstractmethod
+    async def ServiceUserList(self, stream: 'grpclib.server.Stream[google.protobuf.empty_pb2.Empty, modal_proto.api_pb2.ServiceUserListResponse]') -> None:
+        pass
+
     @abc.abstractmethod
     async def SharedVolumeDelete(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.SharedVolumeDeleteRequest, google.protobuf.empty_pb2.Empty]') -> None:
         pass
@@ -787,6 +803,10 @@ class ModalClientBase(abc.ABC):
     async def WorkspaceDashboardUrlGet(self, stream: 'grpclib.server.Stream[modal_proto.api_pb2.WorkspaceDashboardUrlRequest, modal_proto.api_pb2.WorkspaceDashboardUrlResponse]') -> None:
         pass
 
+    @abc.abstractmethod
+    async def WorkspaceMembersList(self, stream: 'grpclib.server.Stream[google.protobuf.empty_pb2.Empty, modal_proto.api_pb2.WorkspaceMembersListResponse]') -> None:
+        pass
+
     @abc.abstractmethod
     async def WorkspaceNameLookup(self, stream: 'grpclib.server.Stream[google.protobuf.empty_pb2.Empty, modal_proto.api_pb2.WorkspaceNameLookupResponse]') -> None:
         pass
@@ -1165,6 +1185,12 @@ class ModalClientBase(abc.ABC):
                 modal_proto.api_pb2.EnvironmentDeleteRequest,
                 google.protobuf.empty_pb2.Empty,
             ),
+            '/modal.client.ModalClient/EnvironmentGetManaged': grpclib.const.Handler(
+                self.EnvironmentGetManaged,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.api_pb2.EnvironmentGetManagedRequest,
+                modal_proto.api_pb2.EnvironmentGetManagedResponse,
+            ),
             '/modal.client.ModalClient/EnvironmentGetOrCreate': grpclib.const.Handler(
                 self.EnvironmentGetOrCreate,
                 grpclib.const.Cardinality.UNARY_UNARY,
@@ -1177,6 +1203,18 @@ class ModalClientBase(abc.ABC):
                 google.protobuf.empty_pb2.Empty,
                 modal_proto.api_pb2.EnvironmentListResponse,
             ),
+            '/modal.client.ModalClient/EnvironmentRoleSet': grpclib.const.Handler(
+                self.EnvironmentRoleSet,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.api_pb2.EnvironmentRoleSetRequest,
+                google.protobuf.empty_pb2.Empty,
+            ),
+            '/modal.client.ModalClient/EnvironmentSetManaged': grpclib.const.Handler(
+                self.EnvironmentSetManaged,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                modal_proto.api_pb2.EnvironmentSetManagedRequest,
+                google.protobuf.empty_pb2.Empty,
+            ),
             '/modal.client.ModalClient/EnvironmentUpdate': grpclib.const.Handler(
                 self.EnvironmentUpdate,
                 grpclib.const.Cardinality.UNARY_UNARY,
@@ -1699,6 +1737,12 @@ class ModalClientBase(abc.ABC):
                 modal_proto.api_pb2.SecretUpdateRequest,
                 google.protobuf.empty_pb2.Empty,
             ),
+            '/modal.client.ModalClient/ServiceUserList': grpclib.const.Handler(
+                self.ServiceUserList,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                google.protobuf.empty_pb2.Empty,
+                modal_proto.api_pb2.ServiceUserListResponse,
+            ),
             '/modal.client.ModalClient/SharedVolumeDelete': grpclib.const.Handler(
                 self.SharedVolumeDelete,
                 grpclib.const.Cardinality.UNARY_UNARY,
@@ -1945,6 +1989,12 @@ class ModalClientBase(abc.ABC):
                 modal_proto.api_pb2.WorkspaceDashboardUrlRequest,
                 modal_proto.api_pb2.WorkspaceDashboardUrlResponse,
             ),
+            '/modal.client.ModalClient/WorkspaceMembersList': grpclib.const.Handler(
+                self.WorkspaceMembersList,
+                grpclib.const.Cardinality.UNARY_UNARY,
+                google.protobuf.empty_pb2.Empty,
+                modal_proto.api_pb2.WorkspaceMembersListResponse,
+            ),
             '/modal.client.ModalClient/WorkspaceNameLookup': grpclib.const.Handler(
                 self.WorkspaceNameLookup,
                 grpclib.const.Cardinality.UNARY_UNARY,
@@ -2329,6 +2379,12 @@ class ModalClientStub:
             modal_proto.api_pb2.EnvironmentDeleteRequest,
             google.protobuf.empty_pb2.Empty,
         )
+        self.EnvironmentGetManaged = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.client.ModalClient/EnvironmentGetManaged',
+            modal_proto.api_pb2.EnvironmentGetManagedRequest,
+            modal_proto.api_pb2.EnvironmentGetManagedResponse,
+        )
         self.EnvironmentGetOrCreate = grpclib.client.UnaryUnaryMethod(
             channel,
             '/modal.client.ModalClient/EnvironmentGetOrCreate',
@@ -2341,6 +2397,18 @@ class ModalClientStub:
             google.protobuf.empty_pb2.Empty,
             modal_proto.api_pb2.EnvironmentListResponse,
         )
+        self.EnvironmentRoleSet = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.client.ModalClient/EnvironmentRoleSet',
+            modal_proto.api_pb2.EnvironmentRoleSetRequest,
+            google.protobuf.empty_pb2.Empty,
+        )
+        self.EnvironmentSetManaged = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.client.ModalClient/EnvironmentSetManaged',
+            modal_proto.api_pb2.EnvironmentSetManagedRequest,
+            google.protobuf.empty_pb2.Empty,
+        )
         self.EnvironmentUpdate = grpclib.client.UnaryUnaryMethod(
             channel,
             '/modal.client.ModalClient/EnvironmentUpdate',
@@ -2863,6 +2931,12 @@ class ModalClientStub:
             modal_proto.api_pb2.SecretUpdateRequest,
             google.protobuf.empty_pb2.Empty,
         )
+        self.ServiceUserList = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.client.ModalClient/ServiceUserList',
+            google.protobuf.empty_pb2.Empty,
+            modal_proto.api_pb2.ServiceUserListResponse,
+        )
         self.SharedVolumeDelete = grpclib.client.UnaryUnaryMethod(
             channel,
             '/modal.client.ModalClient/SharedVolumeDelete',
@@ -3109,6 +3183,12 @@ class ModalClientStub:
             modal_proto.api_pb2.WorkspaceDashboardUrlRequest,
             modal_proto.api_pb2.WorkspaceDashboardUrlResponse,
         )
+        self.WorkspaceMembersList = grpclib.client.UnaryUnaryMethod(
+            channel,
+            '/modal.client.ModalClient/WorkspaceMembersList',
+            google.protobuf.empty_pb2.Empty,
+            modal_proto.api_pb2.WorkspaceMembersListResponse,
+        )
         self.WorkspaceNameLookup = grpclib.client.UnaryUnaryMethod(
             channel,
             '/modal.client.ModalClient/WorkspaceNameLookup',