prefect-client 3.7.5.dev4__tar.gz → 3.7.5.dev5__tar.gz

{prefect_client-3.7.5.dev4 → prefect_client-3.7.5.dev5}/PKG-INFO

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

prefect_client-3.7.5.dev5/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.7.5.dev5"
+__build_date__ = "2026-06-17 09:45:07.057987+00:00"
+__git_commit__ = "7107dafe23e062b1f8eb4239e85a0a9ef756e79e"
+__dirty__ = False

{prefect_client-3.7.5.dev4 → prefect_client-3.7.5.dev5}/src/prefect/_internal/concurrency/threads.py

@@ -38,9 +38,18 @@ def _reset_after_fork_in_child():
 
     This handler is called by os.register_at_fork() in the child process after fork().
     """
+    global _global_loop, _run_sync_loop
+
     for instance in list(_active_instances):
         instance.reset_for_fork()
 
+    # Clear global references so get_global_loop() / get_run_sync_loop() create
+    # fresh instances.  After fork the old EventLoopThread objects have a stale
+    # Thread whose is_alive() still returns True (the dead thread's tstate lock
+    # is held), so the "is the thread alive?" guard in the getters never fires.
+    _global_loop = None
+    _run_sync_loop = None
+
 
 # Register fork handler if supported (POSIX systems)
 if hasattr(os, "register_at_fork"):

{prefect_client-3.7.5.dev4 → prefect_client-3.7.5.dev5}/src/prefect/server/api/background_workers.py

@@ -14,6 +14,7 @@ from prefect.server.models.work_queues import mark_work_queues_ready
 from prefect.server.services.cancellation_cleanup import (
     cancel_child_task_runs,
     cancel_subflow_run,
+    handle_cancelling_timeout,
 )
 from prefect.server.services.db_vacuum import (
     vacuum_events_with_retention_overrides,
@@ -40,6 +41,7 @@ task_functions: list[Callable[..., Any]] = [
     delete_task_run_logs,
     delete_flow_run_logs,
     # Find-and-flood pattern tasks used by perpetual services
+    handle_cancelling_timeout,
     cancel_child_task_runs,
     cancel_subflow_run,
     fail_expired_pause,

{prefect_client-3.7.5.dev4 → prefect_client-3.7.5.dev5}/src/prefect/server/api/flow_runs.py

@@ -13,13 +13,14 @@ from uuid import UUID
 import orjson
 import sqlalchemy as sa
 from docket import Depends as DocketDepends
-from docket import Retry
+from docket import Docket, Retry
 from fastapi import (
     Body,
     Depends,
     HTTPException,
     Path,
     Query,
+    Request,
     Response,
 )
 from fastapi.encoders import jsonable_encoder
@@ -53,6 +54,9 @@ from prefect.server.schemas.responses import (
     FlowRunPaginationResponse,
     OrchestrationResult,
 )
+from prefect.server.services.cancellation_cleanup import (
+    maybe_schedule_cancelling_timeout_check_for_state,
+)
 from prefect.server.utilities.server import PrefectRouter
 from prefect.types import DateTime
 from prefect.types._datetime import earliest_possible_datetime, now
@@ -66,9 +70,41 @@ logger: "logging.Logger" = get_logger("server.api")
 router: PrefectRouter = PrefectRouter(prefix="/flow_runs", tags=["Flow Runs"])
 
 
+def _get_request_docket(request: Request) -> Docket | None:
+    return getattr(request.app.state, "docket", None)
+
+
+async def _maybe_schedule_cancelling_timeout_check_for_state(
+    *,
+    request: Request,
+    flow_run_id: UUID,
+    state: schemas.states.State | None,
+) -> None:
+    docket = _get_request_docket(request)
+    if docket is None:
+        return
+
+    try:
+        await maybe_schedule_cancelling_timeout_check_for_state(
+            docket=docket,
+            flow_run_id=flow_run_id,
+            state=state,
+        )
+    except Exception:
+        logger.exception(
+            "Failed to schedule CANCELLING timeout check; allowing accepted "
+            "state transition to proceed",
+            extra={
+                "flow_run_id": str(flow_run_id),
+                "flow_run_state_id": str(state.id) if state and state.id else None,
+            },
+        )
+
+
 @router.post("/")
 async def create_flow_run(
     flow_run: schemas.actions.FlowRunCreate,
+    request: Request,
     db: PrefectDBInterface = Depends(provide_database_interface),
     response: Response = None,  # type: ignore
     created_by: Optional[schemas.core.CreatedBy] = Depends(dependencies.get_created_by),
@@ -124,13 +160,27 @@ async def create_flow_run(
             flow_run=flow_run_object,
             orchestration_parameters=orchestration_parameters,
         )
-        if model.created >= right_now:
-            response.status_code = status.HTTP_201_CREATED
-
-        return schemas.responses.FlowRunResponse.model_validate(
+        created = model.created >= right_now
+        timeout_check_state = (
+            schemas.states.State.from_orm_without_result(model.state)
+            if created and model.state
+            else None
+        )
+        flow_run_id = model.id
+        flow_run_response = schemas.responses.FlowRunResponse.model_validate(
             model, from_attributes=True
         )
 
+    await _maybe_schedule_cancelling_timeout_check_for_state(
+        request=request,
+        flow_run_id=flow_run_id,
+        state=timeout_check_state,
+    )
+    if created:
+        response.status_code = status.HTTP_201_CREATED
+
+    return flow_run_response
+
 
 @router.patch("/{id:uuid}", status_code=status.HTTP_204_NO_CONTENT)
 async def update_flow_run(
@@ -669,6 +719,7 @@ async def bulk_delete_flow_runs(
 
 @router.post("/bulk_set_state")
 async def bulk_set_flow_run_state(
+    request: Request,
     flow_runs: Optional[schemas.filters.FlowRunFilter] = Body(
         None, description="Filter criteria for flow runs to update"
     ),
@@ -718,6 +769,7 @@ async def bulk_set_flow_run_state(
 
     # Process flow runs sequentially to avoid session conflicts
     for flow_run in db_flow_runs:
+        state_to_schedule: schemas.states.State | None = None
         async with db.session_context(
             begin_transaction=True, with_for_update=True
         ) as session:
@@ -739,6 +791,11 @@ async def bulk_set_flow_run_state(
                         details=orchestration_result.details,
                     )
                 )
+                if (
+                    orchestration_result.status
+                    == schemas.responses.SetStateStatus.ACCEPT
+                ):
+                    state_to_schedule = orchestration_result.state
             except Exception as e:
                 results.append(
                     FlowRunOrchestrationResult(
@@ -748,6 +805,14 @@ async def bulk_set_flow_run_state(
                         details=schemas.responses.StateAbortDetails(reason=str(e)),
                     )
                 )
+                continue
+
+        if state_to_schedule is not None:
+            await _maybe_schedule_cancelling_timeout_check_for_state(
+                request=request,
+                flow_run_id=flow_run.id,
+                state=state_to_schedule,
+            )
 
     return FlowRunBulkSetStateResponse(results=results)
 
@@ -755,6 +820,7 @@ async def bulk_set_flow_run_state(
 @router.post("/{id:uuid}/set_state")
 async def set_flow_run_state(
     response: Response,
+    request: Request,
     flow_run_id: UUID = Path(..., description="The flow run id", alias="id"),
     state: schemas.actions.StateCreate = Body(..., description="The intended state."),
     force: bool = Body(
@@ -796,6 +862,13 @@ async def set_flow_run_state(
             client_version=client_version,
         )
 
+    if orchestration_result.status == schemas.responses.SetStateStatus.ACCEPT:
+        await _maybe_schedule_cancelling_timeout_check_for_state(
+            request=request,
+            flow_run_id=flow_run_id,
+            state=orchestration_result.state,
+        )
+
     # set the 201 if a new state was created
     if orchestration_result.state and orchestration_result.state.timestamp >= right_now:
         response.status_code = status.HTTP_201_CREATED

{prefect_client-3.7.5.dev4 → prefect_client-3.7.5.dev5}/src/prefect/settings/_types.py

@@ -138,6 +138,7 @@ SettingAccessor: TypeAlias = Literal[
     "server.memoize_block_auto_registration",
     "server.metrics_enabled",
     "server.register_blocks_on_start",
+    "server.services.cancellation_cleanup.cancelling_timeout_seconds",
     "server.services.cancellation_cleanup.enabled",
     "server.services.cancellation_cleanup.loop_seconds",
     "server.services.cleanup_reconciler.batch_size",

{prefect_client-3.7.5.dev4 → prefect_client-3.7.5.dev5}/src/prefect/settings/models/server/services.py

@@ -44,6 +44,12 @@ class ServerServicesCancellationCleanupSettings(ServicesBaseSetting):
         ),
     )
 
+    cancelling_timeout_seconds: float = Field(
+        default=300,
+        gt=0,
+        description="The cancellation cleanup service will enqueue worker cleanup for flow runs that remain in CANCELLING longer than this many seconds. Defaults to `300`.",
+    )
+
 
 _VALID_VACUUM_TYPES = frozenset({"events", "flow_runs"})
 

prefect_client-3.7.5.dev4/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.7.5.dev4"
-__build_date__ = "2026-06-16 09:53:11.473641+00:00"
-__git_commit__ = "d26ca5871053fae88d1f757a6c1e9eab7ae48ac1"
-__dirty__ = False