modal 0.72.18__py3-none-any.whl → 0.72.20__py3-none-any.whl

modal_proto/api_pb2.pyi

@@ -8,6 +8,7 @@ import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
+import google.protobuf.struct_pb2
 import google.protobuf.wrappers_pb2
 import sys
 import typing
@@ -6037,6 +6038,193 @@ class NetworkAccess(google.protobuf.message.Message):
 
 global___NetworkAccess = NetworkAccess
 
+class NotebookKernelPublishResultsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class ExecuteReply(google.protobuf.message.Message):
+        """See kernelshim.py for the differences between this and `ExecuteResult`.
+        https://jupyter-client.readthedocs.io/en/stable/messaging.html#execution-results
+        """
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        STATUS_FIELD_NUMBER: builtins.int
+        EXECUTION_COUNT_FIELD_NUMBER: builtins.int
+        status: builtins.str
+        execution_count: builtins.int
+        def __init__(
+            self,
+            *,
+            status: builtins.str = ...,
+            execution_count: builtins.int = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["execution_count", b"execution_count", "status", b"status"]) -> None: ...
+
+    class CellResult(google.protobuf.message.Message):
+        """IOPub message or reply received from the kernel for a cell."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        CELL_ID_FIELD_NUMBER: builtins.int
+        OUTPUT_FIELD_NUMBER: builtins.int
+        CLEAR_OUTPUTS_FIELD_NUMBER: builtins.int
+        EXECUTE_REPLY_FIELD_NUMBER: builtins.int
+        cell_id: builtins.str
+        @property
+        def output(self) -> global___NotebookOutput:
+            """Persistent output that is saved in the notebook."""
+        clear_outputs: builtins.bool
+        """Clear all previous outputs of the cell."""
+        @property
+        def execute_reply(self) -> global___NotebookKernelPublishResultsRequest.ExecuteReply:
+            """Cell has finished executing, return the kernel's execute_reply."""
+        def __init__(
+            self,
+            *,
+            cell_id: builtins.str = ...,
+            output: global___NotebookOutput | None = ...,
+            clear_outputs: builtins.bool = ...,
+            execute_reply: global___NotebookKernelPublishResultsRequest.ExecuteReply | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["clear_outputs", b"clear_outputs", "execute_reply", b"execute_reply", "output", b"output", "result_type", b"result_type"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing_extensions.Literal["cell_id", b"cell_id", "clear_outputs", b"clear_outputs", "execute_reply", b"execute_reply", "output", b"output", "result_type", b"result_type"]) -> None: ...
+        def WhichOneof(self, oneof_group: typing_extensions.Literal["result_type", b"result_type"]) -> typing_extensions.Literal["output", "clear_outputs", "execute_reply"] | None: ...
+
+    NOTEBOOK_ID_FIELD_NUMBER: builtins.int
+    RESULTS_FIELD_NUMBER: builtins.int
+    notebook_id: builtins.str
+    @property
+    def results(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___NotebookKernelPublishResultsRequest.CellResult]: ...
+    def __init__(
+        self,
+        *,
+        notebook_id: builtins.str = ...,
+        results: collections.abc.Iterable[global___NotebookKernelPublishResultsRequest.CellResult] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["notebook_id", b"notebook_id", "results", b"results"]) -> None: ...
+
+global___NotebookKernelPublishResultsRequest = NotebookKernelPublishResultsRequest
+
+class NotebookOutput(google.protobuf.message.Message):
+    """A single output from a notebook. When you execute a cell, it produces an
+    array of these outputs as the code runs.
+
+    https://github.com/jupyter/nbformat/blob/v5.10.4/nbformat/v4/nbformat.v4.schema.json#L301-L309
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class ExecuteResult(google.protobuf.message.Message):
+        """Result of executing a code cell."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        EXECUTION_COUNT_FIELD_NUMBER: builtins.int
+        DATA_FIELD_NUMBER: builtins.int
+        METADATA_FIELD_NUMBER: builtins.int
+        execution_count: builtins.int
+        @property
+        def data(self) -> google.protobuf.struct_pb2.Struct:
+            """mimebundle"""
+        @property
+        def metadata(self) -> google.protobuf.struct_pb2.Struct: ...
+        def __init__(
+            self,
+            *,
+            execution_count: builtins.int = ...,
+            data: google.protobuf.struct_pb2.Struct | None = ...,
+            metadata: google.protobuf.struct_pb2.Struct | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["data", b"data", "metadata", b"metadata"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing_extensions.Literal["data", b"data", "execution_count", b"execution_count", "metadata", b"metadata"]) -> None: ...
+
+    class DisplayData(google.protobuf.message.Message):
+        """Data displayed as a result of code cell execution."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        DATA_FIELD_NUMBER: builtins.int
+        METADATA_FIELD_NUMBER: builtins.int
+        @property
+        def data(self) -> google.protobuf.struct_pb2.Struct:
+            """mimebundle"""
+        @property
+        def metadata(self) -> google.protobuf.struct_pb2.Struct: ...
+        def __init__(
+            self,
+            *,
+            data: google.protobuf.struct_pb2.Struct | None = ...,
+            metadata: google.protobuf.struct_pb2.Struct | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["data", b"data", "metadata", b"metadata"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing_extensions.Literal["data", b"data", "metadata", b"metadata"]) -> None: ...
+
+    class Stream(google.protobuf.message.Message):
+        """Stream output from a code cell (stdout / stderr)."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        NAME_FIELD_NUMBER: builtins.int
+        TEXT_FIELD_NUMBER: builtins.int
+        name: builtins.str
+        """stdout | stderr"""
+        text: builtins.str
+        """multiline_string"""
+        def __init__(
+            self,
+            *,
+            name: builtins.str = ...,
+            text: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["name", b"name", "text", b"text"]) -> None: ...
+
+    class Error(google.protobuf.message.Message):
+        """Output of an error that occurred during code cell execution."""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        ENAME_FIELD_NUMBER: builtins.int
+        EVALUE_FIELD_NUMBER: builtins.int
+        TRACEBACK_FIELD_NUMBER: builtins.int
+        ename: builtins.str
+        evalue: builtins.str
+        @property
+        def traceback(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+        def __init__(
+            self,
+            *,
+            ename: builtins.str = ...,
+            evalue: builtins.str = ...,
+            traceback: collections.abc.Iterable[builtins.str] | None = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["ename", b"ename", "evalue", b"evalue", "traceback", b"traceback"]) -> None: ...
+
+    EXECUTE_RESULT_FIELD_NUMBER: builtins.int
+    DISPLAY_DATA_FIELD_NUMBER: builtins.int
+    STREAM_FIELD_NUMBER: builtins.int
+    ERROR_FIELD_NUMBER: builtins.int
+    @property
+    def execute_result(self) -> global___NotebookOutput.ExecuteResult: ...
+    @property
+    def display_data(self) -> global___NotebookOutput.DisplayData: ...
+    @property
+    def stream(self) -> global___NotebookOutput.Stream: ...
+    @property
+    def error(self) -> global___NotebookOutput.Error: ...
+    def __init__(
+        self,
+        *,
+        execute_result: global___NotebookOutput.ExecuteResult | None = ...,
+        display_data: global___NotebookOutput.DisplayData | None = ...,
+        stream: global___NotebookOutput.Stream | None = ...,
+        error: global___NotebookOutput.Error | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["display_data", b"display_data", "error", b"error", "execute_result", b"execute_result", "output_type", b"output_type", "stream", b"stream"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["display_data", b"display_data", "error", b"error", "execute_result", b"execute_result", "output_type", b"output_type", "stream", b"stream"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["output_type", b"output_type"]) -> typing_extensions.Literal["execute_result", "display_data", "stream", "error"] | None: ...
+
+global___NotebookOutput = NotebookOutput
+
 class Object(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -6985,13 +7173,20 @@ class SandboxGetTaskIdRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     SANDBOX_ID_FIELD_NUMBER: builtins.int
+    TIMEOUT_FIELD_NUMBER: builtins.int
+    WAIT_UNTIL_READY_FIELD_NUMBER: builtins.int
     sandbox_id: builtins.str
+    timeout: builtins.float
+    wait_until_ready: builtins.bool
+    """If true, waits until the container's postStart hook has been run before returning. Useful for detecting init failures."""
     def __init__(
         self,
         *,
         sandbox_id: builtins.str = ...,
+        timeout: builtins.float = ...,
+        wait_until_ready: builtins.bool = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["sandbox_id", b"sandbox_id"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["sandbox_id", b"sandbox_id", "timeout", b"timeout", "wait_until_ready", b"wait_until_ready"]) -> None: ...
 
 global___SandboxGetTaskIdRequest = SandboxGetTaskIdRequest
 
@@ -6999,13 +7194,24 @@ class SandboxGetTaskIdResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     TASK_ID_FIELD_NUMBER: builtins.int
+    TASK_RESULT_FIELD_NUMBER: builtins.int
     task_id: builtins.str
+    """This is None if the sandbox was terminated before a task could be scheduled."""
+    @property
+    def task_result(self) -> global___GenericResult:
+        """If the task has already exited, this is the result."""
     def __init__(
         self,
         *,
-        task_id: builtins.str = ...,
+        task_id: builtins.str | None = ...,
+        task_result: global___GenericResult | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["task_id", b"task_id"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_task_id", b"_task_id", "_task_result", b"_task_result", "task_id", b"task_id", "task_result", b"task_result"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_task_id", b"_task_id", "_task_result", b"_task_result", "task_id", b"task_id", "task_result", b"task_result"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_task_id", b"_task_id"]) -> typing_extensions.Literal["task_id"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_task_result", b"_task_result"]) -> typing_extensions.Literal["task_result"] | None: ...
 
 global___SandboxGetTaskIdResponse = SandboxGetTaskIdResponse
 

modal_proto/api_pb2_grpc.py

@@ -405,6 +405,11 @@ class ModalClientStub(object):
                 request_serializer=modal__proto_dot_api__pb2.MountPutFileRequest.SerializeToString,
                 response_deserializer=modal__proto_dot_api__pb2.MountPutFileResponse.FromString,
                 )
+        self.NotebookKernelPublishResults = channel.unary_unary(
+                '/modal.client.ModalClient/NotebookKernelPublishResults',
+                request_serializer=modal__proto_dot_api__pb2.NotebookKernelPublishResultsRequest.SerializeToString,
+                response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+                )
         self.ProxyCreate = channel.unary_unary(
                 '/modal.client.ModalClient/ProxyCreate',
                 request_serializer=modal__proto_dot_api__pb2.ProxyCreateRequest.SerializeToString,
@@ -1177,6 +1182,13 @@ class ModalClientServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def NotebookKernelPublishResults(self, request, context):
+        """Notebooks
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def ProxyCreate(self, request, context):
         """Proxies
         """
@@ -1923,6 +1935,11 @@ def add_ModalClientServicer_to_server(servicer, server):
                     request_deserializer=modal__proto_dot_api__pb2.MountPutFileRequest.FromString,
                     response_serializer=modal__proto_dot_api__pb2.MountPutFileResponse.SerializeToString,
             ),
+            'NotebookKernelPublishResults': grpc.unary_unary_rpc_method_handler(
+                    servicer.NotebookKernelPublishResults,
+                    request_deserializer=modal__proto_dot_api__pb2.NotebookKernelPublishResultsRequest.FromString,
+                    response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
+            ),
             'ProxyCreate': grpc.unary_unary_rpc_method_handler(
                     servicer.ProxyCreate,
                     request_deserializer=modal__proto_dot_api__pb2.ProxyCreateRequest.FromString,
@@ -3544,6 +3561,23 @@ class ModalClient(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def NotebookKernelPublishResults(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/modal.client.ModalClient/NotebookKernelPublishResults',
+            modal__proto_dot_api__pb2.NotebookKernelPublishResultsRequest.SerializeToString,
+            google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def ProxyCreate(request,
             target,

modal_proto/api_pb2_grpc.pyi

@@ -336,6 +336,11 @@ class ModalClientStub:
         modal_proto.api_pb2.MountPutFileRequest,
         modal_proto.api_pb2.MountPutFileResponse,
     ]
+    NotebookKernelPublishResults: grpc.UnaryUnaryMultiCallable[
+        modal_proto.api_pb2.NotebookKernelPublishResultsRequest,
+        google.protobuf.empty_pb2.Empty,
+    ]
+    """Notebooks"""
     ProxyCreate: grpc.UnaryUnaryMultiCallable[
         modal_proto.api_pb2.ProxyCreateRequest,
         modal_proto.api_pb2.ProxyCreateResponse,
@@ -1060,6 +1065,13 @@ class ModalClientServicer(metaclass=abc.ABCMeta):
         context: grpc.ServicerContext,
     ) -> modal_proto.api_pb2.MountPutFileResponse: ...
     @abc.abstractmethod
+    def NotebookKernelPublishResults(
+        self,
+        request: modal_proto.api_pb2.NotebookKernelPublishResultsRequest,
+        context: grpc.ServicerContext,
+    ) -> google.protobuf.empty_pb2.Empty:
+        """Notebooks"""
+    @abc.abstractmethod
     def ProxyCreate(
         self,
         request: modal_proto.api_pb2.ProxyCreateRequest,

modal_proto/modal_api_grpc.py

@@ -90,6 +90,7 @@ class ModalClientModal:
         self.ImageJoinStreaming = modal.client.UnaryStreamWrapper(grpclib_stub.ImageJoinStreaming, client)
         self.MountGetOrCreate = modal.client.UnaryUnaryWrapper(grpclib_stub.MountGetOrCreate, client)
         self.MountPutFile = modal.client.UnaryUnaryWrapper(grpclib_stub.MountPutFile, client)
+        self.NotebookKernelPublishResults = modal.client.UnaryUnaryWrapper(grpclib_stub.NotebookKernelPublishResults, client)
         self.ProxyCreate = modal.client.UnaryUnaryWrapper(grpclib_stub.ProxyCreate, client)
         self.ProxyDelete = modal.client.UnaryUnaryWrapper(grpclib_stub.ProxyDelete, client)
         self.ProxyGet = modal.client.UnaryUnaryWrapper(grpclib_stub.ProxyGet, client)

modal_version/_version_generated.py

@@ -1,4 +1,4 @@
 # Copyright Modal Labs 2025
 
 # Note: Reset this value to -1 whenever you make a minor `0.X` release of the client.
-build_number = 18  # git: 6608b5d
+build_number = 20  # git: edfa115