prefect-client 3.8.0__tar.gz → 3.8.1.dev2__tar.gz

{prefect_client-3.8.0 → prefect_client-3.8.1.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.8.0
+Version: 3.8.1.dev2
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://docs.prefect.io

{prefect_client-3.8.0 → prefect_client-3.8.1.dev2}/src/prefect/AGENTS.md

@@ -23,6 +23,7 @@ There is no formal public/private boundary beyond the `_` prefix convention. Mod
 - **Engine ordering matters.** The engines apply features (retries, caching, result persistence, transactions) in a specific order. Changing the order or forgetting a feature in one engine path is the most common source of breakage.
 - **Set run metadata before state transitions.** Any run attribute that event subscribers observe at the moment of a state transition (e.g., custom `flow_run_name`) must be persisted to the server *before* `set_state()` is called for that transition. Metadata set after the call appears stale to subscribers of that state event.
 - **Sync and async must stay in sync.** Both `flow_engine.py` and `task_engine.py` have sync and async paths. Any behavior change must be applied to both.
+- **Event and log WebSocket subscribers are parallel implementations.** `events/clients.py::PrefectEventSubscriber` and `logging/clients.py::PrefectLogsSubscriber` duplicate the same reconnect/backoff/dedup logic, including the `reconnect_on_clean_close` option (used by `events/subscribers.py::FlowRunSubscriber` so a clean disconnect doesn't hide a terminal event). Mirror any reconnection-behavior change across both.
 - **Use `whenever` compat helpers for all datetime conversions on Python 3.13+.** `types/_datetime.py` provides `now()`, `_whenever_to_stdlib()`, `_whenever_zdt_from_py()`, and `_whenever_pdt_from_py()` to abstract over API differences between whenever 0.7.x–0.9.x and ≥ 0.10.0. The `_WHENEVER_NEW_API` flag (True when whenever ≥ 0.10.0) guards version-specific code. Never call `DateTime.now()` or `pendulum.now()` directly — use `now()` from `types/_datetime.py` instead. Avoid no-argument `datetime.datetime.now()` because it returns a tz-naive datetime that raises `TypeError` when compared to the tz-aware datetimes produced by `now()`. Explicitly timezone-aware stdlib calls such as `datetime.now(ZoneInfo("UTC"))` are allowed when needed to preserve timezone information. Never call `ZonedDateTime.from_py_datetime()`, `PlainDateTime.from_py_datetime()`, or `.py_datetime()` directly — use the helpers instead. Violations will silently break on whichever whenever version the helpers weren't written for.
 - **Flow and task engines advance state differently.** The flow engine makes blocking API calls to the server to propose and advance states. The task engine emits `prefect.task_run.*` events (delivered via WebSockets) but advances state locally through `set_state` calls with polling/backoff — do not assume the two engines work the same way.
 - **Flow state transitions go through the server.** The flow engine proposes states to the server, which accepts or rejects them via orchestration rules. The task engine, by contrast, manages state transitions locally via `set_state` and emits `prefect.task_run.*` events — it does not propose states to the server.

prefect_client-3.8.1.dev2/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.8.1.dev2"
+__build_date__ = "2026-07-27 09:12:40.986268+00:00"
+__git_commit__ = "d630146acb690f2967e89c3f429ec3425644d87a"
+__dirty__ = False

prefect_client-3.8.1.dev2/src/prefect/concurrency/context.py

@@ -0,0 +1,53 @@
+import logging
+from contextvars import ContextVar
+from typing import Any, ClassVar
+from uuid import UUID
+
+from typing_extensions import Self
+
+from prefect.client.orchestration import get_client
+from prefect.context import ContextModel, Field
+from prefect.logging.loggers import get_logger, get_run_logger
+
+
+def _cleanup_logger() -> "logging.Logger | logging.LoggerAdapter[logging.Logger]":
+    try:
+        # Use a run logger if available so failures reach the run logs
+        return get_run_logger()
+    except Exception:
+        return get_logger("concurrency")
+
+
+class ConcurrencyContext(ContextModel):
+    __var__: ClassVar[ContextVar[Self]] = ContextVar("concurrency")
+
+    # Track the leases that have been acquired but were not able to be released
+    # due to cancellation or some other error. These leases are revoked when
+    # the context manager exits.
+    cleanup_lease_ids: list[UUID] = Field(default_factory=lambda: [])
+
+    def __exit__(self, *exc_info: Any) -> None:
+        # Releasing these leases is best-effort: a failure for one lease must not
+        # strand the others or skip the context teardown below.
+        if self.cleanup_lease_ids:
+            logger = _cleanup_logger()
+            try:
+                with get_client(sync_client=True) as client:
+                    for lease_id in self.cleanup_lease_ids:
+                        try:
+                            client.release_concurrency_slots_with_lease(
+                                lease_id=lease_id
+                            )
+                        except Exception:
+                            logger.warning(
+                                "Failed to release concurrency lease %s during cleanup",
+                                lease_id,
+                                exc_info=True,
+                            )
+            except Exception:
+                logger.warning(
+                    "Failed to release concurrency leases during cleanup",
+                    exc_info=True,
+                )
+
+        return super().__exit__(*exc_info)

prefect_client-3.8.1.dev2/src/prefect/concurrency/v1/context.py

@@ -0,0 +1,56 @@
+import logging
+from contextvars import ContextVar
+from typing import Any, ClassVar
+from uuid import UUID
+
+from typing_extensions import Self
+
+from prefect.client.orchestration import get_client
+from prefect.context import ContextModel, Field
+from prefect.logging.loggers import get_logger, get_run_logger
+
+
+def _cleanup_logger() -> "logging.Logger | logging.LoggerAdapter[logging.Logger]":
+    try:
+        # Use a run logger if available so failures reach the run logs
+        return get_run_logger()
+    except Exception:
+        return get_logger("concurrency")
+
+
+class ConcurrencyContext(ContextModel):
+    __var__: ClassVar[ContextVar[Self]] = ContextVar("concurrency_v1")
+
+    # Track the limits that have been acquired but were not able to be released
+    # due to cancellation or some other error. These limits are released when
+    # the context manager exits.
+    cleanup_slots: list[tuple[list[str], float, UUID]] = Field(default_factory=list)
+
+    def __exit__(self, *exc_info: Any) -> None:
+        # Releasing these slots is best-effort: a failure for one entry must not
+        # strand the others or skip the context teardown below.
+        if self.cleanup_slots:
+            logger = _cleanup_logger()
+            try:
+                with get_client(sync_client=True) as client:
+                    for names, occupancy_seconds, task_run_id in self.cleanup_slots:
+                        try:
+                            client.decrement_v1_concurrency_slots(
+                                names=names,
+                                occupancy_seconds=occupancy_seconds,
+                                task_run_id=task_run_id,
+                            )
+                        except Exception:
+                            logger.warning(
+                                "Failed to release concurrency slots %s for task run %s during cleanup",
+                                names,
+                                task_run_id,
+                                exc_info=True,
+                            )
+            except Exception:
+                logger.warning(
+                    "Failed to release concurrency slots during cleanup",
+                    exc_info=True,
+                )
+
+        return super().__exit__(*exc_info)

prefect_client-3.8.0/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.8.0"
-__build_date__ = "2026-07-23 21:49:48.145061+00:00"
-__git_commit__ = "218fb46646bda2a229d74ce80c35dcf9baa2f0f3"
-__dirty__ = False

prefect_client-3.8.0/src/prefect/concurrency/context.py

@@ -1,25 +0,0 @@
-from contextvars import ContextVar
-from typing import Any, ClassVar
-from uuid import UUID
-
-from typing_extensions import Self
-
-from prefect.client.orchestration import get_client
-from prefect.context import ContextModel, Field
-
-
-class ConcurrencyContext(ContextModel):
-    __var__: ClassVar[ContextVar[Self]] = ContextVar("concurrency")
-
-    # Track the leases that have been acquired but were not able to be released
-    # due to cancellation or some other error. These leases are revoked when
-    # the context manager exits.
-    cleanup_lease_ids: list[UUID] = Field(default_factory=lambda: [])
-
-    def __exit__(self, *exc_info: Any) -> None:
-        if self.cleanup_lease_ids:
-            with get_client(sync_client=True) as client:
-                for lease_id in self.cleanup_lease_ids:
-                    client.release_concurrency_slots_with_lease(lease_id=lease_id)
-
-        return super().__exit__(*exc_info)

prefect_client-3.8.0/src/prefect/concurrency/v1/context.py

@@ -1,29 +0,0 @@
-from contextvars import ContextVar
-from typing import Any, ClassVar
-from uuid import UUID
-
-from typing_extensions import Self
-
-from prefect.client.orchestration import get_client
-from prefect.context import ContextModel, Field
-
-
-class ConcurrencyContext(ContextModel):
-    __var__: ClassVar[ContextVar[Self]] = ContextVar("concurrency_v1")
-
-    # Track the limits that have been acquired but were not able to be released
-    # due to cancellation or some other error. These limits are released when
-    # the context manager exits.
-    cleanup_slots: list[tuple[list[str], float, UUID]] = Field(default_factory=list)
-
-    def __exit__(self, *exc_info: Any) -> None:
-        if self.cleanup_slots:
-            with get_client(sync_client=True) as client:
-                for names, occupancy_seconds, task_run_id in self.cleanup_slots:
-                    client.decrement_v1_concurrency_slots(
-                        names=names,
-                        occupancy_seconds=occupancy_seconds,
-                        task_run_id=task_run_id,
-                    )
-
-        return super().__exit__(*exc_info)