durable-workflow 2.1.1__tar.gz → 2.1.2__tar.gz

{durable_workflow-2.1.1/src/durable_workflow.egg-info → durable_workflow-2.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durable-workflow
-Version: 2.1.1
+Version: 2.1.2
 Summary: Python client and worker SDK for Durable Workflow Cloud and self-hosted Server
 Author: Durable Workflow Contributors
 License-Expression: MIT

{durable_workflow-2.1.1 → durable_workflow-2.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "durable-workflow"
-version = "2.1.1"
+version = "2.1.2"
 description = "Python client and worker SDK for Durable Workflow Cloud and self-hosted Server"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -71,8 +71,8 @@ durable-workflow-replay-conformance = "durable_workflow.replay_conformance:main"
 durable-workflow-workflow-updates-conformance = "durable_workflow.workflow_updates_conformance:main"
 
 [tool.durable-workflow]
-product-train = "2.1.1"
-registry-version = "2.1.1"
+product-train = "2.1.2"
+registry-version = "2.1.2"
 supported-server-versions = "2.4.0"
 worker-protocol-version = "1.19"
 control-plane-version = "2"

{durable_workflow-2.1.1 → durable_workflow-2.1.2}/src/durable_workflow/worker.py

@@ -162,6 +162,15 @@ def _is_storage_admission_error(error: BaseException) -> bool:
     )
 
 
+def _poll_error_summary(error: BaseException) -> str:
+    if not isinstance(error, ServerError):
+        return str(error)
+    reason = error.reason()
+    if not isinstance(reason, str) or not reason:
+        return f"server returned {error.status}"
+    return f"server returned {error.status} reason={reason[:64]!r}"
+
+
 def _command_payload_codec(codec: object) -> str:
     if codec is None:
         return serializer.AVRO_CODEC
@@ -2184,7 +2193,7 @@ class Worker:
                 if _is_storage_admission_error(e):
                     raise
                 self._record_poll_metrics("workflow", "error", time.perf_counter() - poll_start)
-                log.warning("workflow poll error: %s", e)
+                log.warning("workflow poll error: %s", _poll_error_summary(e))
                 await asyncio.sleep(1.0)
                 continue
             if task is None:
@@ -2282,7 +2291,7 @@ class Worker:
                 if _is_storage_admission_error(e):
                     raise
                 self._record_poll_metrics("activity", "error", time.perf_counter() - poll_start)
-                log.warning("activity poll error: %s", e)
+                log.warning("activity poll error: %s", _poll_error_summary(e))
                 await asyncio.sleep(1.0)
                 continue
             if task is None:
@@ -2333,7 +2342,7 @@ class Worker:
                 if _is_storage_admission_error(e):
                     raise
                 self._record_poll_metrics("query", "error", time.perf_counter() - poll_start)
-                log.warning("query poll error: %s", e)
+                log.warning("query poll error: %s", _poll_error_summary(e))
                 await asyncio.sleep(1.0)
                 continue
             if task is None:

{durable_workflow-2.1.1 → durable_workflow-2.1.2/src/durable_workflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durable-workflow
-Version: 2.1.1
+Version: 2.1.2
 Summary: Python client and worker SDK for Durable Workflow Cloud and self-hosted Server
 Author: Durable Workflow Contributors
 License-Expression: MIT

{durable_workflow-2.1.1 → durable_workflow-2.1.2}/tests/test_worker.py

@@ -820,6 +820,65 @@ class TestWorkerRegistration:
         assert query_called.is_set()
         assert query_kwargs["build_id"] == "release-2026.04.22-a1"
 
+    @pytest.mark.parametrize(
+        ("poll_method", "loop_method", "task_kind"),
+        [
+            ("poll_workflow_task", "_poll_workflow_tasks", "workflow"),
+            ("poll_activity_task", "_poll_activity_tasks", "activity"),
+            ("poll_query_task", "_poll_query_tasks", "query"),
+        ],
+    )
+    @pytest.mark.asyncio
+    async def test_poll_warning_omits_large_server_response(
+        self,
+        mock_client: AsyncMock,
+        monkeypatch: pytest.MonkeyPatch,
+        caplog: pytest.LogCaptureFixture,
+        poll_method: str,
+        loop_method: str,
+        task_kind: str,
+    ) -> None:
+        worker = Worker(
+            mock_client,
+            task_queue="q1",
+            workflows=[TestWorkflow],
+            activities=[echo_activity],
+        )
+        getattr(mock_client, poll_method).side_effect = ServerError(
+            429,
+            {
+                "reason": "long_poll_capacity_exhausted",
+                "server_capabilities": {"large": "x" * 10_000},
+            },
+        )
+
+        async def stop_after_backoff(seconds: float) -> None:
+            assert seconds == 1.0
+            worker._stop.set()
+
+        monkeypatch.setattr(worker_module.asyncio, "sleep", stop_after_backoff)
+        with caplog.at_level(logging.WARNING, logger="durable_workflow.worker"):
+            await getattr(worker, loop_method)()
+
+        warnings = [
+            record.getMessage()
+            for record in caplog.records
+            if record.name == "durable_workflow.worker"
+        ]
+        assert warnings == [
+            f"{task_kind} poll error: server returned 429 "
+            "reason='long_poll_capacity_exhausted'"
+        ]
+        assert "server_capabilities" not in caplog.text
+
+    def test_poll_error_summary_bounds_reason(self) -> None:
+        summary = worker_module._poll_error_summary(
+            ServerError(429, {"reason": "x" * 10_000})
+        )
+        assert summary == f"server returned 429 reason={'x' * 64!r}"
+        assert worker_module._poll_error_summary(ServerError(503, {})) == "server returned 503"
+        assert worker_module._poll_error_summary(ValueError("transport failed")) == "transport failed"
+
     @pytest.mark.asyncio
     async def test_register_omits_build_id_when_not_configured(
         self, mock_client: AsyncMock