Bump pydocket version and remove background worker shutdown timeout (#19969)

Author: desertaxle

pyproject.toml

@@ -80,7 +80,7 @@ dependencies = [
     "whenever>=0.7.3,<0.10.0; python_version>='3.13'",
     "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"

src/prefect/server/api/background_workers.py

@@ -47,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"
@@ -58,24 +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:
-                logger.debug(
-                    "Waiting for background worker to finish after cancellation..."
-                )
-                await asyncio.wait_for(worker_task, timeout=5.0)
-                logger.debug(
-                    "Background worker finished successfully after cancellation"
-                )
-            except asyncio.TimeoutError:
-                logger.debug(
-                    "Background worker did not finish within 5 seconds after cancellation. Proceeding with shutdown"
-                )
-            except asyncio.CancelledError:
-                pass
+        finally:
+            if worker_task:
+                worker_task.cancel()
+                try:
+                    await worker_task
+                except asyncio.CancelledError:
+                    pass