[Feat]: Cluster mode: detect and recover tasks abandoned by a crashed replica (heartbeat/lease)
メンテナーはふだん 2 日以内に返信
@rohityan がすでに取り組んでいます。
2026年10月7日 から。
評価
この issue はまだ評価されていません。
説明
Is your feature request related to a problem? Please describe.
Cluster mode (#1281) handles replicas that are alive and disagree: version checks on saves, cancel from any replica, resubscribe from any replica. It does nothing for a replica that dies while running execute() (OOM kill, node loss, a rolling deploy without enough grace time).
When that happens, the task stays non-terminal forever:
GetTaskreturnsTASK_STATE_WORKINGindefinitely.- Every
SubscribeToTaskon another replica keeps pollingtask_eventsand never ends, because no terminal event will ever be written. - Nothing ever re-runs or fails the task. The only way out is a client-issued
CancelTask. - A push-only client never hears anything again.
The client can't tell a dead replica from a slow agent. status.timestamp only moves when the agent emits an event, so a long LLM or tool call looks exactly like a crash.
Detecting abandoned tasks was a stated goal of the Go SDK's distributed mode (a2a-go discussion #139: "Supporting abandoned task detection (e.g. on process reboot)"), and a2a-go ships it (a2a-go #115, a2asrv/workqueue). The Python design discussion (#1224) doesn't cover it.
Describe the solution you'd like
An optional extension point for cluster mode, similar to a2a-go's workqueue.Heartbeater / LeaseManager, along these lines:
- Lease while executing. While
execute()runs, the replica renews a lease (task_id,owner,expires_at, plus the task's owner scope) on an interval. - Release when the turn ends. Release the lease when the task moves into a terminal or interrupted state, so a task waiting on
INPUT_REQUIRED/AUTH_REQUIREDis never treated as abandoned. - Sweep. On every replica, a sweeper atomically claims leases that have expired (for example
FOR UPDATE SKIP LOCKED, so no leader is needed) and applies a policy:fail(default): writeTASK_STATE_FAILED;re-execute: run the turn again, with an attempt counter.
- Write the failure through
VersionedTaskStore.save(..., event=...). This matters in cluster mode:- the event row ends every remote
SubscribeToTaskstream withFAILED; - the version bump fences a replica that wasn't really dead (a long GC pause or a network partition). Its next save gets
ConcurrentTaskModificationError, it re-reads, seesFAILED, and cancels its producer, the same way a remote cancel works today; - push senders can then notify
FAILED.
- the event row ends every remote
Pitfalls we hit running a heartbeat plus sweeper outside the SDK. They're probably worth designing in from the start:
- Release after the durable save, not when
execute()returns. The terminal or interrupted status event may still be on the event queue whenexecute()returns. A crash in that gap leaves a non-terminal task with no lease, which no sweep can ever find. - Release on a state change, not on whatever state is being saved. On a resumed turn, the first save is the held user-message flush, and it re-saves the stale
INPUT_REQUIREDstate (update_with_messagedoesn't changestatus.state). Releasing on "saved an interrupted state" drops the lease at the start of the resumed turn. - A renewal must not recreate a released lease. If renewal is an upsert, a heartbeat that lands just after the release brings the lease back. The sweeper later claims it and fails a task that was correctly paused.
- The sweeper needs the owner scope.
DatabaseTaskStore/VersionedDatabaseTaskStorefilter reads byowner_resolver(context). A sweeper callingget()with an emptyServerCallContextgetsNonefor authenticated tasks and silently does nothing. Store the owner scope with the lease and rebuild the context from it.
Describe alternatives you've considered
- Handle it in the application (what we do today): heartbeat rows, an orphan sweeper, and a
TaskStorewrapper that releases the lease after the durable save. It works, but every deployment has to rebuild it. In cluster mode it also needs the versioned save path and the event table to end remote streams correctly, and those are SDK internals an application shouldn't have to coordinate with. - Leave it out of scope, as a2a-rs did (a2a-rs#149). That's reasonable for single-process servers. But cluster mode is aimed specifically at multi-replica deployments, and replicas dying is the normal failure in that setting.
- A client-side overall timeout is a useful backstop, but the client can't tell which tasks are actually dead.
- The spec-level "Expectations" proposal (A2A#2209) is about business deadlines, not process liveness.
Additional context
- Defaults that worked for us: heartbeat every 30s, a task counts as abandoned after 120s without one, and the sweep runs every 30s. That's roughly 2–2.5 minutes to detect a dead replica.
- Happy to share more detail or help with an implementation if the maintainers think this belongs in the SDK.
Code of Conduct
- I agree to follow this project's Code of Conduct
- 主要言語
- Python
- スター
- 2.2k
- フォーク
- 509
- 平均マージ
- 3日 18時間
- マージ済み PR(30日)
- 44
環境構築
- Dockerfile・Docker Compose ファイルなし
- プルリクエストのテンプレートあり
- コントリビューションガイドを読む
はじめの一歩
- issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
- 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
- リポジトリをフォークし、ブランチを切って変更します。
- issue 番号を参照したプルリクエストを送ります。
a2aproject/a2a-python のほかの issue
-
[Bug]: REST task/request id sanitization対応中かも @Linux2010 が 103 日前に担当しました。 オープンmaintainers-only
難易度 2/5 1〜3時間 初心者へのやさしさ 65/100
a2aproject/a2a-python#805 · コメント 1 件 ·
メンテナーはふだん 2 日以内に返信
-
[Bug]: Cluster mode: SubscribeToTask on a replica holding a paused ActiveTask returns a stale INPUT_REQUIRED snapshot and closes対応中かも @rohityan が 2 日前に担当しました。 オープンcomponent: server
a2aproject/a2a-python#1323 · 担当者 1 名 ·
メンテナーはふだん 2 日以内に返信
-
[Bug]: After a streamed task is cancelled, its background producer never finishes対応中かも @rohityan が 2 日前に担当しました。 オープンcomponent: server status:awaiting response
a2aproject/a2a-python#1322 · コメント 1 件 · 担当者 1 名 ·
メンテナーはふだん 2 日以内に返信
-
v0.3 gRPC and REST SendMessage without configuration run non-blocking対応中かも @rohityan が 3 日前に担当しました。 オープンquestion status:awaiting response
a2aproject/a2a-python#1321 · コメント 1 件 · 担当者 1 名 ·
メンテナーはふだん 2 日以内に返信
-
[Bug]: Push notification store failure rewrites a completed task as FAILED (DefaultRequestHandlerV2)対応中かも @rohityan が 4 日前に担当しました。 オープンcomponent: server status:awaiting response
a2aproject/a2a-python#1313 · コメント 1 件 · 担当者 1 名 ·
メンテナーはふだん 2 日以内に返信
a2aproject/a2a-python の issue をすべて見る
似ている issue
-
難易度 2/5 1〜3時間 初心者へのやさしさ 82/100
メンテナーはふだん 1 日以内に返信
-
難易度 2/5 1〜3時間 初心者へのやさしさ 62/100
-
難易度 2/5 1〜3時間 初心者へのやさしさ 72/100
メンテナーはふだん 1 日以内に返信
-
[BUG] Multi-day events show "Ended" while still in progress対応中かも @tarunagnihotri534 が今日担当しました。 オープンbug
難易度 2/5 1〜3時間 初心者へのやさしさ 85/100
data-umbrella/du-event-board#231 · コメント 2 件 ·
-
avl_automation: the generated control surface block isn't valid XML (typo in avl_out_parse.py)対応中かも @brksol が今日担当しました。 オープン
難易度 1/5 1時間未満 初心者へのやさしさ 92/100
PX4/PX4-gazebo-models#164 ·