prefect 3.6.7__py3-none-any.whl → 3.6.8__py3-none-any.whl

prefect/_build_info.py

@@ -1,5 +1,5 @@
 # Generated by versioningit
-__version__ = "3.6.7"
-__build_date__ = "2025-12-18 19:54:23.981675+00:00"
-__git_commit__ = "ebfef643e5d55b20a7b0b6a80e9756987ac415cb"
+__version__ = "3.6.8"
+__build_date__ = "2025-12-24 19:25:13.006611+00:00"
+__git_commit__ = "8ae19ff770f3b9095c478314f845b0d699ba852f"
 __dirty__ = False

prefect/blocks/notifications.py

@@ -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/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/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/flows.py

@@ -2204,6 +2204,61 @@ class InfrastructureBoundFlow(Flow[P, R]):
 
         return run_coro_as_sync(submit_func())
 
+    async def retry(
+        self,
+        flow_run: "FlowRun",
+    ) -> R | State[R]:
+        """
+        EXPERIMENTAL: This method is experimental and may be removed or changed in future
+            releases.
+
+        Retry an existing flow run on remote infrastructure.
+
+        This method allows retrying a flow run that was previously executed,
+        reusing the same flow run ID and incrementing the run_count.
+
+        Args:
+            flow_run: The existing flow run to retry
+            return_state: If True, return the final state instead of the result
+
+        Returns:
+            The flow result or final state
+
+        Example:
+            ```python
+            from prefect import flow
+            from prefect_aws.experimental import ecs
+
+            @ecs(work_pool="my-pool")
+            @flow
+            def my_flow():
+                ...
+
+            # Original run
+            my_flow()  # Creates flow run abc123
+
+            # Later, retry the same flow run
+            flow_run = client.read_flow_run("abc123")
+            await my_flow.retry(flow_run)
+            ```
+        """
+        try:
+            async with self.worker_cls(work_pool_name=self.work_pool) as worker:
+                future = await worker.submit(
+                    flow=self,
+                    parameters=flow_run.parameters,
+                    job_variables=self.job_variables,
+                    flow_run=flow_run,
+                )
+                return await future.aresult()
+        except (ExceptionGroup, BaseExceptionGroup) as exc:
+            # For less verbose tracebacks
+            exceptions = exc.exceptions
+            if len(exceptions) == 1:
+                raise exceptions[0] from None
+            else:
+                raise
+
     def submit_to_work_pool(
         self, *args: P.args, **kwargs: P.kwargs
     ) -> PrefectFlowRunFuture[R]:

prefect/futures.py

@@ -373,7 +373,7 @@ class PrefectFlowRunFuture(PrefectFuture[R]):
     async def wait_async(self, timeout: float | None = None) -> None:
         if self._final_state:
             logger.debug(
-                "Final state already set for %s. Returning...", self.task_run_id
+                "Final state already set for %s. Returning...", self.flow_run_id
             )
             return
 
@@ -425,7 +425,7 @@ class PrefectFlowRunFuture(PrefectFuture[R]):
             await self.wait_async(timeout=timeout)
             if not self._final_state:
                 raise TimeoutError(
-                    f"Task run {self.task_run_id} did not complete within {timeout} seconds"
+                    f"Flow run {self.flow_run_id} did not complete within {timeout} seconds"
                 )
 
         return await self._final_state.aresult(raise_on_failure=raise_on_failure)

prefect/server/api/background_workers.py

@@ -1,9 +1,11 @@
 import asyncio
 from contextlib import asynccontextmanager
+from logging import Logger
 from typing import Any, AsyncGenerator, Callable
 
 from docket import Docket, Worker
 
+from prefect.logging import get_logger
 from prefect.server.api.flow_runs import delete_flow_run_logs
 from prefect.server.api.task_runs import delete_task_run_logs
 from prefect.server.events.services import triggers as _triggers_module  # noqa: F401
@@ -20,6 +22,8 @@ from prefect.server.services.perpetual_services import (
 )
 from prefect.server.services.repossessor import revoke_expired_lease
 
+logger: Logger = get_logger(__name__)
+
 # Task functions to register with docket for background processing
 task_functions: list[Callable[..., Any]] = [
     # Simple background tasks (from Alex's PR #19377)
@@ -43,7 +47,7 @@ async def background_worker(
     webserver_only: bool = False,
 ) -> AsyncGenerator[None, None]:
     worker_task: asyncio.Task[None] | None = None
-    try:
+    async with Worker(docket) as worker:
         # Register background task functions
         docket.register_collection(
             "prefect.server.api.background_workers:task_functions"
@@ -54,14 +58,14 @@ async def background_worker(
             docket, ephemeral=ephemeral, webserver_only=webserver_only
         )
 
-        async with Worker(docket) as worker:
+        try:
             worker_task = asyncio.create_task(worker.run_forever())
             yield
 
-    finally:
-        if worker_task:
-            worker_task.cancel()
-            try:
-                await worker_task
-            except asyncio.CancelledError:
-                pass
+        finally:
+            if worker_task:
+                worker_task.cancel()
+                try:
+                    await worker_task
+                except asyncio.CancelledError:
+                    pass

prefect/server/events/schemas/events.py

@@ -34,6 +34,7 @@ from prefect.settings import (
     PREFECT_EVENTS_MAXIMUM_LABELS_PER_RESOURCE,
     PREFECT_EVENTS_MAXIMUM_RELATED_RESOURCES,
 )
+from prefect.utilities.urls import url_for
 
 if TYPE_CHECKING:
     import logging
@@ -212,6 +213,12 @@ class ReceivedEvent(Event):
         description="When the event was received by Prefect Cloud",
     )
 
+    @property
+    def url(self) -> Optional[str]:
+        """Returns the UI URL for this event, allowing users to link to events
+        in automation templates without parsing date strings."""
+        return url_for(self, url_type="ui")
+
     def as_database_row(self) -> dict[str, Any]:
         row = self.model_dump()
         row["resource_id"] = self.resource.id

prefect/tasks.py

@@ -456,6 +456,14 @@ class Task(Generic[P, R]):
         update_wrapper(self, fn)
         self.fn = fn
 
+        # Capture source code for cache key computation
+        # This is stored on the task so it survives cloudpickle serialization
+        # to remote environments where the source file is not available
+        try:
+            self.source_code: str | None = inspect.getsource(fn)
+        except (TypeError, OSError):
+            self.source_code = None
+
         # the task is considered async if its function is async or an async
         # generator
         self.isasync: bool = inspect.iscoroutinefunction(

prefect/utilities/schema_tools/validation.py

@@ -257,7 +257,7 @@ def preprocess_schema(
 
     if "definitions" in schema:  # Also process definitions for reused models
         definitions = cast(dict[str, Any], schema["definitions"])
-        for definition in definitions.values():
+        for definition in list(definitions.values()):
             if "properties" in definition:
                 required_fields = definition.get("required", [])
                 process_properties(

prefect/utilities/urls.py

@@ -4,7 +4,7 @@ import socket
 import urllib.parse
 from logging import Logger
 from string import Formatter
-from typing import TYPE_CHECKING, Any, Literal, Optional, Union
+from typing import TYPE_CHECKING, Any, Literal, Optional, Union, cast
 from urllib.parse import urlparse
 from uuid import UUID
 
@@ -251,9 +251,13 @@ def url_for(
     )
     assert url_format is not None
 
-    if isinstance(obj, ReceivedEvent):
+    # Use duck-typing to handle both client-side and server-side ReceivedEvent
+    if name == "received-event" and hasattr(obj, "occurred"):
+        # Cast to ReceivedEvent for type checking - we've verified it has the
+        # required attributes via hasattr and name check above
+        event = cast(ReceivedEvent, obj)
         url = url_format.format(
-            occurred=obj.occurred.strftime("%Y-%m-%d"), obj_id=obj_id
+            occurred=event.occurred.strftime("%Y-%m-%d"), obj_id=obj_id
         )
     else:
         obj_keys = [

prefect/workers/base.py

@@ -766,6 +766,7 @@ class BaseWorker(abc.ABC, Generic[C, V, R]):
         flow: "Flow[..., FR]",
         parameters: dict[str, Any] | None = None,
         job_variables: dict[str, Any] | None = None,
+        flow_run: "FlowRun | None" = None,
     ) -> "PrefectFlowRunFuture[FR]":
         """
         EXPERIMENTAL: The interface for this method is subject to change.
@@ -775,9 +776,11 @@ class BaseWorker(abc.ABC, Generic[C, V, R]):
         Args:
             flow: The flow to submit
             parameters: The parameters to pass to the flow
+            job_variables: Job variables for infrastructure configuration
+            flow_run: Optional existing flow run to retry (reuses ID instead of creating new)
 
         Returns:
-            A flow run object
+            A flow run future
         """
         warnings.warn(
             "Ad-hoc flow submission via workers is experimental. The interface "
@@ -793,6 +796,7 @@ class BaseWorker(abc.ABC, Generic[C, V, R]):
                 flow=flow,
                 parameters=parameters,
                 job_variables=job_variables,
+                flow_run=flow_run,
             ),
         )
         return PrefectFlowRunFuture(flow_run_id=flow_run.id)
@@ -803,9 +807,17 @@ class BaseWorker(abc.ABC, Generic[C, V, R]):
         parameters: dict[str, Any] | None = None,
         job_variables: dict[str, Any] | None = None,
         task_status: anyio.abc.TaskStatus["FlowRun"] | None = None,
+        flow_run: "FlowRun | None" = None,
     ):
         """
         Submits a flow for the worker to kick off execution for.
+
+        Args:
+            flow: The flow to submit
+            parameters: The parameters to pass to the flow
+            job_variables: Job variables for infrastructure configuration
+            task_status: Task status for signaling when the flow run is ready
+            flow_run: Optional existing flow run to retry (reuses ID instead of creating new)
         """
         from prefect._experimental.bundles import (
             aupload_bundle_to_storage,
@@ -862,28 +874,40 @@ class BaseWorker(abc.ABC, Generic[C, V, R]):
         job_variables = (job_variables or {}) | {"command": " ".join(execute_command)}
         parameters = parameters or {}
 
-        # Create a parent task run if this is a child flow run to ensure it shows up as a child flow in the UI
-        parent_task_run = None
-        if flow_run_ctx := FlowRunContext.get():
-            parent_task = Task[Any, Any](
-                name=flow.name,
-                fn=flow.fn,
-                version=flow.version,
+        if flow_run is None:
+            # Create new flow run (standard behavior)
+            # Create a parent task run if this is a child flow run to ensure it shows up as a child flow in the UI
+            parent_task_run = None
+            if flow_run_ctx := FlowRunContext.get():
+                parent_task = Task[Any, Any](
+                    name=flow.name,
+                    fn=flow.fn,
+                    version=flow.version,
+                )
+                parent_task_run = await parent_task.create_run(
+                    flow_run_context=flow_run_ctx,
+                    parameters=parameters,
+                )
+
+            flow_run = await self.client.create_flow_run(
+                flow,
+                parameters=flow.serialize_parameters(parameters),
+                state=Pending(),
+                job_variables=job_variables,
+                work_pool_name=self.work_pool.name,
+                tags=TagsContext.get().current_tags,
+                parent_task_run_id=getattr(parent_task_run, "id", None),
             )
-            parent_task_run = await parent_task.create_run(
-                flow_run_context=flow_run_ctx,
-                parameters=parameters,
+        else:
+            # Reuse existing flow run - set state to Pending for retry
+            await self.client.set_flow_run_state(
+                flow_run_id=flow_run.id,
+                state=Pending(message="Retrying on remote infrastructure"),
+                force=True,
             )
+            # Re-fetch to get updated state
+            flow_run = await self.client.read_flow_run(flow_run.id)
 
-        flow_run = await self.client.create_flow_run(
-            flow,
-            parameters=flow.serialize_parameters(parameters),
-            state=Pending(),
-            job_variables=job_variables,
-            work_pool_name=self.work_pool.name,
-            tags=TagsContext.get().current_tags,
-            parent_task_run_id=getattr(parent_task_run, "id", None),
-        )
         if task_status is not None:
             # Emit the flow run object to .submit to allow it to return a future as soon as possible
             task_status.started(flow_run)

prefect/workers/process.py

@@ -263,18 +263,27 @@ class ProcessWorker(
         parameters: dict[str, Any] | None = None,
         job_variables: dict[str, Any] | None = None,
         task_status: anyio.abc.TaskStatus["FlowRun"] | None = None,
+        flow_run: "FlowRun | None" = None,
     ):
         from prefect._experimental.bundles import (
             create_bundle_for_flow_run,
         )
 
-        flow_run = await self.client.create_flow_run(
-            flow,
-            parameters=parameters,
-            state=Pending(),
-            job_variables=job_variables,
-            work_pool_name=self.work_pool.name,
-        )
+        if flow_run is None:
+            flow_run = await self.client.create_flow_run(
+                flow,
+                parameters=parameters,
+                state=Pending(),
+                job_variables=job_variables,
+                work_pool_name=self.work_pool.name,
+            )
+        else:
+            # Reuse existing flow run - set state to Pending for retry
+            await self.client.set_flow_run_state(
+                flow_run.id,
+                Pending(),
+                force=True,
+            )
         if task_status is not None:
             # Emit the flow run object to .submit to allow it to return a future as soon as possible
             task_status.started(flow_run)

{prefect-3.6.7.dist-info → prefect-3.6.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect
-Version: 3.6.7
+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,7 +70,7 @@ 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: uvicorn!=0.29.0,>=0.14.0
 Requires-Dist: websockets<16.0,>=15.0.1

{prefect-3.6.7.dist-info → prefect-3.6.8.dist-info}/RECORD

@@ -2,7 +2,7 @@ prefect/.prefectignore,sha256=awSprvKT0vI8a64mEOLrMxhxqcO-b0ERQeYpA2rNKVQ,390
 prefect/AGENTS.md,sha256=qmCZAuKIF9jQyp5TrW_T8bsM_97-QaiCoQp71A_b2Lg,1008
 prefect/__init__.py,sha256=-cFYePxZEyhJ4zSCwCu9QhTgvA6yRoUYMlAluL8voyE,5441
 prefect/__main__.py,sha256=WFjw3kaYJY6pOTA7WDOgqjsz8zUEUZHCcj3P5wyVa-g,66
-prefect/_build_info.py,sha256=95B1mmlb6G30YylFu5MG8GuW0dSygM8wu90-AA1FpeQ,180
+prefect/_build_info.py,sha256=wDAQcSSV0PgD03tSk9mYPD8xeBX61wboIbNbFQJypLw,180
 prefect/_result_records.py,sha256=S6QmsODkehGVSzbMm6ig022PYbI6gNKz671p_8kBYx4,7789
 prefect/_states.py,sha256=_BIcTNbnExppu4sRIbeOHxdt5DeaIUAa9qKE8sNIkQ0,9775
 prefect/_versioning.py,sha256=YqR5cxXrY4P6LM1Pmhd8iMo7v_G2KJpGNdsf4EvDFQ0,14132
@@ -10,15 +10,15 @@ prefect/_waiters.py,sha256=Ia2ITaXdHzevtyWIgJoOg95lrEXQqNEOquHvw3T33UQ,9026
 prefect/agent.py,sha256=dPvG1jDGD5HSH7aM2utwtk6RaJ9qg13XjkA0lAIgQmY,287
 prefect/artifacts.py,sha256=ZdMLJeJGK82hibtRzbsVa-g95dMa0D2UP1LiESoXmf4,23951
 prefect/automations.py,sha256=ZzPxn2tINdlXTQo805V4rIlbXuNWxd7cdb3gTJxZIeY,12567
-prefect/cache_policies.py,sha256=jH1aDW6vItTcsEytuTCrNYyjbq87IQPwdOgF0yxiUts,12749
+prefect/cache_policies.py,sha256=K0cymkMbEQBuwY1fSehefllN4Y4RblpNzYsh-LtjdXM,13156
 prefect/context.py,sha256=zA5j-7HJIdfOft92NW8dxkvgqnKkC-MhhOHfAmM5DoE,34706
 prefect/engine.py,sha256=uB5JN4l045i5JTlRQNT1x7MwlSiGQ5Bop2Q6jHHOgxY,3699
 prefect/exceptions.py,sha256=E67przt2JKW2hzXQGC2XMIMxM8PNtnDunwhW55VUsM8,12598
 prefect/filesystems.py,sha256=PrDsWxT5mfmJSs3ib3pNaKkjEEBExY2kGrhh5fPWFIM,22393
 prefect/flow_engine.py,sha256=bWGHTJ3WZs7taLAzbDvkr8Gvg_NgB7riHmjPtstj8xA,61774
 prefect/flow_runs.py,sha256=gZBtxwSk2k-bUXUvb_Eipg00F2hEm5_batIse85ngXE,18495
-prefect/flows.py,sha256=9lacxyuc2K16KcwcauOdsggt3f21NZDjJ4g8F9AxVI0,128521
-prefect/futures.py,sha256=HwxR17RRuBHT_r5c7XuhmuYG8NnreFCUEuO_FpRBxPw,25670
+prefect/flows.py,sha256=mV7B-WJJU7oBa_qv1-V4fvuv-luShPpDU253M0xh2us,130277
+prefect/futures.py,sha256=F0cLoC_DDqNMCgjFDcvHNt2jjusqndTt9S0dQd1DQ-k,25670
 prefect/main.py,sha256=swco0ugs87OPf0nVFtHSmFUAyyMkNFUKC_6Gncp-FFU,2547
 prefect/plugins.py,sha256=6bPNLs5Cab1VqwFAlfwJi7Wj3ASx0ZLdT5_-s8_Blyo,2510
 prefect/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -30,7 +30,7 @@ prefect/task_engine.py,sha256=OtcVWHcp45TLLraz-gK5a6exlYQhbSxLSSHpdRM9NFY,70691
 prefect/task_runners.py,sha256=UlDBdwIB0YBcTG6rMf3ICnGUOZt9WNscaXt2QonigBM,34524
 prefect/task_runs.py,sha256=2wPYMCJet_9PoT8rbacvwt7B7noGR3ksHEP1xXUiXpU,10093
 prefect/task_worker.py,sha256=RifZ3bOl6ppoYPiOAd4TQp2_GEw9eDQoW483rq1q52Q,20805
-prefect/tasks.py,sha256=Vip7I60wvW-ZF-hAx1-ObtQ3sw1JUhC0dK22S0gChQI,82741
+prefect/tasks.py,sha256=Twbmri_cw3Uq2FVwXzpcaytHFhZPtBWrYpisxD1TPNs,83099
 prefect/transactions.py,sha256=JzRHA_B6Oy-XwmMqZUeiBLOvaf-aCEAdxPEQsYI2opM,26819
 prefect/variables.py,sha256=L1WrzyGF47GMACI8Jji8nlCcAAl1wtLhJmhf4qV0-CQ,9526
 prefect/_experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -90,7 +90,7 @@ prefect/blocks/__init__.py,sha256=D0hB72qMfgqnBB2EMZRxUxlX9yLfkab5zDChOwJZmkY,22
 prefect/blocks/abstract.py,sha256=mpOAWopSR_RrzdxeurBTXVSKisP8ne-k8LYos-tp7go,17021
 prefect/blocks/core.py,sha256=V7a00ewzL37wPkSOrAtdty0gxpVxtqTt3RYOQUfiW9I,67055
 prefect/blocks/fields.py,sha256=1m507VVmkpOnMF_7N-qboRjtw4_ceIuDneX3jZ3Jm54,63
-prefect/blocks/notifications.py,sha256=3Rbicq2IGBLM9B0nfHOLeEs9GXuVWW1VyOjmli6ieFs,37042
+prefect/blocks/notifications.py,sha256=5GCwOyg3lnQepSbLFwnLg0hxN2To9MExwytFI7KWjuI,37345
 prefect/blocks/redis.py,sha256=fn6nR_ZoFfJqO_sWc-4WUY3D2No7RSXUntgZtRFNXYo,7682
 prefect/blocks/system.py,sha256=9OE9gs0cH0bH6DV5dvf4oprthWBKA6Zh56aSnn2pIU4,2325
 prefect/blocks/webhook.py,sha256=xylFigbDOsn-YzxahkTzNqYwrIA7wwS6204P0goLY3A,2907
@@ -109,7 +109,7 @@ prefect/cli/dev.py,sha256=1toGWrEvskTjir1agQXsCVWezFuvZjCkjNsFYExBmNE,11996
 prefect/cli/events.py,sha256=E_QDm9eEDKxS64QRdmYsp7qPXQXfgIpNq2h575foDxw,6076
 prefect/cli/experimental.py,sha256=skZ_c-agYqMT-YlDzfqjt6AvoXussbaIVGByYlUw12Q,5267
 prefect/cli/flow.py,sha256=lBHC9y6ZeLgeZifsfd7CukGYqp7XrbaLBE7X1yh6OME,5789
-prefect/cli/flow_run.py,sha256=7Ju-3k8Px_-Xm0yRkHubtVu8gGbJdl-bMrY9INLBGEM,14684
+prefect/cli/flow_run.py,sha256=8I-9UkJBzp_hLx8QS1e78O7aJOrXp0m5tasao_2HVHI,24685
 prefect/cli/flow_runs_watching.py,sha256=nOeLoTn1a-jzR4A1PTjC6reaFO8Qds3bQhhQ40xlUFA,6805
 prefect/cli/global_concurrency_limit.py,sha256=AiFBv2tqrs4QI5mASDa-TGodXSIpvpuRbwcJek8nmr0,14080
 prefect/cli/profile.py,sha256=v1nc2oaWFAwlZNv3yKjcY3qMhencgHkN4-QF-AEQBsA,14712
@@ -290,7 +290,7 @@ prefect/server/api/__init__.py,sha256=SpRTXHC6ApqR4_Y3wNHM7TvvH9FZ8N3tH-RZIgmubg
 prefect/server/api/admin.py,sha256=nINYSrux7XPAV4MMDQUts3X2dddrc3mJtd3iPl5N-jI,2644
 prefect/server/api/artifacts.py,sha256=B19bxxnADiVTo0XtzvsbAHHtsJwr7S1IunjGl-qsSd4,7314
 prefect/server/api/automations.py,sha256=KlQuhEOOKkxAI6f5aZedCKMELi99MdknulIleRP-JVE,8285
-prefect/server/api/background_workers.py,sha256=yMoNgw2thesg56_Uxjihio9rA9Ngikgi3TLOHmSKwA0,2281
+prefect/server/api/background_workers.py,sha256=ywc5bWP9x_b-zBgSu1MJPwwbfzcU-jR5iR0p5mgxj9E,2414
 prefect/server/api/block_capabilities.py,sha256=0x1vtC2CtSRVcCWydgNbylmfv_LnJHIoPnt7hRiJkN4,828
 prefect/server/api/block_documents.py,sha256=Hj-J10XMnfJGa2-AbJfZHun-iZrUwzxG8h7cXGfBXjg,5838
 prefect/server/api/block_schemas.py,sha256=H9kxnKPlTvz62SMcX1ssHnQ8KACU2728ebCtSp2iLcs,5508
@@ -577,7 +577,7 @@ prefect/server/events/ordering/db.py,sha256=5yf1Hk1qz4GsKPBG95YsGn2r5UEOG0hhJxld
 prefect/server/events/ordering/memory.py,sha256=il-LWxYFEVpj-HgwvIJL_8MgpO4IVk5g1dOLLg8Fd08,12601
 prefect/server/events/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/server/events/schemas/automations.py,sha256=0oEbdn4oRuHKMiqVaeWJdhDcDpiKmSqvVJvx_hSOFNU,27249
-prefect/server/events/schemas/events.py,sha256=R5gJZ9dowyuwZcNPyRWzjH0iZGMad1QuykWy_ludLzk,12137
+prefect/server/events/schemas/events.py,sha256=k10wMIBPEUTMDFQCJy1frT8LEP7O0EvZxvL5blzBkAU,12419
 prefect/server/events/schemas/labelling.py,sha256=McGy7dq6Ry2GY3ejnMQnkuL_h77F5MnHXQkyCdePlLU,3103
 prefect/server/events/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prefect/server/events/services/actions.py,sha256=o8m89I740YCZfO7DQgyngvHgIXOzpT86pjOvvvDeRmA,1850
@@ -883,20 +883,20 @@ prefect/utilities/slugify.py,sha256=57Vb14t13F3zm1P65KAu8nVeAz0iJCd1Qc5eMG-R5y8,
 prefect/utilities/templating.py,sha256=AnJ-w7wqVl2nfqKHQawEGhloxCU_zihrCtfx5pWFcYU,16017
 prefect/utilities/text.py,sha256=cuXb5EwRP5qFV7w-3_axEft4rDIJAMS8jgCg0kqNGKQ,758
 prefect/utilities/timeout.py,sha256=y7ILFZDpQMnmKB6XloulGDRUYRwnJR_b-XecQLNS7Ts,1265
-prefect/utilities/urls.py,sha256=AtwAt_uBjixUsOqDPTkS23eIbLdvDsWct32ewgcq5uY,9184
+prefect/utilities/urls.py,sha256=fUcUKnVQkzpfcARBdUIbfZGnvd6hy2tS31Jqhiy2u44,9476
 prefect/utilities/visualization.py,sha256=5rywcA_q2D0KHb5ZoZnrZ_A-cXaAFcQ9-wcZaCn99fM,7421
 prefect/utilities/schema_tools/__init__.py,sha256=At3rMHd2g_Em2P3_dFQlFgqR_EpBwrYtU2N2OJd0pDE,345
 prefect/utilities/schema_tools/hydration.py,sha256=NkRhWkNfxxFmVGhNDfmxdK_xeKaEhs3a42q83Sg9cT4,9436
-prefect/utilities/schema_tools/validation.py,sha256=Wix26IVR-ZJ32-6MX2pHhrwm3reB-Q4iB6_phn85OKE,10743
+prefect/utilities/schema_tools/validation.py,sha256=UhRLWStdT9__u4yz-mnc3lcdt-VJmRGJBGV_f_9um-Y,10749
 prefect/workers/__init__.py,sha256=EaM1F0RZ-XIJaGeTKLsXDnfOPHzVWk5bk0_c4BVS44M,64
-prefect/workers/base.py,sha256=n0tcQWGg_gQX1zJ-0WHnSCQMzYyg69Lx8fuwSduoGik,64663
+prefect/workers/base.py,sha256=uZFMqtIs1I4WtTXIUjeD9Sd-KQ8hy30a-Gl3-EWOuyc,65883
 prefect/workers/block.py,sha256=dPvG1jDGD5HSH7aM2utwtk6RaJ9qg13XjkA0lAIgQmY,287
 prefect/workers/cloud.py,sha256=dPvG1jDGD5HSH7aM2utwtk6RaJ9qg13XjkA0lAIgQmY,287
-prefect/workers/process.py,sha256=jvxBQyR8-G4Svt0as-BNg94YpvRMYZBIXKtNwC0igNY,11902
+prefect/workers/process.py,sha256=O_HZ39lBjedQat4gmIh2SiC2N_Q5q7grBGD13HXB2eo,12235
 prefect/workers/server.py,sha256=bWnYfMfJf5_IO3y3aJOpia7p9lFKC3ZZjiMvHox-UKY,1992
 prefect/workers/utilities.py,sha256=VfPfAlGtTuDj0-Kb8WlMgAuOfgXCdrGAnKMapPSBrwc,2483
-prefect-3.6.7.dist-info/METADATA,sha256=gC6jkDLoTHWSorFX05xqxpOdh0aphEjs3z__A511F2E,13654
-prefect-3.6.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-prefect-3.6.7.dist-info/entry_points.txt,sha256=HlY8up83iIq2vU2r33a0qSis4eOFSyb1mRH4l7Xt9X8,126
-prefect-3.6.7.dist-info/licenses/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
-prefect-3.6.7.dist-info/RECORD,,
+prefect-3.6.8.dist-info/METADATA,sha256=5PYAj4eUkiXV0gSWisj1g3yw6i-FDltadp-xyjMhzn4,13654
+prefect-3.6.8.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+prefect-3.6.8.dist-info/entry_points.txt,sha256=HlY8up83iIq2vU2r33a0qSis4eOFSyb1mRH4l7Xt9X8,126
+prefect-3.6.8.dist-info/licenses/LICENSE,sha256=MCxsn8osAkzfxKC4CC_dLcUkU8DZLkyihZ8mGs3Ah3Q,11357
+prefect-3.6.8.dist-info/RECORD,,