prefect 3.6.7.dev3__tar.gz → 3.6.8__tar.gz

{prefect-3.6.7.dev3 → prefect-3.6.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect
-Version: 3.6.7.dev3
+Version: 3.6.8
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://docs.prefect.io
@@ -57,7 +57,7 @@ Requires-Dist: pydantic!=2.11.0,!=2.11.1,!=2.11.2,!=2.11.3,!=2.11.4,<3.0.0,>=2.1
 Requires-Dist: pydantic-core<3.0.0,>=2.12.0
 Requires-Dist: pydantic-extra-types<3.0.0,>=2.8.2
 Requires-Dist: pydantic-settings!=2.9.0,<3.0.0,>2.2.1
-Requires-Dist: pydocket>=0.13.0
+Requires-Dist: pydocket>=0.16.2
 Requires-Dist: python-dateutil<3.0.0,>=2.8.2
 Requires-Dist: python-slugify<9.0,>=5.0
 Requires-Dist: pytz<2026,>=2021.1
@@ -70,9 +70,8 @@ Requires-Dist: semver>=3.0.4
 Requires-Dist: sniffio<2.0.0,>=1.3.0
 Requires-Dist: sqlalchemy[asyncio]<3.0.0,>=2.0
 Requires-Dist: toml>=0.10.0
-Requires-Dist: typer<0.20.0,>=0.16.0
+Requires-Dist: typer<0.21.0,>=0.16.0
 Requires-Dist: typing-extensions<5.0.0,>=4.10.0
-Requires-Dist: uv>=0.6.0
 Requires-Dist: uvicorn!=0.29.0,>=0.14.0
 Requires-Dist: websockets<16.0,>=15.0.1
 Requires-Dist: whenever<0.10.0,>=0.7.3; python_version >= '3.13'
@@ -82,6 +81,8 @@ Provides-Extra: azure
 Requires-Dist: prefect-azure>=0.4.0; extra == 'azure'
 Provides-Extra: bitbucket
 Requires-Dist: prefect-bitbucket>=0.3.0; extra == 'bitbucket'
+Provides-Extra: bundles
+Requires-Dist: uv>=0.6.0; extra == 'bundles'
 Provides-Extra: dask
 Requires-Dist: prefect-dask>=0.3.0; extra == 'dask'
 Provides-Extra: databricks

{prefect-3.6.7.dev3 → prefect-3.6.8}/pyproject.toml

@@ -38,7 +38,7 @@ dependencies = [
     "pytz>=2021.1,<2026",
     "readchar>=4.0.0,<5.0.0",
     "sqlalchemy[asyncio]>=2.0,<3.0.0",
-    "typer>=0.16.0,<0.20.0",
+    "typer>=0.16.0,<0.21.0",
     # Client dependencies
     # If you modify this list, make the same modification in client/pyproject.toml
     "anyio>=4.4.0,<5.0.0",
@@ -78,10 +78,9 @@ dependencies = [
     "uvicorn>=0.14.0,!=0.29.0",
     "websockets>=15.0.1,<16.0",
     "whenever>=0.7.3,<0.10.0; python_version>='3.13'",
-    "uv>=0.6.0",
     "semver>=3.0.4",
     "pluggy>=1.6.0",
-    "pydocket>=0.13.0",
+    "pydocket>=0.16.2",
 ]
 [project.urls]
 Changelog = "https://github.com/PrefectHQ/prefect/releases"
@@ -98,6 +97,9 @@ otel = [
     "opentelemetry-test-utils>=0.48b0,<1.0.0",
 ]
 
+# bundles extra for bundle features (requires uv)
+bundles = ["uv>=0.6.0"]
+
 # Infrastructure extras
 aws = ["prefect-aws>=0.5.8"]
 azure = ["prefect-azure>=0.4.0"]
@@ -158,7 +160,7 @@ dev = [
     "watchfiles>=1.0.4",
     "respx>=0.22.0",
     "codespell==2.4.1",
-    "ruff==0.14.8",
+    "ruff==0.14.10",
     "mypy==1.15.0",
     "types-cachetools>=5.3.1",
     "types-PyYAML>=6.0.12",

{prefect-3.6.7.dev3 → prefect-3.6.8}/src/prefect/__init__.py

@@ -12,21 +12,22 @@ if TYPE_CHECKING:
     from importlib.machinery import ModuleSpec
     from .main import (
         allow_failure,
+        aresume_flow_run,
+        aserve,
         flow,
         Flow,
         get_client,
         get_run_logger,
+        pause_flow_run,
+        resume_flow_run,
+        serve,
         State,
+        suspend_flow_run,
         tags,
         task,
         Task,
         Transaction,
         unmapped,
-        serve,
-        aserve,
-        pause_flow_run,
-        resume_flow_run,
-        suspend_flow_run,
     )
     from prefect.deployments.runner import deploy
 
@@ -118,6 +119,7 @@ _initialize_plugins()
 
 _public_api: dict[str, tuple[Optional[str], str]] = {
     "allow_failure": (__spec__.parent, ".main"),
+    "aresume_flow_run": (__spec__.parent, ".main"),
     "aserve": (__spec__.parent, ".main"),
     "deploy": (__spec__.parent, ".deployments.runner"),
     "flow": (__spec__.parent, ".main"),
@@ -140,6 +142,7 @@ _public_api: dict[str, tuple[Optional[str], str]] = {
 __all__ = [
     "__version__",
     "allow_failure",
+    "aresume_flow_run",
     "aserve",
     "deploy",
     "flow",

prefect-3.6.8/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.6.8"
+__build_date__ = "2025-12-24 19:25:05.398703+00:00"
+__git_commit__ = "8ae19ff770f3b9095c478314f845b0d699ba852f"
+__dirty__ = False

{prefect-3.6.7.dev3 → prefect-3.6.8}/src/prefect/_experimental/bundles/__init__.py

@@ -38,14 +38,18 @@ logger: logging.Logger = get_logger(__name__)
 
 
 def _get_uv_path() -> str:
+    """
+    Get the path to the uv binary.
+
+    First tries to use the uv Python package to find the binary.
+    Falls back to "uv" string (assumes uv is in PATH).
+    """
     try:
         import uv
 
-        uv_path = uv.find_uv_bin()
+        return uv.find_uv_bin()
     except (ImportError, ModuleNotFoundError, FileNotFoundError):
-        uv_path = "uv"
-
-    return uv_path
+        return "uv"
 
 
 class SerializedBundle(TypedDict):

{prefect-3.6.7.dev3 → prefect-3.6.8}/src/prefect/blocks/notifications.py

@@ -679,7 +679,7 @@ class MattermostWebhook(AbstractAppriseNotificationBlock):
                 token=self.token.get_secret_value(),
                 fullpath=self.path,
                 host=self.hostname,
-                botname=self.botname,
+                user=self.botname,
                 channels=self.channels,
                 include_image=self.include_image,
                 port=self.port,
@@ -844,11 +844,12 @@ class CustomWebhookNotificationBlock(NotificationBlock):
             ' "{{tokenFromSecrets}}"}'
         ],
     )
-    form_data: Optional[dict[str, str]] = Field(
+    form_data: Optional[dict[str, str] | str] = Field(
         default=None,
         title="Form Data",
         description=(
-            "Send form data as payload. Should not be used together with _JSON Data_."
+            "Send form data as payload. Should not be used together with _JSON Data_. "
+            "Can be a dictionary for form-encoded data or a string for raw body content."
         ),
         examples=[
             '{"text": "{{subject}}\\n{{body}}", "title": "{{name}}", "token":'
@@ -882,13 +883,18 @@ class CustomWebhookNotificationBlock(NotificationBlock):
                 "name": self.name,
             }
         )
+        # httpx uses 'data' for form-encoded dicts, 'content' for raw string/bytes
+        if isinstance(self.form_data, str):
+            data_key = "content"
+        else:
+            data_key = "data"
         # do substution
         return apply_values(
             {
                 "method": self.method,
                 "url": self.url,
                 "params": self.params,
-                "data": self.form_data,
+                data_key: self.form_data,
                 "json": self.json_data,
                 "headers": self.headers,
                 "cookies": self.cookies,

{prefect-3.6.7.dev3 → prefect-3.6.8}/src/prefect/cache_policies.py

@@ -1,6 +1,7 @@
 import inspect
 from copy import deepcopy
 from dataclasses import dataclass, field
+from logging import Logger
 from pathlib import Path
 from typing import (
     TYPE_CHECKING,
@@ -16,6 +17,7 @@ from typing_extensions import Self
 
 from prefect.context import TaskRunContext
 from prefect.exceptions import HashError
+from prefect.logging import get_logger
 from prefect.utilities.hashing import hash_objects
 
 if TYPE_CHECKING:
@@ -25,6 +27,8 @@ if TYPE_CHECKING:
 
 STABLE_TRANSFORMS: dict[type, Callable[[Any], Any]] = {}
 
+logger: Logger = get_logger(__name__)
+
 
 def _register_stable_transforms() -> None:
     """
@@ -291,6 +295,13 @@ class TaskSource(CachePolicy):
     ) -> Optional[str]:
         if not task_ctx:
             return None
+
+        # Use stored source code if available (works after cloudpickle serialization)
+        lines = getattr(task_ctx.task, "source_code", None)
+        if lines is not None:
+            return hash_objects(lines, raise_on_failure=True)
+
+        # Fall back to inspect.getsource for local execution
         try:
             lines = inspect.getsource(task_ctx.task)
         except TypeError:
@@ -300,6 +311,7 @@ class TaskSource(CachePolicy):
                 lines = task_ctx.task.fn.__code__.co_code
             else:
                 raise
+
         return hash_objects(lines, raise_on_failure=True)
 
 

{prefect-3.6.7.dev3 → prefect-3.6.8}/src/prefect/cli/flow_run.py

@@ -10,9 +10,15 @@ import signal
 import threading
 import webbrowser
 from types import FrameType
-from typing import List, Optional
+from typing import TYPE_CHECKING, List, Optional
 from uuid import UUID
 
+from prefect.utilities.callables import get_call_parameters, parameters_to_args_kwargs
+
+if TYPE_CHECKING:
+    from prefect.client.orchestration import PrefectClient
+    from prefect.client.schemas.objects import FlowRun
+
 import httpx
 import orjson
 import typer
@@ -32,7 +38,7 @@ from prefect.client.schemas.sorting import FlowRunSort, LogSort
 from prefect.exceptions import ObjectNotFound
 from prefect.logging import get_logger
 from prefect.runner import Runner
-from prefect.states import State
+from prefect.states import State, exception_to_crashed_state
 from prefect.types._datetime import human_friendly_diff
 from prefect.utilities.asyncutils import run_sync_in_worker_thread
 from prefect.utilities.urls import url_for
@@ -43,6 +49,65 @@ flow_run_app: PrefectTyper = PrefectTyper(
 app.add_typer(flow_run_app, aliases=["flow-runs"])
 
 LOGS_DEFAULT_PAGE_SIZE = 200
+
+
+async def _get_flow_run_by_id_or_name(
+    client: "PrefectClient",
+    id_or_name: str,
+) -> "FlowRun":
+    """
+    Resolve a flow run identifier that could be either a UUID or a name.
+
+    Flow run names are not guaranteed to be unique, so this function will
+    error if multiple flow runs match the given name.
+
+    Args:
+        client: The Prefect client to use for API calls
+        id_or_name: Either a UUID string or a flow run name
+
+    Returns:
+        The matching FlowRun object
+
+    Raises:
+        typer.Exit: If flow run not found, or if multiple flow runs match the name
+    """
+    from prefect.client.schemas.filters import FlowRunFilterName
+
+    # First, try parsing as UUID
+    try:
+        flow_run_id = UUID(id_or_name)
+        try:
+            return await client.read_flow_run(flow_run_id)
+        except ObjectNotFound:
+            exit_with_error(f"Flow run '{id_or_name}' not found!")
+    except ValueError:
+        # Not a valid UUID, treat as a name
+        pass
+
+    # Query by name (exact match)
+    flow_runs = await client.read_flow_runs(
+        flow_run_filter=FlowRunFilter(name=FlowRunFilterName(any_=[id_or_name])),
+        limit=100,  # Reasonable limit for displaying matches
+    )
+
+    if not flow_runs:
+        exit_with_error(f"Flow run '{id_or_name}' not found!")
+
+    if len(flow_runs) == 1:
+        return flow_runs[0]
+
+    # Multiple matches - show all and exit with error
+    lines = [f"Multiple flow runs found with name '{id_or_name}':\n"]
+    for fr in flow_runs:
+        state_name = fr.state.name if fr.state else "unknown"
+        timestamp = fr.start_time or fr.created
+        timestamp_str = timestamp.strftime("%Y-%m-%d %H:%M:%S") if timestamp else "N/A"
+        lines.append(f"  - {fr.id} ({state_name}, {timestamp_str})")
+
+    lines.append("\nPlease retry using an explicit flow run ID.")
+    exit_with_error("\n".join(lines))
+
+
 LOGS_WITH_LIMIT_FLAG_DEFAULT_NUM_LOGS = 20
 
 logger: "logging.Logger" = get_logger(__name__)
@@ -276,6 +341,189 @@ async def cancel(id: UUID):
     exit_with_success(f"Flow run '{id}' was successfully scheduled for cancellation.")
 
 
+@flow_run_app.command()
+async def retry(
+    id_or_name: str = typer.Argument(
+        ...,
+        help="The flow run ID (UUID) or name to retry.",
+    ),
+    entrypoint: Optional[str] = typer.Option(
+        None,
+        "--entrypoint",
+        "-e",
+        help=(
+            "The path to a file containing the flow to run, and the name of the flow "
+            "function, in the format `path/to/file.py:flow_function_name`. "
+            "Required if the flow run does not have an associated deployment."
+        ),
+    ),
+):
+    """
+    Retry a failed or completed flow run.
+
+    The flow run can be specified by either its UUID or its name. If multiple
+    flow runs have the same name, you must use the UUID to disambiguate.
+
+    If the flow run has an associated deployment, it will be scheduled for retry
+    and a worker will pick it up. If there is no deployment, you must provide
+    an --entrypoint to the flow code, and the flow will execute locally.
+
+    \b
+    Examples:
+        $ prefect flow-run retry abc123-def456-7890-...
+        $ prefect flow-run retry my-flow-run-name
+        $ prefect flow-run retry abc123 --entrypoint ./flows/my_flow.py:my_flow
+    """
+    from prefect.flow_engine import run_flow
+    from prefect.flows import load_flow_from_entrypoint
+    from prefect.states import Scheduled
+
+    terminal_states = {
+        StateType.COMPLETED,
+        StateType.FAILED,
+        StateType.CANCELLED,
+        StateType.CRASHED,
+    }
+
+    async with get_client() as client:
+        # Resolve flow run by ID or name
+        flow_run = await _get_flow_run_by_id_or_name(client, id_or_name)
+        flow_run_id = flow_run.id
+
+        # Validate flow run is in terminal state
+        if flow_run.state is None or flow_run.state.type not in terminal_states:
+            current_state = flow_run.state.type.value if flow_run.state else "unknown"
+            exit_with_error(
+                f"Flow run '{flow_run_id}' is in state '{current_state}' and cannot be retried. "
+                f"Only flow runs in terminal states (COMPLETED, FAILED, CANCELLED, CRASHED) can be retried."
+            )
+
+        # Branch based on deployment association
+        if flow_run.deployment_id:
+            # Deployment-based retry: set state to Scheduled and exit
+            # Use force=True to bypass orchestration rules that prevent state transitions
+            # from terminal states (e.g., CANCELLED -> SCHEDULED)
+            scheduled_state = Scheduled(message="Retried via CLI")
+            try:
+                result = await client.set_flow_run_state(
+                    flow_run_id=flow_run_id, state=scheduled_state, force=True
+                )
+            except ObjectNotFound:
+                exit_with_error(f"Flow run '{flow_run_id}' not found!")
+
+            if result.status == SetStateStatus.ABORT:
+                exit_with_error(
+                    f"Flow run '{flow_run_id}' could not be retried. Reason: '{result.details.reason}'"
+                )
+
+            exit_with_success(
+                f"Flow run '{flow_run_id}' has been scheduled for retry. "
+                "A worker will pick it up shortly."
+            )
+        else:
+            # Local retry: require entrypoint and execute synchronously
+            if not entrypoint:
+                exit_with_error(
+                    f"Flow run '{flow_run_id}' does not have an associated deployment. "
+                    "Please provide an --entrypoint to the flow code.\n\n"
+                    f"Example: prefect flow-run retry {flow_run_id} --entrypoint ./flows/my_flow.py:my_flow"
+                )
+
+            # Load the flow from entrypoint
+            try:
+                flow = load_flow_from_entrypoint(entrypoint, use_placeholder_flow=False)
+            except Exception as exc:
+                exit_with_error(
+                    f"Failed to load flow from entrypoint '{entrypoint}': {exc}"
+                )
+
+            # Check if this is an infrastructure-bound flow
+            from prefect.flows import InfrastructureBoundFlow
+
+            if isinstance(flow, InfrastructureBoundFlow):
+                app.console.print(
+                    f"Retrying flow run '{flow_run_id}' on remote infrastructure "
+                    f"(work pool: {flow.work_pool})..."
+                )
+
+                try:
+                    # Use the retry method which handles remote execution
+                    await flow.retry(flow_run)
+                except Exception as exc:
+                    exit_with_error(f"Flow run failed: {exc}")
+
+                # Re-fetch to get final state
+                flow_run = await client.read_flow_run(flow_run_id)
+                final_state = flow_run.state.type.value if flow_run.state else "unknown"
+
+                if flow_run.state and flow_run.state.is_completed():
+                    exit_with_success(
+                        f"Flow run '{flow_run_id}' completed successfully."
+                    )
+                else:
+                    exit_with_error(
+                        f"Flow run '{flow_run_id}' finished with state: {final_state}"
+                    )
+            else:
+                # Regular local execution path
+                # Set state to Scheduled with force=True to bypass deployment check
+                scheduled_state = Scheduled(message="Retried via CLI (local execution)")
+                try:
+                    result = await client.set_flow_run_state(
+                        flow_run_id=flow_run_id, state=scheduled_state, force=True
+                    )
+                except ObjectNotFound:
+                    exit_with_error(f"Flow run '{flow_run_id}' not found!")
+
+                if result.status == SetStateStatus.ABORT:
+                    exit_with_error(
+                        f"Flow run '{flow_run_id}' could not be retried. Reason: '{result.details.reason}'"
+                    )
+
+                app.console.print(f"Executing flow run '{flow_run_id}' locally...")
+
+                # Re-fetch the flow run to get updated state
+                flow_run = await client.read_flow_run(flow_run_id)
+
+                try:
+                    call_args, call_kwargs = parameters_to_args_kwargs(
+                        flow.fn, flow_run.parameters if flow_run else {}
+                    )
+                    parameters = get_call_parameters(flow.fn, call_args, call_kwargs)
+                except Exception as exc:
+                    state = await exception_to_crashed_state(exc)
+                    await client.set_flow_run_state(
+                        flow_run_id=flow_run_id, state=state, force=True
+                    )
+                    exit_with_error(
+                        "Failed to use parameters from previous attempt. Please ensure the flow signature has not changed since the last run."
+                    )
+
+                # Execute the flow synchronously, reusing the existing flow run
+                try:
+                    run_flow(
+                        flow=flow,
+                        flow_run=flow_run,
+                        return_type="state",
+                        parameters=parameters,
+                    )
+                except Exception as exc:
+                    exit_with_error(f"Flow run failed: {exc}")
+
+                # Re-fetch to get final state
+                flow_run = await client.read_flow_run(flow_run_id)
+                final_state = flow_run.state.type.value if flow_run.state else "unknown"
+
+                if flow_run.state and flow_run.state.is_completed():
+                    exit_with_success(
+                        f"Flow run '{flow_run_id}' completed successfully."
+                    )
+                else:
+                    exit_with_error(
+                        f"Flow run '{flow_run_id}' finished with state: {final_state}"
+                    )
+
+
 @flow_run_app.command()
 async def logs(
     id: UUID,

{prefect-3.6.7.dev3 → prefect-3.6.8}/src/prefect/cli/server.py

@@ -709,13 +709,29 @@ def run_manager_process():
 
     logger.debug("Manager process started. Starting services...")
     try:
-        asyncio.run(Service.run_services())
+        asyncio.run(_run_all_services())
     except KeyboardInterrupt:
         pass
     finally:
         logger.debug("Manager process has exited.")
 
 
+async def _run_all_services() -> None:
+    """Run Service-based services and docket-based perpetual services."""
+    from docket import Docket
+
+    from prefect.server.api.background_workers import background_worker
+    from prefect.server.services.base import Service
+    from prefect.settings.context import get_current_settings
+
+    docket_url = get_current_settings().server.docket.url
+
+    async with Docket(name="prefect", url=docket_url) as docket:
+        async with background_worker(docket, ephemeral=False, webserver_only=False):
+            # Run Service-based services (will block until shutdown)
+            await Service.run_services()
+
+
 # public, user-facing `prefect server services` commands
 @services_app.command(aliases=["ls"])
 def list_services():
@@ -772,7 +788,7 @@ def start_services(
     if not background:
         app.console.print("\n[blue]Starting services... Press CTRL+C to stop[/]\n")
         try:
-            asyncio.run(Service.run_services())
+            asyncio.run(_run_all_services())
         except KeyboardInterrupt:
             pass
         app.console.print("\n[green]All services stopped.[/]")

{prefect-3.6.7.dev3 → prefect-3.6.8}/src/prefect/client/orchestration/_deployments/client.py

@@ -157,6 +157,12 @@ class DeploymentClient(BaseClient):
             payload["version_info"] = deployment_create.version_info.model_dump(
                 mode="json"
             )
+        if deployment_create.concurrency_options:
+            payload["concurrency_options"] = (
+                deployment_create.concurrency_options.model_dump(
+                    mode="json", exclude_unset=True
+                )
+            )
 
         try:
             response = self.request("POST", "/deployments/", json=payload)
@@ -823,6 +829,12 @@ class DeploymentAsyncClient(BaseAsyncClient):
             payload["version_info"] = deployment_create.version_info.model_dump(
                 mode="json"
             )
+        if deployment_create.concurrency_options:
+            payload["concurrency_options"] = (
+                deployment_create.concurrency_options.model_dump(
+                    mode="json", exclude_unset=True
+                )
+            )
 
         try:
             response = await self.request("POST", "/deployments/", json=payload)

{prefect-3.6.7.dev3 → prefect-3.6.8}/src/prefect/events/clients.py

@@ -281,7 +281,11 @@ class PrefectEventsClient(EventsClient):
         # Don't handle any errors in the initial connection, because these are most
         # likely a permission or configuration issue that should propagate
         await super().__aenter__()
-        await self._reconnect()
+        try:
+            await self._reconnect()
+        except Exception as e:
+            self._log_connection_error(e)
+            raise
         return self
 
     async def __aexit__(
@@ -298,6 +302,18 @@ class PrefectEventsClient(EventsClient):
         message = f"EventsClient(id={id(self)}): " + message
         logger.debug(message, *args, **kwargs)
 
+    def _log_connection_error(self, error: Exception) -> None:
+        logger.warning(
+            "Unable to connect to %r. "
+            "Please check your network settings to ensure websocket connections "
+            "to the API are allowed. Otherwise event data (including task run data) may be lost. "
+            "Reason: %s. "
+            "Set PREFECT_DEBUG_MODE=1 to see the full error.",
+            self._events_socket_url,
+            str(error),
+            exc_info=PREFECT_DEBUG_MODE.value(),
+        )
+
     async def _reconnect(self) -> None:
         logger.debug("Reconnecting websocket connection.")
 
@@ -315,15 +331,10 @@ class PrefectEventsClient(EventsClient):
             await pong
             logger.debug("Pong received. Websocket connected.")
         except Exception as e:
-            # The client is frequently run in a background thread
-            # so we log an additional warning to ensure
-            # surfacing the error to the user.
-            logger.warning(
-                "Unable to connect to %r. "
-                "Please check your network settings to ensure websocket connections "
-                "to the API are allowed. Otherwise event data (including task run data) may be lost. "
-                "Reason: %s. "
-                "Set PREFECT_DEBUG_MODE=1 to see the full error.",
+            # Log at debug level during reconnection attempts - the warning will
+            # only be logged if all reconnection attempts fail (in _emit)
+            logger.debug(
+                "Unable to connect to %r, will retry. Reason: %s",
                 self._events_socket_url,
                 str(e),
                 exc_info=PREFECT_DEBUG_MODE.value(),
@@ -391,10 +402,11 @@ class PrefectEventsClient(EventsClient):
                 await self._checkpoint()
 
                 return
-            except ConnectionClosed:
+            except ConnectionClosed as e:
                 self._log_debug("Got ConnectionClosed error.")
                 if i == self._reconnection_attempts:
-                    # this was our final chance, raise the most recent error
+                    # this was our final chance, log warning and raise
+                    self._log_connection_error(e)
                     raise
 
                 if i > 2: