prefect 3.9.2.dev5__py3-none-any.whl → 3.9.2.dev6__py3-none-any.whl

prefect/_build_info.py

@@ -1,5 +1,5 @@
 # Generated by versioningit
-__version__ = "3.9.2.dev5"
-__build_date__ = "2026-11-05 08:14:46.658497+00:00"
-__git_commit__ = "7d8cf20e30b8d22e4fdc99bef3a99050ad0d7cf8"
+__version__ = "3.9.2.dev6"
+__build_date__ = "2026-11-06 08:14:36.397592+00:00"
+__git_commit__ = "802fc9435423bed0d16fe30862de25a8df62493f"
 __dirty__ = False

prefect/flow_engine.py

@@ -11,7 +11,7 @@ import signal
 import sys
 import threading
 import time
-from collections.abc import Mapping
+from collections.abc import Callable, Mapping
 from contextlib import (
     AsyncExitStack,
     ExitStack,
@@ -334,28 +334,40 @@ def _send_heartbeats(
     resource, related = engine._build_heartbeat_event_template()
 
     stop_event = threading.Event()
+    # The heartbeat thread checks for a stop and emits while holding this lock,
+    # so once `stop_heartbeats` returns, no heartbeat is mid-emit and none will
+    # be emitted.
+    emit_lock = threading.Lock()
 
-    def heartbeat_loop() -> None:
-        while not stop_event.is_set():
-            # Check state before emitting - don't emit if final
-            if (
-                engine.flow_run
-                and engine.flow_run.state
-                and engine.flow_run.state.is_final()
-            ):
-                engine.logger.debug("Flow run in terminal state, stopping heartbeat")
-                return
-
-            try:
-                engine._emit_flow_run_heartbeat(resource, related)
-            except Exception:
-                engine.logger.debug("Failed to emit heartbeat", exc_info=True)
+    def stop_heartbeats() -> None:
+        with emit_lock:
+            stop_event.set()
 
-            # Sleep in increments to allow quick shutdown
-            for _ in range(heartbeat_seconds):
+    def heartbeat_loop() -> None:
+        while True:
+            with emit_lock:
                 if stop_event.is_set():
                     return
-                time.sleep(1)
+
+                # Check state before emitting - don't emit if final
+                if (
+                    engine.flow_run
+                    and engine.flow_run.state
+                    and engine.flow_run.state.is_final()
+                ):
+                    engine.logger.debug(
+                        "Flow run in terminal state, stopping heartbeat"
+                    )
+                    return
+
+                try:
+                    engine._emit_flow_run_heartbeat(resource, related)
+                except Exception:
+                    engine.logger.debug("Failed to emit heartbeat", exc_info=True)
+
+            # Waiting on the event wakes the thread as soon as heartbeats stop
+            if stop_event.wait(heartbeat_seconds):
+                return
 
     # Copy the current context so the heartbeat thread sees the same
     # `SettingsContext` (and therefore the same `PREFECT_API_URL`) as the
@@ -370,13 +382,14 @@ def _send_heartbeats(
     thread = threading.Thread(
         target=heartbeat_ctx.run, args=(heartbeat_loop,), daemon=True
     )
+    engine._stop_heartbeats = stop_heartbeats
     thread.start()
     engine.logger.debug("Started flow run heartbeat context")
 
     try:
         yield
     finally:
-        stop_event.set()
+        stop_heartbeats()
         if join_on_exit:
             thread.join(timeout=2)
         engine.logger.debug("Stopped flow run heartbeat context")
@@ -428,6 +441,15 @@ class BaseFlowRunEngine(Generic[P, R]):
         default_factory=FlowRunSuspensionRequest
     )
     _attempt_conclusion: EngineOutcomeReceipt | None = field(default=None, init=False)
+    # Stops the heartbeat thread started by the most recent `_send_heartbeats`.
+    # `set_state` calls it before proposing a terminal state: the server
+    # timestamps that state when it accepts it, but the heartbeat thread sees
+    # it only after the response arrives.  A heartbeat sent in that gap occurs
+    # after the terminal state, and an automation that expects every heartbeat
+    # to be followed by another event then fires on the finished run.
+    _stop_heartbeats: Callable[[], None] | None = field(
+        default=None, init=False, repr=False
+    )
 
     def __post_init__(self) -> None:
         if self.flow is None and self.flow_run_id is None:
@@ -761,6 +783,9 @@ class FlowRunEngine(BaseFlowRunEngine[P, R]):
         if self.short_circuit:
             return self.state
 
+        if state.is_final() and self._stop_heartbeats is not None:
+            self._stop_heartbeats()
+
         state = propose_state_sync(
             self.client, state, flow_run_id=self.flow_run.id, force=force
         )  # type: ignore
@@ -1500,6 +1525,9 @@ class AsyncFlowRunEngine(BaseFlowRunEngine[P, R]):
         if self.short_circuit:
             return self.state
 
+        if state.is_final() and self._stop_heartbeats is not None:
+            self._stop_heartbeats()
+
         state = await propose_state(
             self.client, state, flow_run_id=self.flow_run.id, force=force
         )  # type: ignore

{prefect-3.9.2.dev5.dist-info → prefect-3.9.2.dev6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: prefect
-Version: 3.9.2.dev5
+Version: 3.9.2.dev6
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://www.prefect.io/docs

{prefect-3.9.2.dev5.dist-info → prefect-3.9.2.dev6.dist-info}/RECORD

@@ -2,7 +2,7 @@ prefect/.prefectignore,sha256=awSprvKT0vI8a64mEOLrMxhxqcO-b0ERQeYpA2rNKVQ,390
 prefect/AGENTS.md,sha256=BI3DGSwZTEHIg_d3pwt2tGx659DXCUi9HgjpHhnfJRY,14281
 prefect/__init__.py,sha256=FtzUynR70uuE9WgVABo7cZkGGeNdL8Q1gJVTtmqvpyg,6772
 prefect/__main__.py,sha256=WFjw3kaYJY6pOTA7WDOgqjsz8zUEUZHCcj3P5wyVa-g,66
-prefect/_build_info.py,sha256=RwiYqHagFMM0E-yEq-jdf6ZvPj_eUae8mWxXc0m5GnM,185
+prefect/_build_info.py,sha256=IhZ_c3TJ-QIk6XHxP0n_7qUKMiJ2CO28iMufduaxXTY,185
 prefect/_flow_run_suspension.py,sha256=5zTTB7ZIBHzoS0pVrhNn23-9hK51qZ3CQA6C-azluC0,4144
 prefect/agent.py,sha256=dPvG1jDGD5HSH7aM2utwtk6RaJ9qg13XjkA0lAIgQmY,287
 prefect/artifacts.py,sha256=ydlAad-FUepGUvTzmDkSmZLFRdMHohqVhwKnufJkeOw,23955
@@ -12,7 +12,7 @@ prefect/context.py,sha256=0jvq1dacNBpEwm_t5GQC2ngg1yI7OYcHpIx_ap_j9t4,39646
 prefect/engine.py,sha256=uM2-08TDIPf9RILKrg0Hzc9zuhCbtPDo9CQjKw_1c-s,6301
 prefect/exceptions.py,sha256=MRWzDG8zVzl9cep90bfFS1JEAR3CXoXDyy7qaxWPnLg,12727
 prefect/filesystems.py,sha256=eCgfNECSwlaqB_9UEHt8RgeHZBsa54Jmq0mlVXH_oeg,28900
-prefect/flow_engine.py,sha256=h9I7lOhzXjbMh8Un0Pv_WieUAzwlBoUBJzDwYr9cFO4,93570
+prefect/flow_engine.py,sha256=PdnlnJf1uMWB6QUhNjZ9rzMVR1-gI978BgckgbmRsCk,94824
 prefect/flow_runs.py,sha256=heh7pdxDBNPU_uSa3zczJRoXra7t600AoAcBZFBdP28,30670
 prefect/flows.py,sha256=L9I26RDn5R6DpwRquYtYYSTEl_QM-DD9cL_po4iMBmc,150053
 prefect/futures.py,sha256=sbtEOrwyegEmIWjEHanidV-IcYICgrodHF8tSOmydLI,33343
@@ -1240,8 +1240,8 @@ prefect/workers/_worker_channel/_protocol.py,sha256=GpCNh1o3qmmqHA_UOOTge1QVC6IR
 prefect/workers/_worker_channel/_state.py,sha256=eQTFZtAVDZH1vVWps3SdeY6aW3qu2wx1UKYQXK3AyuE,5369
 prefect/workers/_worker_channel/_sync.py,sha256=G5G8_UaQYbeLebi5Mb1Z_KgGfXfyjXo5uT9la5_zwqY,14984
 prefect/workers/_worker_channel/_transport.py,sha256=6VXxiZKIu0rqImlif14ojoAL3UGOBTmNpnlYxNggaLE,9423
-prefect-3.9.2.dev5.dist-info/METADATA,sha256=qWqV55UHgyeeTmmhLth9W_JhnZAXfaWZb_yXFFOv_KU,14407
-prefect-3.9.2.dev5.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
-prefect-3.9.2.dev5.dist-info/entry_points.txt,sha256=HlY8up83iIq2vU2r33a0qSis4eOFSyb1mRH4l7Xt9X8,126
-prefect-3.9.2.dev5.dist-info/licenses/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
-prefect-3.9.2.dev5.dist-info/RECORD,,
+prefect-3.9.2.dev6.dist-info/METADATA,sha256=g_ByggJVO9wKEr5ZFH9BTAaQKm6hfamhgCyELtaU8cc,14407
+prefect-3.9.2.dev6.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
+prefect-3.9.2.dev6.dist-info/entry_points.txt,sha256=HlY8up83iIq2vU2r33a0qSis4eOFSyb1mRH4l7Xt9X8,126
+prefect-3.9.2.dev6.dist-info/licenses/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
+prefect-3.9.2.dev6.dist-info/RECORD,,