isolate 0.12.16__py3-none-any.whl → 0.13.1__py3-none-any.whl

isolate/_isolate_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.12.16'
-__version_tuple__ = version_tuple = (0, 12, 16)
+__version__ = version = '0.13.1'
+__version_tuple__ = version_tuple = (0, 13, 1)

isolate/connections/grpc/_base.py

@@ -16,6 +16,7 @@ from isolate.connections.common import serialize_object
 from isolate.connections.grpc import agent, definitions
 from isolate.connections.grpc.configuration import get_default_options
 from isolate.connections.grpc.interface import from_grpc
+from isolate.logger import logger
 from isolate.logs import LogLevel, LogSource
 
 
@@ -36,7 +37,7 @@ class GRPCExecutionBase(EnvironmentConnection):
     def _establish_bridge(
         self,
         *,
-        max_wait_timeout: float = 10.0,
+        max_wait_timeout: float = 20.0,
     ) -> Iterator[definitions.AgentStub]:
         with self.start_agent() as (address, credentials):
             with grpc.secure_channel(
@@ -148,5 +149,5 @@ class LocalPythonGRPC(PythonExecutionBase[str], GRPCExecutionBase):
         ]
 
     def handle_agent_log(self, line: str, level: LogLevel, source: LogSource) -> None:
-        print(f"[{source}] [{level}] {line}")
+        logger.log(level, line, source)
         self.log(line, level=level, source=source)

isolate/connections/grpc/agent.py

@@ -22,6 +22,8 @@ from isolate.connections.common import SerializationError, serialize_object
 from isolate.connections.grpc import definitions
 from isolate.connections.grpc.configuration import get_default_options
 from isolate.connections.grpc.interface import from_grpc
+from isolate.logger import logger
+from isolate.logs import LogLevel, LogSource
 
 
 @dataclass
@@ -149,7 +151,7 @@ class AgentServicer(definitions.AgentServicer):
             definition = serialize_object(serialization_method, result)
         except SerializationError:
             if stringized_tb:
-                print(stringized_tb, file=sys.stderr)
+                logger.log(LogLevel.ERROR, stringized_tb, LogSource.BRIDGE)
             raise AbortException(
                 "Error while serializing the execution result "
                 f"(object of type {type(result)})."

isolate/connections/grpc/definitions/agent_pb2.py

@@ -1,12 +1,12 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: agent.proto
+# Protobuf Python Version: 5.26.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
-
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -14,17 +14,16 @@ _sym_db = _symbol_database.Default()
 
 from isolate.connections.grpc.definitions import common_pb2 as common__pb2
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x0b\x61gent.proto\x1a\x0c\x63ommon.proto"n\n\x0c\x46unctionCall\x12#\n\x08\x66unction\x18\x01 \x01(\x0b\x32\x11.SerializedObject\x12*\n\nsetup_func\x18\x02 \x01(\x0b\x32\x11.SerializedObjectH\x00\x88\x01\x01\x42\r\n\x0b_setup_func24\n\x05\x41gent\x12+\n\x03Run\x12\r.FunctionCall\x1a\x11.PartialRunResult"\x00\x30\x01\x62\x06proto3'
-)
 
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "agent_pb2", globals())
-if _descriptor._USE_C_DESCRIPTORS == False:
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0b\x61gent.proto\x1a\x0c\x63ommon.proto\"n\n\x0c\x46unctionCall\x12#\n\x08\x66unction\x18\x01 \x01(\x0b\x32\x11.SerializedObject\x12*\n\nsetup_func\x18\x02 \x01(\x0b\x32\x11.SerializedObjectH\x00\x88\x01\x01\x42\r\n\x0b_setup_func24\n\x05\x41gent\x12+\n\x03Run\x12\r.FunctionCall\x1a\x11.PartialRunResult\"\x00\x30\x01\x62\x06proto3')
 
-    DESCRIPTOR._options = None
-    _FUNCTIONCALL._serialized_start = 29
-    _FUNCTIONCALL._serialized_end = 139
-    _AGENT._serialized_start = 141
-    _AGENT._serialized_end = 193
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'agent_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_FUNCTIONCALL']._serialized_start=29
+  _globals['_FUNCTIONCALL']._serialized_end=139
+  _globals['_AGENT']._serialized_start=141
+  _globals['_AGENT']._serialized_end=193
 # @@protoc_insertion_point(module_scope)

isolate/connections/grpc/definitions/agent_pb2.pyi

@@ -2,20 +2,16 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 from isolate.connections.grpc.definitions import common_pb2
 import google.protobuf.descriptor
 import google.protobuf.message
-import sys
-
-if sys.version_info >= (3, 8):
-    import typing as typing_extensions
-else:
-    import typing_extensions
+import typing
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
-@typing_extensions.final
+@typing.final
 class FunctionCall(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -24,6 +20,7 @@ class FunctionCall(google.protobuf.message.Message):
     @property
     def function(self) -> common_pb2.SerializedObject:
         """The function to execute and return the results to."""
+
     @property
     def setup_func(self) -> common_pb2.SerializedObject:
         """Optionally the setup function which will be passed
@@ -31,36 +28,15 @@ class FunctionCall(google.protobuf.message.Message):
         has to be an idempotent step since the result for
         this executable will be cached.
         """
+
     def __init__(
         self,
         *,
         function: common_pb2.SerializedObject | None = ...,
         setup_func: common_pb2.SerializedObject | None = ...,
     ) -> None: ...
-    def HasField(
-        self,
-        field_name: typing_extensions.Literal[
-            "_setup_func",
-            b"_setup_func",
-            "function",
-            b"function",
-            "setup_func",
-            b"setup_func",
-        ],
-    ) -> builtins.bool: ...
-    def ClearField(
-        self,
-        field_name: typing_extensions.Literal[
-            "_setup_func",
-            b"_setup_func",
-            "function",
-            b"function",
-            "setup_func",
-            b"setup_func",
-        ],
-    ) -> None: ...
-    def WhichOneof(
-        self, oneof_group: typing_extensions.Literal["_setup_func", b"_setup_func"]
-    ) -> typing_extensions.Literal["setup_func"] | None: ...
+    def HasField(self, field_name: typing.Literal["_setup_func", b"_setup_func", "function", b"function", "setup_func", b"setup_func"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["_setup_func", b"_setup_func", "function", b"function", "setup_func", b"setup_func"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["_setup_func", b"_setup_func"]) -> typing.Literal["setup_func"] | None: ...
 
 global___FunctionCall = FunctionCall

isolate/connections/grpc/definitions/agent_pb2_grpc.py

@@ -1,10 +1,35 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
+import warnings
 
 from isolate.connections.grpc.definitions import agent_pb2 as agent__pb2
 from isolate.connections.grpc.definitions import common_pb2 as common__pb2
 
+GRPC_GENERATED_VERSION = '1.64.0'
+GRPC_VERSION = grpc.__version__
+EXPECTED_ERROR_RELEASE = '1.65.0'
+SCHEDULED_RELEASE_DATE = 'June 25, 2024'
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    warnings.warn(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in agent_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+        + f' This warning will become an error in {EXPECTED_ERROR_RELEASE},'
+        + f' scheduled for release on {SCHEDULED_RELEASE_DATE}.',
+        RuntimeWarning
+    )
+
 
 class AgentStub(object):
     """Missing associated documentation comment in .proto file."""
@@ -16,55 +41,56 @@ class AgentStub(object):
             channel: A grpc.Channel.
         """
         self.Run = channel.unary_stream(
-            "/Agent/Run",
-            request_serializer=agent__pb2.FunctionCall.SerializeToString,
-            response_deserializer=common__pb2.PartialRunResult.FromString,
-        )
+                '/Agent/Run',
+                request_serializer=agent__pb2.FunctionCall.SerializeToString,
+                response_deserializer=common__pb2.PartialRunResult.FromString,
+                _registered_method=True)
 
 
 class AgentServicer(object):
     """Missing associated documentation comment in .proto file."""
 
     def Run(self, request, context):
-        """Start running the given function, and stream results back."""
+        """Start running the given function, and stream results back.
+        """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details("Method not implemented!")
-        raise NotImplementedError("Method not implemented!")
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
 
 
 def add_AgentServicer_to_server(servicer, server):
     rpc_method_handlers = {
-        "Run": grpc.unary_stream_rpc_method_handler(
-            servicer.Run,
-            request_deserializer=agent__pb2.FunctionCall.FromString,
-            response_serializer=common__pb2.PartialRunResult.SerializeToString,
-        ),
+            'Run': grpc.unary_stream_rpc_method_handler(
+                    servicer.Run,
+                    request_deserializer=agent__pb2.FunctionCall.FromString,
+                    response_serializer=common__pb2.PartialRunResult.SerializeToString,
+            ),
     }
-    generic_handler = grpc.method_handlers_generic_handler("Agent", rpc_method_handlers)
+    generic_handler = grpc.method_handlers_generic_handler(
+            'Agent', rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('Agent', rpc_method_handlers)
 
 
-# This class is part of an EXPERIMENTAL API.
+ # This class is part of an EXPERIMENTAL API.
 class Agent(object):
     """Missing associated documentation comment in .proto file."""
 
     @staticmethod
-    def Run(
-        request,
-        target,
-        options=(),
-        channel_credentials=None,
-        call_credentials=None,
-        insecure=False,
-        compression=None,
-        wait_for_ready=None,
-        timeout=None,
-        metadata=None,
-    ):
+    def Run(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_stream(
             request,
             target,
-            "/Agent/Run",
+            '/Agent/Run',
             agent__pb2.FunctionCall.SerializeToString,
             common__pb2.PartialRunResult.FromString,
             options,
@@ -75,4 +101,4 @@ class Agent(object):
             wait_for_ready,
             timeout,
             metadata,
-        )
+            _registered_method=True)

isolate/connections/grpc/definitions/common_pb2.py

@@ -1,38 +1,35 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: common.proto
+# Protobuf Python Version: 5.26.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
-
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
 
 
-from google.protobuf import (
-    timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2,
-)
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x0c\x63ommon.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x89\x01\n\x10SerializedObject\x12\x0e\n\x06method\x18\x01 \x01(\t\x12\x12\n\ndefinition\x18\x02 \x01(\x0c\x12\x15\n\rwas_it_raised\x18\x03 \x01(\x08\x12!\n\x14stringized_traceback\x18\x04 \x01(\tH\x00\x88\x01\x01\x42\x17\n\x15_stringized_traceback"n\n\x10PartialRunResult\x12\x13\n\x0bis_complete\x18\x01 \x01(\x08\x12\x12\n\x04logs\x18\x02 \x03(\x0b\x32\x04.Log\x12&\n\x06result\x18\x03 \x01(\x0b\x32\x11.SerializedObjectH\x00\x88\x01\x01\x42\t\n\x07_result"{\n\x03Log\x12\x0f\n\x07message\x18\x01 \x01(\t\x12\x1a\n\x06source\x18\x02 \x01(\x0e\x32\n.LogSource\x12\x18\n\x05level\x18\x03 \x01(\x0e\x32\t.LogLevel\x12-\n\ttimestamp\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp*.\n\tLogSource\x12\x0b\n\x07\x42UILDER\x10\x00\x12\n\n\x06\x42RIDGE\x10\x01\x12\x08\n\x04USER\x10\x02*Z\n\x08LogLevel\x12\t\n\x05TRACE\x10\x00\x12\t\n\x05\x44\x45\x42UG\x10\x01\x12\x08\n\x04INFO\x10\x02\x12\x0b\n\x07WARNING\x10\x03\x12\t\n\x05\x45RROR\x10\x04\x12\n\n\x06STDOUT\x10\x05\x12\n\n\x06STDERR\x10\x06\x62\x06proto3'
-)
 
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "common_pb2", globals())
-if _descriptor._USE_C_DESCRIPTORS == False:
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0c\x63ommon.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x89\x01\n\x10SerializedObject\x12\x0e\n\x06method\x18\x01 \x01(\t\x12\x12\n\ndefinition\x18\x02 \x01(\x0c\x12\x15\n\rwas_it_raised\x18\x03 \x01(\x08\x12!\n\x14stringized_traceback\x18\x04 \x01(\tH\x00\x88\x01\x01\x42\x17\n\x15_stringized_traceback\"n\n\x10PartialRunResult\x12\x13\n\x0bis_complete\x18\x01 \x01(\x08\x12\x12\n\x04logs\x18\x02 \x03(\x0b\x32\x04.Log\x12&\n\x06result\x18\x03 \x01(\x0b\x32\x11.SerializedObjectH\x00\x88\x01\x01\x42\t\n\x07_result\"{\n\x03Log\x12\x0f\n\x07message\x18\x01 \x01(\t\x12\x1a\n\x06source\x18\x02 \x01(\x0e\x32\n.LogSource\x12\x18\n\x05level\x18\x03 \x01(\x0e\x32\t.LogLevel\x12-\n\ttimestamp\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp*.\n\tLogSource\x12\x0b\n\x07\x42UILDER\x10\x00\x12\n\n\x06\x42RIDGE\x10\x01\x12\x08\n\x04USER\x10\x02*Z\n\x08LogLevel\x12\t\n\x05TRACE\x10\x00\x12\t\n\x05\x44\x45\x42UG\x10\x01\x12\x08\n\x04INFO\x10\x02\x12\x0b\n\x07WARNING\x10\x03\x12\t\n\x05\x45RROR\x10\x04\x12\n\n\x06STDOUT\x10\x05\x12\n\n\x06STDERR\x10\x06\x62\x06proto3')
 
-    DESCRIPTOR._options = None
-    _LOGSOURCE._serialized_start = 426
-    _LOGSOURCE._serialized_end = 472
-    _LOGLEVEL._serialized_start = 474
-    _LOGLEVEL._serialized_end = 564
-    _SERIALIZEDOBJECT._serialized_start = 50
-    _SERIALIZEDOBJECT._serialized_end = 187
-    _PARTIALRUNRESULT._serialized_start = 189
-    _PARTIALRUNRESULT._serialized_end = 299
-    _LOG._serialized_start = 301
-    _LOG._serialized_end = 424
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'common_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_LOGSOURCE']._serialized_start=426
+  _globals['_LOGSOURCE']._serialized_end=472
+  _globals['_LOGLEVEL']._serialized_start=474
+  _globals['_LOGLEVEL']._serialized_end=564
+  _globals['_SERIALIZEDOBJECT']._serialized_start=50
+  _globals['_SERIALIZEDOBJECT']._serialized_end=187
+  _globals['_PARTIALRUNRESULT']._serialized_start=189
+  _globals['_PARTIALRUNRESULT']._serialized_end=299
+  _globals['_LOG']._serialized_start=301
+  _globals['_LOG']._serialized_end=424
 # @@protoc_insertion_point(module_scope)

isolate/connections/grpc/definitions/common_pb2.pyi

@@ -2,6 +2,7 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import collections.abc
 import google.protobuf.descriptor
@@ -23,10 +24,7 @@ class _LogSource:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _LogSourceEnumTypeWrapper(
-    google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_LogSource.ValueType],
-    builtins.type,
-):
+class _LogSourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_LogSource.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     BUILDER: _LogSource.ValueType  # 0
     BRIDGE: _LogSource.ValueType  # 1
@@ -43,10 +41,7 @@ class _LogLevel:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _LogLevelEnumTypeWrapper(
-    google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_LogLevel.ValueType],
-    builtins.type,
-):
+class _LogLevelEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_LogLevel.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     TRACE: _LogLevel.ValueType  # 0
     DEBUG: _LogLevel.ValueType  # 1
@@ -67,7 +62,7 @@ STDOUT: LogLevel.ValueType  # 5
 STDERR: LogLevel.ValueType  # 6
 global___LogLevel = LogLevel
 
-@typing_extensions.final
+@typing.final
 class SerializedObject(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -95,40 +90,13 @@ class SerializedObject(google.protobuf.message.Message):
         was_it_raised: builtins.bool = ...,
         stringized_traceback: builtins.str | None = ...,
     ) -> None: ...
-    def HasField(
-        self,
-        field_name: typing_extensions.Literal[
-            "_stringized_traceback",
-            b"_stringized_traceback",
-            "stringized_traceback",
-            b"stringized_traceback",
-        ],
-    ) -> builtins.bool: ...
-    def ClearField(
-        self,
-        field_name: typing_extensions.Literal[
-            "_stringized_traceback",
-            b"_stringized_traceback",
-            "definition",
-            b"definition",
-            "method",
-            b"method",
-            "stringized_traceback",
-            b"stringized_traceback",
-            "was_it_raised",
-            b"was_it_raised",
-        ],
-    ) -> None: ...
-    def WhichOneof(
-        self,
-        oneof_group: typing_extensions.Literal[
-            "_stringized_traceback", b"_stringized_traceback"
-        ],
-    ) -> typing_extensions.Literal["stringized_traceback"] | None: ...
+    def HasField(self, field_name: typing.Literal["_stringized_traceback", b"_stringized_traceback", "stringized_traceback", b"stringized_traceback"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["_stringized_traceback", b"_stringized_traceback", "definition", b"definition", "method", b"method", "stringized_traceback", b"stringized_traceback", "was_it_raised", b"was_it_raised"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["_stringized_traceback", b"_stringized_traceback"]) -> typing.Literal["stringized_traceback"] | None: ...
 
 global___SerializedObject = SerializedObject
 
-@typing_extensions.final
+@typing.final
 class PartialRunResult(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -138,17 +106,15 @@ class PartialRunResult(google.protobuf.message.Message):
     is_complete: builtins.bool
     """A flag indicating whether the run has completed."""
     @property
-    def logs(
-        self,
-    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
-        global___Log
-    ]:
+    def logs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Log]:
         """A list of logs collected during this partial execution. It does
         not include old logs.
         """
+
     @property
     def result(self) -> global___SerializedObject:
         """The result of the run, if it is complete."""
+
     def __init__(
         self,
         *,
@@ -156,32 +122,13 @@ class PartialRunResult(google.protobuf.message.Message):
         logs: collections.abc.Iterable[global___Log] | None = ...,
         result: global___SerializedObject | None = ...,
     ) -> None: ...
-    def HasField(
-        self,
-        field_name: typing_extensions.Literal[
-            "_result", b"_result", "result", b"result"
-        ],
-    ) -> builtins.bool: ...
-    def ClearField(
-        self,
-        field_name: typing_extensions.Literal[
-            "_result",
-            b"_result",
-            "is_complete",
-            b"is_complete",
-            "logs",
-            b"logs",
-            "result",
-            b"result",
-        ],
-    ) -> None: ...
-    def WhichOneof(
-        self, oneof_group: typing_extensions.Literal["_result", b"_result"]
-    ) -> typing_extensions.Literal["result"] | None: ...
+    def HasField(self, field_name: typing.Literal["_result", b"_result", "result", b"result"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["_result", b"_result", "is_complete", b"is_complete", "logs", b"logs", "result", b"result"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["_result", b"_result"]) -> typing.Literal["result"] | None: ...
 
 global___PartialRunResult = PartialRunResult
 
-@typing_extensions.final
+@typing.final
 class Log(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -202,21 +149,7 @@ class Log(google.protobuf.message.Message):
         level: global___LogLevel.ValueType = ...,
         timestamp: google.protobuf.timestamp_pb2.Timestamp | None = ...,
     ) -> None: ...
-    def HasField(
-        self, field_name: typing_extensions.Literal["timestamp", b"timestamp"]
-    ) -> builtins.bool: ...
-    def ClearField(
-        self,
-        field_name: typing_extensions.Literal[
-            "level",
-            b"level",
-            "message",
-            b"message",
-            "source",
-            b"source",
-            "timestamp",
-            b"timestamp",
-        ],
-    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["timestamp", b"timestamp"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["level", b"level", "message", b"message", "source", b"source", "timestamp", b"timestamp"]) -> None: ...
 
 global___Log = Log

isolate/connections/grpc/definitions/common_pb2_grpc.py

@@ -1,3 +1,29 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.64.0'
+GRPC_VERSION = grpc.__version__
+EXPECTED_ERROR_RELEASE = '1.65.0'
+SCHEDULED_RELEASE_DATE = 'June 25, 2024'
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    warnings.warn(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in common_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+        + f' This warning will become an error in {EXPECTED_ERROR_RELEASE},'
+        + f' scheduled for release on {SCHEDULED_RELEASE_DATE}.',
+        RuntimeWarning
+    )

isolate/logger.py

@@ -0,0 +1,17 @@
+import json
+
+
+# NOTE: we probably should've created a proper `logging.getLogger` here,
+# but it handling `source` would be not trivial, so we are better off
+# just keeping it simple for now.
+class IsolateLogger:
+    def log(self, level, message, source):
+        record = {
+            "isolate_source": source.name,
+            "level": level.name,
+            "message": message,
+        }
+        print(json.dumps(record))
+
+
+logger = IsolateLogger()

isolate/server/definitions/server.proto

@@ -7,6 +7,9 @@ service Isolate {
     // Run the given function on the specified environment. Streams logs
     // and the result originating from that function.
     rpc Run (BoundFunction) returns (stream PartialRunResult) {}
+
+    // Submit a function to be run without waiting for results.
+    rpc Submit (SubmitRequest) returns (SubmitResponse) {}
 }
 
 message BoundFunction {
@@ -23,3 +26,12 @@ message EnvironmentDefinition {
     // Whether to force-create this environment or not.
     bool force = 3;
 }
+
+message SubmitRequest {
+    // The function to run.
+    BoundFunction function = 1;
+}
+
+message SubmitResponse {
+    // Reserved for future use.
+}

isolate/server/definitions/server_pb2.py

@@ -1,34 +1,36 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: server.proto
+# Protobuf Python Version: 5.26.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
-
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
 
 
-from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
-
 from isolate.connections.grpc.definitions import common_pb2 as common__pb2
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x0cserver.proto\x1a\x0c\x63ommon.proto\x1a\x1cgoogle/protobuf/struct.proto"\x9d\x01\n\rBoundFunction\x12,\n\x0c\x65nvironments\x18\x01 \x03(\x0b\x32\x16.EnvironmentDefinition\x12#\n\x08\x66unction\x18\x02 \x01(\x0b\x32\x11.SerializedObject\x12*\n\nsetup_func\x18\x03 \x01(\x0b\x32\x11.SerializedObjectH\x00\x88\x01\x01\x42\r\n\x0b_setup_func"d\n\x15\x45nvironmentDefinition\x12\x0c\n\x04kind\x18\x01 \x01(\t\x12.\n\rconfiguration\x18\x02 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\r\n\x05\x66orce\x18\x03 \x01(\x08\x32\x37\n\x07Isolate\x12,\n\x03Run\x12\x0e.BoundFunction\x1a\x11.PartialRunResult"\x00\x30\x01\x62\x06proto3'
-)
-
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "server_pb2", globals())
-if _descriptor._USE_C_DESCRIPTORS == False:
 
-    DESCRIPTOR._options = None
-    _BOUNDFUNCTION._serialized_start = 61
-    _BOUNDFUNCTION._serialized_end = 218
-    _ENVIRONMENTDEFINITION._serialized_start = 220
-    _ENVIRONMENTDEFINITION._serialized_end = 320
-    _ISOLATE._serialized_start = 322
-    _ISOLATE._serialized_end = 377
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0cserver.proto\x1a\x0c\x63ommon.proto\x1a\x1cgoogle/protobuf/struct.proto\"\x9d\x01\n\rBoundFunction\x12,\n\x0c\x65nvironments\x18\x01 \x03(\x0b\x32\x16.EnvironmentDefinition\x12#\n\x08\x66unction\x18\x02 \x01(\x0b\x32\x11.SerializedObject\x12*\n\nsetup_func\x18\x03 \x01(\x0b\x32\x11.SerializedObjectH\x00\x88\x01\x01\x42\r\n\x0b_setup_func\"d\n\x15\x45nvironmentDefinition\x12\x0c\n\x04kind\x18\x01 \x01(\t\x12.\n\rconfiguration\x18\x02 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\r\n\x05\x66orce\x18\x03 \x01(\x08\"1\n\rSubmitRequest\x12 \n\x08\x66unction\x18\x01 \x01(\x0b\x32\x0e.BoundFunction\"\x10\n\x0eSubmitResponse2d\n\x07Isolate\x12,\n\x03Run\x12\x0e.BoundFunction\x1a\x11.PartialRunResult\"\x00\x30\x01\x12+\n\x06Submit\x12\x0e.SubmitRequest\x1a\x0f.SubmitResponse\"\x00\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'server_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_BOUNDFUNCTION']._serialized_start=61
+  _globals['_BOUNDFUNCTION']._serialized_end=218
+  _globals['_ENVIRONMENTDEFINITION']._serialized_start=220
+  _globals['_ENVIRONMENTDEFINITION']._serialized_end=320
+  _globals['_SUBMITREQUEST']._serialized_start=322
+  _globals['_SUBMITREQUEST']._serialized_end=371
+  _globals['_SUBMITRESPONSE']._serialized_start=373
+  _globals['_SUBMITRESPONSE']._serialized_end=389
+  _globals['_ISOLATE']._serialized_start=391
+  _globals['_ISOLATE']._serialized_end=491
 # @@protoc_insertion_point(module_scope)