loom-kernel 2.17.1__tar.gz → 2.17.2__tar.gz

{loom_kernel-2.17.1 → loom_kernel-2.17.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: loom-kernel
-Version: 2.17.1
+Version: 2.17.2
 Summary: Loom Python project
 License-File: LICENSE
 Classifier: Programming Language :: Python :: 3

{loom_kernel-2.17.1 → loom_kernel-2.17.2}/src/loom/etl/executor/_dispatcher.py

@@ -18,6 +18,7 @@ Swap implementations to move from local threads to distributed workers::
 
 from __future__ import annotations
 
+import contextvars
 from collections.abc import Callable, Sequence
 from concurrent.futures import ThreadPoolExecutor
 from typing import Protocol, runtime_checkable
@@ -27,6 +28,9 @@ from typing import Protocol, runtime_checkable
 class ParallelDispatcher(Protocol):
     """Protocol for executing a group of independent tasks in parallel.
 
+    Each task should run in a copy of the caller's context, so a step keeps
+    the run's context variables (its flow run, its trace parent).
+
     All tasks in a group are independent — the executor guarantees no
     data dependency between them.  Implementations must:
 
@@ -82,7 +86,8 @@ class ThreadDispatcher:
             return
         workers = self._max_workers or len(tasks)
         with ThreadPoolExecutor(max_workers=workers) as pool:
-            futures = [pool.submit(task) for task in tasks]
+            # Each task runs in a copy of the caller's context, as asyncio.to_thread does.
+            futures = [pool.submit(contextvars.copy_context().run, task) for task in tasks]
             first_exc: Exception | None = None
             for future in futures:
                 try:

{loom_kernel-2.17.1 → loom_kernel-2.17.2}/src/loom/prefect/observer/_logging_bridge.py

@@ -56,8 +56,8 @@ def current_flow_run_id() -> uuid.UUID | None:
 class PrefectLogBridgeHandler(logging.Handler):
     """logging.Handler that ships records to the Prefect logs API.
 
-    Records carry the active ``task_run_id`` from the ContextVar if one
-    is bound; otherwise they fall back to the installed ``flow_run_id``.
+    Records carry the installed ``flow_run_id`` and, inside a step, the
+    active ``task_run_id`` from the ContextVar too.
     A daemon worker drains the queue and retries with exponential
     backoff up to ``_MAX_ATTEMPTS`` before incrementing ``dropped_logs``.
     """
@@ -120,11 +120,10 @@ class PrefectLogBridgeHandler(logging.Handler):
             "level": record.levelno,
             "message": record.getMessage(),
             "timestamp": datetime.fromtimestamp(record.created, tz=UTC),
+            "flow_run_id": flow_run_id,
         }
         if task_run_id is not None:
             payload["task_run_id"] = task_run_id
-        else:
-            payload["flow_run_id"] = flow_run_id
         return payload
 
     def _drain_loop(self) -> None: