isolate 0.14.1__tar.gz → 0.14.2__tar.gz

{isolate-0.14.1 → isolate-0.14.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: isolate
-Version: 0.14.1
+Version: 0.14.2
 Summary: Managed isolated environments for Python
 Author-email: Features & Labels <hello@fal.ai>
 Project-URL: Issues, https://github.com/fal-ai/isolate/issues

{isolate-0.14.1 → isolate-0.14.2}/src/isolate/_isolate_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.14.1'
-__version_tuple__ = version_tuple = (0, 14, 1)
+__version__ = version = '0.14.2'
+__version_tuple__ = version_tuple = (0, 14, 2)

{isolate-0.14.1 → isolate-0.14.2}/src/isolate/backends/common.py

@@ -117,6 +117,7 @@ def _io_observer(
                 return  # Nothing to read
 
             for line in raw_data.splitlines():
+                # TODO: parse the lines to include `extra={...}` added by the logger?
                 hook(line)
 
     def _reader():

{isolate-0.14.1 → isolate-0.14.2}/src/isolate/logger.py

@@ -9,6 +9,8 @@ from isolate.logs import LogLevel, LogSource
 # but it handling `source` would be not trivial, so we are better off
 # just keeping it simple for now.
 class IsolateLogger:
+    extra_labels: Dict[str, str] = {}
+
     def __init__(self, log_labels: Dict[str, str]):
         self.log_labels = log_labels
 
@@ -18,6 +20,7 @@ class IsolateLogger:
             "level": level.name,
             "message": message,
             **self.log_labels,
+            **self.extra_labels,
         }
         print(json.dumps(record))
 

{isolate-0.14.1 → isolate-0.14.2}/src/isolate/server/server.py

@@ -367,9 +367,7 @@ class IsolateServicer(definitions.IsolateServicer):
 
         task = self.background_tasks[request.task_id]
 
-        task.logger = IsolateLogger.with_env_expanded(
-            dict(request.metadata.logger_labels)
-        )
+        task.logger.extra_labels = dict(request.metadata.logger_labels)
 
         return definitions.SetMetadataResponse()
 
@@ -379,13 +377,19 @@ class IsolateServicer(definitions.IsolateServicer):
         context: ServicerContext,
     ) -> Iterator[definitions.PartialRunResult]:
         try:
-            yield from self._run_task(RunTask(request=request))
+            # HACK: we can support only one task at a time for Run
+            # TODO: move away from this when we use submit for env-aware tasks
+            task = RunTask(request=request)
+            self.background_tasks["RUN"] = task
+            yield from self._run_task(task)
         except GRPCException as exc:
             return self.abort_with_msg(
                 exc.message,
                 context,
                 code=exc.code,
             )
+        finally:
+            self.background_tasks.pop("RUN", None)
 
     def List(
         self,

{isolate-0.14.1 → isolate-0.14.2}/src/isolate.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: isolate
-Version: 0.14.1
+Version: 0.14.2
 Summary: Managed isolated environments for Python
 Author-email: Features & Labels <hello@fal.ai>
 Project-URL: Issues, https://github.com/fal-ai/isolate/issues