[WORKFLOW SDK FEATURE REQUEST] Retry WaitForInstanceCompletion/Start on a server-sent CANCELLED
Maintainers usually reply within 2 days
Nobody has claimed this yet.
Assessment
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Newbie friendliness
- 55/100
- Issue type
- Feature
- Clarity
- Mostly clear
- Activity status
- Active
- Tech stack
- python
- Domain
- backend, distributed-systems
Research direction
Start in dapr/ext/workflow/_durabletask/client.py around lines 335-349 and the matching retry logic in aio/client.py. Read the three cancellation tests in tests/ext/workflow/durabletask/test_orchestration_wait.py and test_client_async.py, then confirm the intended retry and timeout behavior with maintainers. Done means both clients re-issue eligible waits after server CANCELLED responses while preserving caller deadlines and the updated tests pass.
Written by the indexing model from the issue text.
Description
Proposal
wait_for_workflow_completion and wait_for_workflow_start (sync and aio) should re-issue WaitForInstanceCompletion / WaitForInstanceStart when the server ends the call with CANCELLED and the caller's own timeout hasn't expired. Today that status reaches the caller as an error, even though the workflow is still running. This asks maintainers to decide on the behaviour first, because it reverses assertions added in #1112. It's a proposal, not a PR.
Why the server sends CANCELLED
When daprd shuts down or restarts mid-wait (rollout, pod restart, hot-reload, a config change that restarts the runtime), the actors router cancels every in-flight call it tracks. The wait then returns CANCELLED "context canceled", even though the caller is still waiting. The mechanism and a repro are in dapr/dapr#10566.
Reproduced against native daprd 1.18.0:
- a 60 s workflow with
wait_for_workflow_completion(id)and no timeout returns normally; - the same run with daprd sent
SIGTERMabout 10 s in fails withStatusCode.CANCELLED"context canceled".
Why retrying is safe
- A blocking unary call only gets
CANCELLEDfrom the server. The sync client can't cancel it mid-flight, and in the aio client a caller's cancellation raisesasyncio.CancelledError, notAioRpcError. A client-side deadline shows up asDEADLINE_EXCEEDED, which already maps toTimeoutError. - The wait only reads state, so re-issuing it is idempotent. It returns straight away if the workflow already finished.
What would change
- Add
CANCELLEDto the retried codes, when the caller's timeout hasn't expired, in_durabletask/client.pyL335 and_durabletask/aio/client.pyL230._is_deadline_cancellation(L349) already turns aCANCELLEDafter the deadline intoTimeoutError. - Restart the 30 s transient-retry window after each successful re-issue, rather than anchoring it on the first transient error. Otherwise a sidecar that restarts more than once during a long wait would still fail it.
- These tests from #1112 currently assert that a
CANCELLEDreaches the caller, and would flip:
Relationship to the runtime fix
The proper fix is in daprd: return UNAVAILABLE in this case (dapr/dapr#10566). The SDK already retries UNAVAILABLE. This change would protect users on current and older runtimes until that ships, and it would still help afterwards with proxies that reset the stream.
Question for maintainers
Is it acceptable to treat a server-sent CANCELLED on these two waits as retryable, and change the #1112 tests? If yes, it's a small change in both clients plus tests.
- Dominant language
- Python
- Stars
- 272
- Forks
- 152
- Avg merge
- 2d 22h
- Merged PRs (30d)
- 7
Getting set up
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
More from dapr/python-sdk
-
dapr-ext-workflow good first issue kind/enhancement P2
Difficulty 2/5 1-2 days Newbie friendliness 72/100
dapr/python-sdk#853 · 4 comments ·
Maintainers usually reply within 2 days
-
kind/bug
Difficulty 4/5 3-5 days Newbie friendliness 65/100
dapr/python-sdk#1233 ·
Maintainers usually reply within 2 days
-
kind/bug
Difficulty 4/5 3-5 days Newbie friendliness 68/100
dapr/python-sdk#1232 ·
Maintainers usually reply within 2 days
-
kind/bug
Difficulty 4/5 3-5 days Newbie friendliness 45/100
dapr/python-sdk#1230 ·
Maintainers usually reply within 2 days
-
dapr-ext-workflow kind/enhancement
Difficulty 3/5 1-2 days Newbie friendliness 78/100
dapr/python-sdk#1213 ·
Maintainers usually reply within 2 days
Similar issues
-
Difficulty 2/5 1-3 hours Newbie friendliness 88/100
Maintainers usually reply within 1 day
-
Difficulty 2/5 1-3 hours Newbie friendliness 84/100
spec-kitty/spec-kitty#5319 ·
Maintainers usually reply within 1 day
-
backend::vllm diffusion multimodal
Difficulty 2/5 1-3 hours Newbie friendliness 72/100
Maintainers usually reply within 1 day
-
Difficulty 2/5 1-3 hours Newbie friendliness 84/100
openai/openai-agents-python#5229 ·
Maintainers usually reply within 1 day
-
Difficulty 2/5 1-3 hours Newbie friendliness 84/100
Maintainers usually reply within 1 day