prefect-client 3.4.13.dev3__tar.gz → 3.4.14.dev2__tar.gz

{prefect_client-3.4.13.dev3 → prefect_client-3.4.14.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.4.13.dev3
+Version: 3.4.14.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.4.14.dev2/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.4.14.dev2"
+__build_date__ = "2025-08-19 08:09:09.042950+00:00"
+__git_commit__ = "fe7b84c9f7c5aae03463def73507a15421eb22fb"
+__dirty__ = False

{prefect_client-3.4.13.dev3 → prefect_client-3.4.14.dev2}/src/prefect/blocks/notifications.py

@@ -14,7 +14,7 @@ from prefect.utilities.asyncutils import sync_compatible
 from prefect.utilities.templating import apply_values, find_placeholders
 from prefect.utilities.urls import validate_restricted_url
 
-PREFECT_NOTIFY_TYPE_DEFAULT = "prefect_default"
+PREFECT_NOTIFY_TYPE_DEFAULT = "info"  # Use a valid apprise type as default
 
 
 class AbstractAppriseNotificationBlock(NotificationBlock, ABC):
@@ -22,24 +22,12 @@ class AbstractAppriseNotificationBlock(NotificationBlock, ABC):
     An abstract class for sending notifications using Apprise.
     """
 
-    notify_type: Literal["prefect_default", "info", "success", "warning", "failure"] = (
-        Field(
-            default=PREFECT_NOTIFY_TYPE_DEFAULT,
-            description=(
-                "The type of notification being performed; the prefect_default "
-                "is a plain notification that does not attach an image."
-            ),
-        )
+    notify_type: Literal["info", "success", "warning", "failure"] = Field(
+        default=PREFECT_NOTIFY_TYPE_DEFAULT,
+        description="The type of notification being performed.",
     )
 
     def __init__(self, *args: Any, **kwargs: Any):
-        from apprise import (
-            NOTIFY_TYPES,  # pyright: ignore[reportAttributeAccessIssue, reportUnknownVariableType] incomplete type hints in apprise
-        )
-
-        if PREFECT_NOTIFY_TYPE_DEFAULT not in NOTIFY_TYPES:
-            NOTIFY_TYPES += (PREFECT_NOTIFY_TYPE_DEFAULT,)  # pyright: ignore[reportUnknownVariableType]
-
         super().__init__(*args, **kwargs)
 
     def _start_apprise_client(self, url: SecretStr):
@@ -227,8 +215,7 @@ class PagerDutyWebHook(AbstractAppriseNotificationBlock):
         "https://docs.prefect.io/latest/automate/events/automations-triggers#sending-notifications-with-automations"
     )
 
-    # The default cannot be prefect_default because NotifyPagerDuty's
-    # PAGERDUTY_SEVERITY_MAP only has these notify types defined as keys
+    # PagerDuty requires a valid severity level from its PAGERDUTY_SEVERITY_MAP
     notify_type: Literal["info", "success", "warning", "failure"] = Field(  # pyright: ignore[reportIncompatibleVariableOverride]
         default="info", description="The severity of the notification."
     )

{prefect_client-3.4.13.dev3 → prefect_client-3.4.14.dev2}/src/prefect/client/orchestration/_flow_runs/client.py

@@ -317,6 +317,54 @@ class FlowRunClient(BaseClient):
 
         return FlowRun.model_validate_list(response.json())
 
+    def count_flow_runs(
+        self,
+        *,
+        flow_filter: "FlowFilter | None" = None,
+        flow_run_filter: "FlowRunFilter | None" = None,
+        task_run_filter: "TaskRunFilter | None" = None,
+        deployment_filter: "DeploymentFilter | None" = None,
+        work_pool_filter: "WorkPoolFilter | None" = None,
+        work_queue_filter: "WorkQueueFilter | None" = None,
+    ) -> int:
+        """
+        Returns the count of flow runs matching all criteria for flow runs.
+
+        Args:
+            flow_filter: filter criteria for flows
+            flow_run_filter: filter criteria for flow runs
+            task_run_filter: filter criteria for task runs
+            deployment_filter: filter criteria for deployments
+            work_pool_filter: filter criteria for work pools
+            work_queue_filter: filter criteria for work pool queues
+
+        Returns:
+            count of flow runs
+        """
+        body: dict[str, Any] = {
+            "flows": flow_filter.model_dump(mode="json") if flow_filter else None,
+            "flow_runs": (
+                flow_run_filter.model_dump(mode="json", exclude_unset=True)
+                if flow_run_filter
+                else None
+            ),
+            "task_runs": (
+                task_run_filter.model_dump(mode="json") if task_run_filter else None
+            ),
+            "deployments": (
+                deployment_filter.model_dump(mode="json") if deployment_filter else None
+            ),
+            "work_pools": (
+                work_pool_filter.model_dump(mode="json") if work_pool_filter else None
+            ),
+            "work_pool_queues": (
+                work_queue_filter.model_dump(mode="json") if work_queue_filter else None
+            ),
+        }
+
+        response = self.request("POST", "/flow_runs/count", json=body)
+        return response.json()
+
     def set_flow_run_state(
         self,
         flow_run_id: "UUID | str",
@@ -767,6 +815,54 @@ class FlowRunAsyncClient(BaseAsyncClient):
 
         return FlowRun.model_validate_list(response.json())
 
+    async def count_flow_runs(
+        self,
+        *,
+        flow_filter: "FlowFilter | None" = None,
+        flow_run_filter: "FlowRunFilter | None" = None,
+        task_run_filter: "TaskRunFilter | None" = None,
+        deployment_filter: "DeploymentFilter | None" = None,
+        work_pool_filter: "WorkPoolFilter | None" = None,
+        work_queue_filter: "WorkQueueFilter | None" = None,
+    ) -> int:
+        """
+        Returns the count of flow runs matching all criteria for flow runs.
+
+        Args:
+            flow_filter: filter criteria for flows
+            flow_run_filter: filter criteria for flow runs
+            task_run_filter: filter criteria for task runs
+            deployment_filter: filter criteria for deployments
+            work_pool_filter: filter criteria for work pools
+            work_queue_filter: filter criteria for work pool queues
+
+        Returns:
+            count of flow runs
+        """
+        body: dict[str, Any] = {
+            "flows": flow_filter.model_dump(mode="json") if flow_filter else None,
+            "flow_runs": (
+                flow_run_filter.model_dump(mode="json", exclude_unset=True)
+                if flow_run_filter
+                else None
+            ),
+            "task_runs": (
+                task_run_filter.model_dump(mode="json") if task_run_filter else None
+            ),
+            "deployments": (
+                deployment_filter.model_dump(mode="json") if deployment_filter else None
+            ),
+            "work_pools": (
+                work_pool_filter.model_dump(mode="json") if work_pool_filter else None
+            ),
+            "work_pool_queues": (
+                work_queue_filter.model_dump(mode="json") if work_queue_filter else None
+            ),
+        }
+
+        response = await self.request("POST", "/flow_runs/count", json=body)
+        return response.json()
+
     async def set_flow_run_state(
         self,
         flow_run_id: "UUID | str",

{prefect_client-3.4.13.dev3 → prefect_client-3.4.14.dev2}/src/prefect/client/schemas/schedules.py

@@ -287,7 +287,7 @@ class RRuleSchedule(PrefectBaseModel):
             kwargs = dict(dtstart=dtstart.replace(tzinfo=timezone))
             if until := _rrule_dt(exr, "_until"):
                 kwargs.update(until=until.replace(tzinfo=timezone))
-            localized_exrules.append(exr.replace(**kwargs))  # pyright: ignore[reportUnknownArgumentType, reportUnknownMemberType] missing type hints
+            localized_exrules.append(exr.replace(**kwargs))  # pyright: ignore[reportArgumentType, reportUnknownArgumentType, reportUnknownMemberType] missing type hints
         setattr(rrule, "_exrule", localized_exrules)
 
         # update rdates

{prefect_client-3.4.13.dev3 → prefect_client-3.4.14.dev2}/src/prefect/concurrency/_leases.py

@@ -109,6 +109,9 @@ async def amaintain_concurrency_lease(
     with AsyncCancelScope() as cancel_scope:
 
         def handle_lease_renewal_failure(task: asyncio.Task[None]):
+            if task.cancelled():
+                # Cancellation is the expected way for this loop to stop
+                return
             exc = task.exception()
             if exc:
                 try:

{prefect_client-3.4.13.dev3 → prefect_client-3.4.14.dev2}/src/prefect/flows.py

@@ -1806,7 +1806,7 @@ class FlowDecorator:
         flow_run_name: Optional[Union[Callable[[], str], str]] = None,
         retries: Optional[int] = None,
         retry_delay_seconds: Optional[Union[int, float]] = None,
-        task_runner: Optional[TaskRunner[PrefectFuture[R]]] = None,
+        task_runner: Optional[TaskRunner[PrefectFuture[Any]]] = None,
         description: Optional[str] = None,
         timeout_seconds: Union[int, float, None] = None,
         validate_parameters: bool = True,
@@ -1831,7 +1831,7 @@ class FlowDecorator:
         flow_run_name: Optional[Union[Callable[[], str], str]] = None,
         retries: Optional[int] = None,
         retry_delay_seconds: Union[int, float, None] = None,
-        task_runner: Optional[TaskRunner[PrefectFuture[R]]] = None,
+        task_runner: Optional[TaskRunner[PrefectFuture[Any]]] = None,
         description: Optional[str] = None,
         timeout_seconds: Union[int, float, None] = None,
         validate_parameters: bool = True,

{prefect_client-3.4.13.dev3 → prefect_client-3.4.14.dev2}/src/prefect/settings/models/tasks.py

@@ -23,6 +23,12 @@ class TasksRunnerSettings(PrefectBaseSettings):
         ),
     )
 
+    process_pool_max_workers: Optional[int] = Field(
+        default=None,
+        gt=0,
+        description="The maximum number of workers for ProcessPoolTaskRunner.",
+    )
+
 
 class TasksSchedulingSettings(PrefectBaseSettings):
     model_config: ClassVar[SettingsConfigDict] = build_settings_config(