durable-workflow 0.4.87__tar.gz → 0.4.89__tar.gz

{durable_workflow-0.4.87/src/durable_workflow.egg-info → durable_workflow-0.4.89}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durable-workflow
-Version: 0.4.87
+Version: 0.4.89
 Summary: Python SDK for the Durable Workflow server (language-neutral HTTP protocol)
 Author: Durable Workflow Contributors
 License-Expression: MIT

{durable_workflow-0.4.87 → durable_workflow-0.4.89}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "durable-workflow"
-version = "0.4.87"
+version = "0.4.89"
 description = "Python SDK for the Durable Workflow server (language-neutral HTTP protocol)"
 readme = "README.md"
 requires-python = ">=3.10"

{durable_workflow-0.4.87 → durable_workflow-0.4.89}/src/durable_workflow/client.py

@@ -3242,7 +3242,14 @@ class Client:
         return {"task": None, "poll_status": "timeout"}
 
     async def poll_workflow_task(
-        self, *, worker_id: str, task_queue: str, timeout: float = 35.0
+        self,
+        *,
+        worker_id: str,
+        task_queue: str,
+        timeout: float = 35.0,
+        build_id: str | None = None,
+        history_page_size: int | None = None,
+        poll_request_id: str | None = None,
     ) -> Any:
         """Long-poll for the next workflow task on ``task_queue``.
 
@@ -3254,6 +3261,9 @@ class Client:
             worker_id=worker_id,
             task_queue=task_queue,
             timeout=timeout,
+            build_id=build_id,
+            history_page_size=history_page_size,
+            poll_request_id=poll_request_id,
         )
 
         return data.get("task")
@@ -3341,7 +3351,13 @@ class Client:
         )
 
     async def poll_query_task(
-        self, *, worker_id: str, task_queue: str, timeout: float = 35.0
+        self,
+        *,
+        worker_id: str,
+        task_queue: str,
+        timeout: float = 35.0,
+        build_id: str | None = None,
+        poll_request_id: str | None = None,
     ) -> Any:
         """Long-poll for the next workflow query task on ``task_queue``.
 
@@ -3351,8 +3367,11 @@ class Client:
         body: dict[str, Any] = {
             "worker_id": worker_id,
             "task_queue": task_queue,
-            "poll_request_id": f"query-poll-{uuid.uuid4().hex}",
+            "poll_request_id": poll_request_id or f"query-poll-{uuid.uuid4().hex}",
         }
+        if build_id:
+            body["build_id"] = build_id
+
         for _ in range(2):
             try:
                 data = await self._request(
@@ -3426,7 +3445,13 @@ class Client:
         )
 
     async def poll_activity_task(
-        self, *, worker_id: str, task_queue: str, timeout: float = 35.0
+        self,
+        *,
+        worker_id: str,
+        task_queue: str,
+        timeout: float = 35.0,
+        build_id: str | None = None,
+        poll_request_id: str | None = None,
     ) -> Any:
         """Long-poll for the next activity task on ``task_queue``.
 
@@ -3436,8 +3461,11 @@ class Client:
         body: dict[str, Any] = {
             "worker_id": worker_id,
             "task_queue": task_queue,
-            "poll_request_id": f"activity-poll-{uuid.uuid4().hex}",
+            "poll_request_id": poll_request_id or f"activity-poll-{uuid.uuid4().hex}",
         }
+        if build_id:
+            body["build_id"] = build_id
+
         for _ in range(2):
             try:
                 data = await self._request(

{durable_workflow-0.4.87 → durable_workflow-0.4.89}/src/durable_workflow/worker.py

@@ -1588,6 +1588,7 @@ class Worker:
                     worker_id=self.worker_id,
                     task_queue=self.task_queue,
                     timeout=self._poll_http_timeout,
+                    build_id=self.build_id,
                 )
             except Exception as e:
                 self._wf_semaphore.release()
@@ -1665,6 +1666,7 @@ class Worker:
                     worker_id=self.worker_id,
                     task_queue=self.task_queue,
                     timeout=self._poll_http_timeout,
+                    build_id=self.build_id,
                 )
             except Exception as e:
                 self._act_semaphore.release()
@@ -1702,6 +1704,7 @@ class Worker:
                     worker_id=self.worker_id,
                     task_queue=self.task_queue,
                     timeout=self._poll_http_timeout,
+                    build_id=self.build_id,
                 )
             except Exception as e:
                 self._record_poll_metrics("query", "error", time.perf_counter() - poll_start)
@@ -1960,6 +1963,7 @@ class Worker:
                         worker_id=self.worker_id,
                         task_queue=self.task_queue,
                         timeout=self._poll_http_timeout,
+                        build_id=self.build_id,
                     )
                     self._record_poll_metrics(
                         "workflow",
@@ -1990,6 +1994,7 @@ class Worker:
                     worker_id=self.worker_id,
                     task_queue=self.task_queue,
                     timeout=self._poll_http_timeout,
+                    build_id=self.build_id,
                 )
                 self._record_poll_metrics(
                     "activity",

{durable_workflow-0.4.87 → durable_workflow-0.4.89/src/durable_workflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durable-workflow
-Version: 0.4.87
+Version: 0.4.89
 Summary: Python SDK for the Durable Workflow server (language-neutral HTTP protocol)
 Author: Durable Workflow Contributors
 License-Expression: MIT

{durable_workflow-0.4.87 → durable_workflow-0.4.89}/tests/test_client.py

@@ -2302,10 +2302,15 @@ class TestFailWorkflowTask:
             "poll_status": "no_compatible_worker",
         }
 
-        with patch.object(client, "poll_workflow_task_response", new_callable=AsyncMock, return_value=response):
-            task = await client.poll_workflow_task(worker_id="worker-v2", task_queue="queue-1")
+        with patch.object(client, "poll_workflow_task_response", new_callable=AsyncMock, return_value=response) as mock:
+            task = await client.poll_workflow_task(
+                worker_id="worker-v2",
+                task_queue="queue-1",
+                build_id="build-v2",
+            )
 
         assert task is None
+        assert mock.await_args.kwargs["build_id"] == "build-v2"
 
     @pytest.mark.asyncio
     async def test_poll_workflow_task_retries_once_with_same_poll_request_id_after_timeout(
@@ -2335,12 +2340,36 @@ class TestFailWorkflowTask:
         response_task = {"task": {"task_id": "activity-task-123"}}
 
         with patch.object(client, "_request", new_callable=AsyncMock, return_value=response_task) as mock:
-            task = await client.poll_activity_task(worker_id="worker-1", task_queue="queue-1")
+            task = await client.poll_activity_task(
+                worker_id="worker-1",
+                task_queue="queue-1",
+                build_id="build-1",
+            )
+
+        assert task == response_task["task"]
+        request_body = mock.await_args.kwargs["json"]
+        assert request_body["worker_id"] == "worker-1"
+        assert request_body["task_queue"] == "queue-1"
+        assert request_body["build_id"] == "build-1"
+        assert isinstance(request_body["poll_request_id"], str)
+        assert request_body["poll_request_id"] != ""
+
+    @pytest.mark.asyncio
+    async def test_poll_query_task_sends_build_id(self, client: Client) -> None:
+        response_task = {"task": {"query_task_id": "query-task-123"}}
+
+        with patch.object(client, "_request", new_callable=AsyncMock, return_value=response_task) as mock:
+            task = await client.poll_query_task(
+                worker_id="worker-1",
+                task_queue="queue-1",
+                build_id="build-1",
+            )
 
         assert task == response_task["task"]
         request_body = mock.await_args.kwargs["json"]
         assert request_body["worker_id"] == "worker-1"
         assert request_body["task_queue"] == "queue-1"
+        assert request_body["build_id"] == "build-1"
         assert isinstance(request_body["poll_request_id"], str)
         assert request_body["poll_request_id"] != ""
 

{durable_workflow-0.4.87 → durable_workflow-0.4.89}/tests/test_release_docs_audit_workflow.py

@@ -26,9 +26,11 @@ def test_publish_workflow_runs_docs_audit_in_recoverable_job() -> None:
         "DOCS_RELEASE_AUDIT_ARTIFACT: sdk-python",
         "DOCS_RELEASE_AUDIT_VERSION: ${{ github.ref_name }}",
         "DOCS_RELEASE_AUDIT_EVIDENCE: docs-release-audit-evidence.json",
+        "DOCS_RELEASE_AUDIT_HANDOFF: docs-release-audit-handoff.json",
         "scripts/ci/check-docs-release-audit.sh",
         "Upload docs release audit evidence",
         "docs-release-audit-evidence.json",
+        "docs-release-audit-handoff.json",
     ]:
         assert expected in workflow
 
@@ -39,10 +41,14 @@ def test_publish_workflow_runs_docs_audit_in_recoverable_job() -> None:
     assert "Upload docs release audit evidence" not in publish_job
     assert "needs: publish" in docs_audit_job
     assert "durable-workflow.release.docs-release-audit-evidence" in auditor
+    assert "durable-workflow.release.docs-artifact-tuple-handoff" in auditor
+    assert "DOCS_RELEASE_AUDIT_HANDOFF" in auditor
     assert "schema: 'durable-workflow.docs.refresh-request'" in auditor
     assert "repository: 'durable-workflow.github.io'" in auditor
     assert "refresh_command: 'npm run refresh:public-artifact-versions'" in auditor
+    assert "refresh_files: refreshFiles" in auditor
     assert "observed_artifact_versions: versions" in auditor
+    assert "docs_artifact_tuple_handoff: handoff" in auditor
 
     publish_offset = workflow.index("Publish to PyPI")
     docs_audit_job_offset = workflow.index("verify-docs-release-audit:")

{durable_workflow-0.4.87 → durable_workflow-0.4.89}/tests/test_worker.py

@@ -403,6 +403,59 @@ class TestWorkerRegistration:
         call_kwargs = mock_client.register_worker.call_args.kwargs
         assert call_kwargs["build_id"] == "release-2026.04.22-a1"
 
+    @pytest.mark.asyncio
+    async def test_poll_loops_forward_build_id_when_configured(
+        self, mock_client: AsyncMock
+    ) -> None:
+        workflow_called = asyncio.Event()
+        activity_called = asyncio.Event()
+        worker = Worker(
+            mock_client,
+            task_queue="q1",
+            workflows=[TestWorkflow],
+            activities=[echo_activity],
+            worker_id="w-build-poll",
+            build_id="release-2026.04.22-a1",
+            poll_timeout=0.01,
+        )
+
+        async def workflow_poll_once(**_: object) -> None:
+            workflow_called.set()
+            await worker.stop()
+            return None
+
+        async def activity_poll_once(**_: object) -> None:
+            activity_called.set()
+            await worker.stop()
+            return None
+
+        mock_client.poll_workflow_task.side_effect = workflow_poll_once
+        await worker._poll_workflow_tasks()
+        workflow_kwargs = mock_client.poll_workflow_task.call_args.kwargs
+        assert workflow_called.is_set()
+        assert workflow_kwargs["build_id"] == "release-2026.04.22-a1"
+
+        worker._stop.clear()
+        mock_client.poll_activity_task.side_effect = activity_poll_once
+        await worker._poll_activity_tasks()
+        activity_kwargs = mock_client.poll_activity_task.call_args.kwargs
+        assert activity_called.is_set()
+        assert activity_kwargs["build_id"] == "release-2026.04.22-a1"
+
+        worker._stop.clear()
+        query_called = asyncio.Event()
+
+        async def query_poll_once(**_: object) -> None:
+            query_called.set()
+            await worker.stop()
+            return None
+
+        mock_client.poll_query_task.side_effect = query_poll_once
+        await worker._poll_query_tasks()
+        query_kwargs = mock_client.poll_query_task.call_args.kwargs
+        assert query_called.is_set()
+        assert query_kwargs["build_id"] == "release-2026.04.22-a1"
+
     @pytest.mark.asyncio
     async def test_register_omits_build_id_when_not_configured(
         self, mock_client: AsyncMock