prefect-client 3.8.9.dev6__tar.gz → 3.8.9.dev7__tar.gz

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: prefect-client
-Version: 3.8.9.dev6
+Version: 3.8.9.dev7
 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.9.dev7/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.8.9.dev7"
+__build_date__ = "2026-10-20 08:19:07.991479+00:00"
+__git_commit__ = "817bc93b6e828459e6853fa8b51ddba52515ad16"
+__dirty__ = False

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/src/prefect/events/__init__.py

@@ -42,6 +42,7 @@ from .actions import (
     PauseAutomation,
     ResumeAutomation,
     DeclareIncident,
+    UnknownAction,
 )
 from .clients import get_events_client, get_events_subscriber
 from .subscribers import FlowRunSubscriber
@@ -90,6 +91,7 @@ __all__ = [
     "PauseAutomation",
     "ResumeAutomation",
     "DeclareIncident",
+    "UnknownAction",
     "emit_event",
     "get_events_client",
     "get_events_subscriber",

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/src/prefect/events/actions.py

@@ -1,9 +1,16 @@
 import abc
 from datetime import timedelta
-from typing import Any, Dict, Optional, Union
+from typing import Annotated, Any, ClassVar, Dict, Optional, Union, get_args
 from uuid import UUID
 
-from pydantic import Field, model_validator
+from pydantic import (
+    ConfigDict,
+    Discriminator,
+    Field,
+    Tag,
+    field_validator,
+    model_validator,
+)
 from typing_extensions import Literal, Self, TypeAlias
 
 from prefect._internal.schemas.bases import PrefectBaseModel
@@ -314,3 +321,56 @@ ActionTypes: TypeAlias = Union[
     # Prefect Cloud only
     DeclareIncident,
 ]
+
+_KNOWN_ACTION_TYPES: frozenset[str] = frozenset(
+    action.model_fields["type"].default for action in get_args(ActionTypes)
+)
+
+
+class UnknownAction(Action):
+    """An action whose type this version of Prefect does not recognize.
+
+    Prefect Cloud can add action types before a client release knows about
+    them. Reading an automation that uses one of these types returns this
+    model instead of failing. The model keeps every field of the action, so
+    updating the automation sends the action back unchanged.
+    """
+
+    model_config: ClassVar[ConfigDict] = ConfigDict(extra="allow")
+
+    type: str = Field(
+        json_schema_extra={"not": {"enum": sorted(_KNOWN_ACTION_TYPES)}},
+    )
+
+    @field_validator("type")
+    @classmethod
+    def _type_is_not_known(cls, value: str) -> str:
+        if value in _KNOWN_ACTION_TYPES:
+            raise ValueError(
+                f"{value!r} is a known action type and cannot be an UnknownAction"
+            )
+        return value
+
+
+def _action_discriminator(value: Any) -> str:
+    """Sends an action with an unrecognized `type` to `UnknownAction` and every
+    other action, including one with no `type`, to `ActionTypes`."""
+    if isinstance(value, UnknownAction):
+        return "unknown"
+    action_type = (
+        value.get("type") if isinstance(value, dict) else getattr(value, "type", None)
+    )
+    if isinstance(action_type, str) and action_type not in _KNOWN_ACTION_TYPES:
+        return "unknown"
+    return "known"
+
+
+ActionTypesOrUnknown: TypeAlias = Annotated[
+    Union[
+        Annotated[ActionTypes, Tag("known")],
+        Annotated[UnknownAction, Tag("unknown")],
+    ],
+    Discriminator(_action_discriminator),
+]
+"""The action types an automation read from the API may contain: any of
+`ActionTypes`, or an `UnknownAction` for a type this version does not know"""

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/src/prefect/events/schemas/automations.py

@@ -22,17 +22,21 @@ from pydantic import (
     Field,
     PrivateAttr,
     Tag,
+    TypeAdapter,
+    ValidationError,
     field_validator,
     model_validator,
 )
 from typing_extensions import Self, TypeAlias
 
 from prefect._internal.schemas.bases import PrefectBaseModel
-from prefect.events.actions import ActionTypes, RunDeployment
+from prefect.events.actions import ActionTypes, ActionTypesOrUnknown, RunDeployment
 from prefect.utilities.collections import AutoEnum
 
 from .events import ResourceSpecification
 
+_timedelta_adapter: TypeAdapter[timedelta] = TypeAdapter(timedelta)
+
 # Dict form accepted by ResourceSpecification via Pydantic coercion
 ResourceSpecificationDict: TypeAlias = Dict[str, Union[str, List[str]]]
 
@@ -96,7 +100,7 @@ class Trigger(PrefectBaseModel, abc.ABC, extra="ignore"):  # type: ignore[call-a
             description=getattr(self, "description", ""),
             enabled=getattr(self, "enabled", True),
             trigger=trigger,
-            actions=self.actions(),
+            actions=list(self.actions()),
             owner_resource=self.owner_resource(),
         )
 
@@ -220,8 +224,15 @@ class EventTrigger(ResourceTrigger):
         posture: Optional[Posture] = data.get("posture")
         within: Optional[timedelta] = data.get("within")
 
-        if isinstance(within, (int, float)):
-            within = timedelta(seconds=within)
+        if within is not None and not isinstance(within, timedelta):
+            # Parse the raw value the same way the field would, so that every
+            # form pydantic accepts for a timedelta reaches the comparisons
+            # below as a timedelta.
+            try:
+                within = _timedelta_adapter.validate_python(within)
+            except ValidationError:
+                # The field validation reports the unparsable value.
+                return data
 
         if posture == Posture.Proactive:
             if not within or within == timedelta(0):
@@ -486,17 +497,17 @@ class AutomationCore(PrefectBaseModel, extra="ignore"):  # type: ignore[call-arg
         ),
     )
 
-    actions: List[ActionTypes] = Field(
+    actions: List[ActionTypesOrUnknown] = Field(
         default=...,
         description="The actions to perform when this Automation triggers",
     )
 
-    actions_on_trigger: List[ActionTypes] = Field(
+    actions_on_trigger: List[ActionTypesOrUnknown] = Field(
         default_factory=list,
         description="The actions to perform when an Automation goes into a triggered state",
     )
 
-    actions_on_resolve: List[ActionTypes] = Field(
+    actions_on_resolve: List[ActionTypesOrUnknown] = Field(
         default_factory=list,
         description="The actions to perform when an Automation goes into a resolving state",
     )

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/src/prefect/flow_runs.py

@@ -273,7 +273,7 @@ async def apause_flow_run(
             if wait_for_input:
                 # The flow run wanted input, so we need to load it and return it
                 # to the user.
-                await wait_for_input.aload(run_input_keyset)
+                return await wait_for_input.aload(run_input_keyset)
 
             return
 

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/src/prefect/flows.py

@@ -578,11 +578,11 @@ class Flow(Generic[P, R]):
                 else self.cache_result_in_memory
             ),
             log_prints=log_prints if log_prints is not NotSet else self.log_prints,
-            on_completion=on_completion or self.on_completion_hooks,
-            on_failure=on_failure or self.on_failure_hooks,
-            on_cancellation=on_cancellation or self.on_cancellation_hooks,
-            on_crashed=on_crashed or self.on_crashed_hooks,
-            on_running=on_running or self.on_running_hooks,
+            on_completion=on_completion or copy(self.on_completion_hooks),
+            on_failure=on_failure or copy(self.on_failure_hooks),
+            on_cancellation=on_cancellation or copy(self.on_cancellation_hooks),
+            on_crashed=on_crashed or copy(self.on_crashed_hooks),
+            on_running=on_running or copy(self.on_running_hooks),
         )
         new_flow._storage = self._storage
         new_flow._entrypoint = self._entrypoint

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/src/prefect/futures.py

@@ -584,7 +584,7 @@ class PrefectFutureList(list[PrefectFuture[R]], Iterator[PrefectFuture[R]]):
         timeout_message = (
             f"Timed out waiting for all futures to complete within {timeout} seconds"
         )
-        deadline = time.monotonic() + timeout if timeout is not None else None
+        deadline = time.monotonic() + timeout if timeout else None
 
         try:
             # `as_completed` de-duplicates internally; each unique future is
@@ -638,7 +638,7 @@ def as_completed(
     pending = unique_futures
     deadline: float | None = None
     try:
-        with timeout_context(timeout, timeout_exc_type=_WaitTimeoutError):
+        with timeout_context(timeout or None, timeout_exc_type=_WaitTimeoutError):
             done = {f for f in unique_futures if f._final_state}  # type: ignore[privateUsage]
             pending = unique_futures - done
             yield from done

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/src/prefect/task_engine.py

@@ -16,6 +16,7 @@ from typing import (
     TYPE_CHECKING,
     Any,
     AsyncGenerator,
+    Awaitable,
     Callable,
     Coroutine,
     Generator,
@@ -459,12 +460,14 @@ class SyncTaskRunEngine(BaseTaskRunEngine[P, R]):
                     message=f"Task run encountered unexpected exception: {repr(exc_or_state)}",
                 )
             )
-            if inspect.iscoroutinefunction(retry_condition):
-                should_retry = run_coro_as_sync(
-                    retry_condition(self.task, self.task_run, state)
-                )
-            elif inspect.isfunction(retry_condition):
+            if callable(retry_condition):
                 should_retry = retry_condition(self.task, self.task_run, state)
+                if inspect.isawaitable(should_retry):
+
+                    async def await_retry_condition(result: Awaitable[bool]) -> bool:
+                        return await result
+
+                    should_retry = run_coro_as_sync(await_retry_condition(should_retry))
             else:
                 should_retry = not retry_condition
             return should_retry
@@ -1077,10 +1080,10 @@ class AsyncTaskRunEngine(BaseTaskRunEngine[P, R]):
                     message=f"Task run encountered unexpected exception: {repr(exc_or_state)}",
                 )
             )
-            if inspect.iscoroutinefunction(retry_condition):
-                should_retry = await retry_condition(self.task, self.task_run, state)
-            elif inspect.isfunction(retry_condition):
+            if callable(retry_condition):
                 should_retry = retry_condition(self.task, self.task_run, state)
+                if inspect.isawaitable(should_retry):
+                    should_retry = await should_retry
             else:
                 should_retry = not retry_condition
             return should_retry

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/src/prefect/tasks.py

@@ -791,7 +791,7 @@ class Task(Generic[P, R]):
             fn=self.fn,
             name=name or self.name,
             description=description or self.description,
-            tags=tags or copy(self.tags),
+            tags=tags if tags is not None else copy(self.tags),
             version=self.version,
             cache_policy=cache_policy
             if cache_policy is not NotSet
@@ -842,9 +842,9 @@ class Task(Generic[P, R]):
             refresh_cache=(
                 refresh_cache if refresh_cache is not NotSet else self.refresh_cache
             ),
-            on_completion=on_completion or self.on_completion_hooks,
-            on_failure=on_failure or self.on_failure_hooks,
-            on_running=on_running or self.on_running_hooks,
+            on_completion=on_completion or copy(self.on_completion_hooks),
+            on_failure=on_failure or copy(self.on_failure_hooks),
+            on_running=on_running or copy(self.on_running_hooks),
             on_rollback=copy(self.on_rollback_hooks),
             on_commit=copy(self.on_commit_hooks),
             retry_condition_fn=retry_condition_fn or self.retry_condition_fn,
@@ -2258,9 +2258,10 @@ class MaterializingTask(Task[P, R]):
                 continue
 
             attr_name = param_to_attr.get(param_name, param_name)
-            init_kwargs[param_name] = task_kwargs.get(
-                param_name, getattr(self, attr_name)
-            )
+            value = getattr(self, attr_name)
+            if attr_name.endswith("_hooks"):
+                value = copy(value)
+            init_kwargs[param_name] = task_kwargs.get(param_name, value)
 
         return MaterializingTask(
             fn=self.fn,

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/src/prefect/utilities/callables/__init__.py

@@ -83,12 +83,21 @@ def get_call_parameters(
     # a FieldInfo object when the user writes `param: T = Field(default=..., ...)`.
     # Unwrap those to their actual default values.
     for key, value in arguments.items():
-        if isinstance(value, pydantic.fields.FieldInfo):
-            arguments[key] = value.default
+        arguments[key] = _resolve_field_info_default(value)
 
     return arguments
 
 
+def _resolve_field_info_default(value: Any) -> Any:
+    """
+    Resolve a `pydantic.Field(...)` used as a signature default to its actual
+    default value. Other values are returned unchanged.
+    """
+    if isinstance(value, pydantic.fields.FieldInfo):
+        return value.default
+    return value
+
+
 def get_parameter_defaults(
     fn: Callable[..., Any],
 ) -> dict[str, Any]:
@@ -101,7 +110,7 @@ def get_parameter_defaults(
 
     for name, param in signature.parameters.items():
         if param.default is not signature.empty:
-            parameter_defaults[name] = param.default
+            parameter_defaults[name] = _resolve_field_info_default(param.default)
 
     return parameter_defaults
 

{prefect_client-3.8.9.dev6 → prefect_client-3.8.9.dev7}/src/prefect/utilities/collections.py

@@ -208,7 +208,7 @@ def extract_instances(
 
     if len(types_collection) == 1:
         [type_] = types_collection
-        return ret[type_]
+        return ret.get(type_, [])
 
     return ret
 

prefect_client-3.8.9.dev6/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.8.9.dev6"
-__build_date__ = "2026-10-18 08:17:13.281213+00:00"
-__git_commit__ = "87b5243845e733ed938b21cf7b9aeda6d6a4e194"
-__dirty__ = False