durable-workflow 0.4.90__tar.gz → 0.4.91__tar.gz

{durable_workflow-0.4.90/src/durable_workflow.egg-info → durable_workflow-0.4.91}/PKG-INFO

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

{durable_workflow-0.4.90 → durable_workflow-0.4.91}/pyproject.toml

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

{durable_workflow-0.4.90 → durable_workflow-0.4.91}/src/durable_workflow/client.py

@@ -19,6 +19,7 @@ methods without repeating the id on every call.
 from __future__ import annotations
 
 import asyncio
+import math
 import os
 import time
 import uuid
@@ -58,6 +59,31 @@ def _default_sdk_version() -> str:
 
 
 DEFAULT_SDK_VERSION = _default_sdk_version()
+_WORKER_POLL_MAX_SERVER_TIMEOUT_SECONDS = 60
+_WORKER_POLL_HTTP_TIMEOUT_GRACE_SECONDS = 5.0
+
+
+def _worker_poll_timeout_seconds(timeout: float | None) -> int | None:
+    if timeout is None:
+        return None
+
+    value = float(timeout)
+    if not math.isfinite(value) or value < 0:
+        raise ValueError("timeout must be a non-negative finite number")
+
+    seconds = int(math.ceil(value)) if value >= 1 else 0
+    return min(seconds, _WORKER_POLL_MAX_SERVER_TIMEOUT_SECONDS)
+
+
+def _worker_poll_http_timeout(timeout: float | None) -> float | None:
+    timeout_seconds = _worker_poll_timeout_seconds(timeout)
+    if timeout_seconds is None:
+        return None
+
+    if timeout_seconds == 0:
+        return max(float(timeout), 1.0)
+
+    return max(float(timeout), float(timeout_seconds) + _WORKER_POLL_HTTP_TIMEOUT_GRACE_SECONDS)
 
 
 def _protocol_version_from_env(name: str, default: str) -> str:
@@ -3217,22 +3243,31 @@ class Client:
         Returns the full worker-protocol response envelope, including
         ``poll_status`` when the server has no task to lease. Worker-plane
         endpoint — most applications use :class:`~durable_workflow.Worker`
-        rather than calling this directly.
+        rather than calling this directly. ``timeout`` controls the server
+        long-poll window; the HTTP request uses a small grace margin.
         """
         body: dict[str, Any] = {
             "worker_id": worker_id,
             "task_queue": task_queue,
             "poll_request_id": poll_request_id or f"wf-poll-{uuid.uuid4().hex}",
         }
+        timeout_seconds = _worker_poll_timeout_seconds(timeout)
+        if timeout_seconds is not None:
+            body["timeout_seconds"] = timeout_seconds
         if build_id:
             body["build_id"] = build_id
         if history_page_size is not None:
             body["history_page_size"] = history_page_size
+        http_timeout = _worker_poll_http_timeout(timeout)
 
         for _ in range(2):
             try:
                 data = await self._request(
-                    "POST", "/worker/workflow-tasks/poll", worker=True, json=body, timeout=timeout
+                    "POST",
+                    "/worker/workflow-tasks/poll",
+                    worker=True,
+                    json=body,
+                    timeout=http_timeout,
                 )
             except httpx.TimeoutException:
                 continue
@@ -3255,7 +3290,8 @@ class Client:
 
         Returns the task payload, or ``None`` on poll timeout. Worker-plane
         endpoint — most applications use :class:`~durable_workflow.Worker`
-        rather than calling this directly.
+        rather than calling this directly. ``timeout`` controls the server
+        long-poll window; the HTTP request uses a small grace margin.
         """
         data = await self.poll_workflow_task_response(
             worker_id=worker_id,
@@ -3363,19 +3399,29 @@ class Client:
 
         Query tasks are ephemeral worker-plane requests created when the server
         must route a control-plane query to a non-PHP workflow runtime.
+        ``timeout`` controls the server long-poll window; the HTTP request
+        uses a small grace margin.
         """
         body: dict[str, Any] = {
             "worker_id": worker_id,
             "task_queue": task_queue,
             "poll_request_id": poll_request_id or f"query-poll-{uuid.uuid4().hex}",
         }
+        timeout_seconds = _worker_poll_timeout_seconds(timeout)
+        if timeout_seconds is not None:
+            body["timeout_seconds"] = timeout_seconds
         if build_id:
             body["build_id"] = build_id
+        http_timeout = _worker_poll_http_timeout(timeout)
 
         for _ in range(2):
             try:
                 data = await self._request(
-                    "POST", "/worker/query-tasks/poll", worker=True, json=body, timeout=timeout
+                    "POST",
+                    "/worker/query-tasks/poll",
+                    worker=True,
+                    json=body,
+                    timeout=http_timeout,
                 )
             except httpx.TimeoutException:
                 continue
@@ -3457,19 +3503,29 @@ class Client:
 
         Returns the task payload, or ``None`` on poll timeout. Worker-plane
         endpoint — typically used by :class:`~durable_workflow.Worker`.
+        ``timeout`` controls the server long-poll window; the HTTP request
+        uses a small grace margin.
         """
         body: dict[str, Any] = {
             "worker_id": worker_id,
             "task_queue": task_queue,
             "poll_request_id": poll_request_id or f"activity-poll-{uuid.uuid4().hex}",
         }
+        timeout_seconds = _worker_poll_timeout_seconds(timeout)
+        if timeout_seconds is not None:
+            body["timeout_seconds"] = timeout_seconds
         if build_id:
             body["build_id"] = build_id
+        http_timeout = _worker_poll_http_timeout(timeout)
 
         for _ in range(2):
             try:
                 data = await self._request(
-                    "POST", "/worker/activity-tasks/poll", worker=True, json=body, timeout=timeout
+                    "POST",
+                    "/worker/activity-tasks/poll",
+                    worker=True,
+                    json=body,
+                    timeout=http_timeout,
                 )
             except httpx.TimeoutException:
                 continue

{durable_workflow-0.4.90 → durable_workflow-0.4.91/src/durable_workflow.egg-info}/PKG-INFO

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

{durable_workflow-0.4.90 → durable_workflow-0.4.91}/tests/test_client.py

@@ -2292,8 +2292,9 @@ class TestFailWorkflowTask:
         assert request_body["task_queue"] == "queue-1"
         assert request_body["build_id"] == "build-v2"
         assert request_body["history_page_size"] == 100
+        assert request_body["timeout_seconds"] == 3
         assert isinstance(request_body["poll_request_id"], str)
-        assert mock.await_args.kwargs["timeout"] == 3.0
+        assert mock.await_args.kwargs["timeout"] == 8.0
 
     @pytest.mark.asyncio
     async def test_poll_workflow_task_keeps_returning_none_for_no_compatible_status(self, client: Client) -> None:
@@ -2351,6 +2352,7 @@ class TestFailWorkflowTask:
         assert request_body["worker_id"] == "worker-1"
         assert request_body["task_queue"] == "queue-1"
         assert request_body["build_id"] == "build-1"
+        assert request_body["timeout_seconds"] == 35
         assert isinstance(request_body["poll_request_id"], str)
         assert request_body["poll_request_id"] != ""
 
@@ -2370,6 +2372,7 @@ class TestFailWorkflowTask:
         assert request_body["worker_id"] == "worker-1"
         assert request_body["task_queue"] == "queue-1"
         assert request_body["build_id"] == "build-1"
+        assert request_body["timeout_seconds"] == 35
         assert isinstance(request_body["poll_request_id"], str)
         assert request_body["poll_request_id"] != ""
 
@@ -3016,6 +3019,8 @@ class TestQueryTasks:
             body = mock.call_args.kwargs["json"]
             assert body["worker_id"] == "w1"
             assert body["task_queue"] == "q1"
+            assert body["timeout_seconds"] == 5
+            assert mock.call_args.kwargs["timeout"] == 10.0
             assert isinstance(body["poll_request_id"], str)
             assert body["poll_request_id"] != ""