Feature Request: Run Cancellation / Abort Support for Java ADK
@hemasekhar-p ci sta già lavorando.
Dal 9/7/2026.
Valutazione
Questa issue non è ancora stata valutata.
Descrizione
Summary
Java ADK currently lacks a mechanism to cancel/abort a running agent invocation. TypeScript ADK already ships mature AbortSignal-based cancellation (docs/runtime/cancel.md), and Python ADK has active work in progress (google/adk-python#4825, google/adk-python#5983). Java ADK has no equivalent — no API, no internal checkpoints, and no graceful shutdown path.
Motivation
Cancellation is essential for production agent deployments:
- User-facing "Stop" button — long-running multi-tool loops need immediate, graceful termination when the user clicks stop.
- Timeout enforcement — server-side request deadlines (e.g., HTTP gateway timeouts) must propagate into the agent execution.
- Resource cleanup — without cooperative cancellation, in-flight LLM streaming connections and tool executions leak resources.
- Cost control — aborting early avoids unnecessary LLM token consumption.
Current State
| Language | Status | Mechanism |
|---|---|---|
| TypeScript | ✅ Shipped | runner.runAsync(params, { signal }) with AbortSignal; checkpoints in LLM call, tool execution, and plugin callbacks; graceful generator completion (no throw) |
| Python | 🚧 In progress | google/adk-python#4825 (stop_event: asyncio.Event), google/adk-python#5983 (/cancel API + task.cancel()) |
| Java | ❌ Missing | Only InvocationContext.endInvocation (internal flag, not externally triggerable); Disposable.dispose() is a hard RxJava cut (not graceful, tools don't check it); no RunConfig timeout/cancel field |
Proposed Design (aligned with TypeScript)
API Surface
// Option A: CancellationToken (similar to .NET / structured concurrency)
CancellationTokenSource cts = new CancellationTokenSource();
RunConfig config = RunConfig.builder()
.cancellationToken(cts.token())
.build();
Flowable<Event> events = runner.runAsync(userId, sessionId, content, config);
// Later:
cts.cancel(); // triggers graceful shutdown
// Option B: RxJava-native (Disposable already exists, but needs cooperative checks)
// Enhance internal checkpoints to check subscription disposal state
Internal Checkpoints (minimum)
Following TypeScript's lead, cancellation should be checked at:
- Before each LLM call —
BaseLlmFlow.runOneStep()(already checksendInvocation, easy to extend) - Before each tool execution —
Functions.handleFunctionCalls()(currently has NO cancellation check) - During LLM streaming — between chunks (for long generations)
- Plugin/callback boundaries —
beforeModel/afterModel/beforeTool/afterTool
Semantics (align with TypeScript cancel.md)
- Graceful: the Flowable completes normally (no error signal), already-emitted events are preserved in the session.
- Cooperative: cancellation is checked at defined points; in-flight atomic operations (single LLM call, single tool run) may complete before the next checkpoint.
- Idempotent: calling cancel multiple times is safe.
Questions for Maintainers
- API preference:
CancellationToken(explicit, testable) vs. leveraging RxJavaDisposablesemantics (less new API surface but less portable)? - Scope: Should cancellation also cover
runLive(streaming/live mode), or start withrunAsynconly? - Event emission on cancel: Should the framework emit a final "cancelled" event (like Python's
temp:cancelledstate proposal), or just complete silently (like TypeScript)?
References
- TypeScript ADK cancel docs: https://google.github.io/adk-docs/runtime/cancel/
- Python ADK cancellation PR: google/adk-python#4825
- Python ADK
/cancelAPI PR: google/adk-python#5983 - Python ADK cancellation discussion: google/adk-python#4156
I'm happy to contribute the implementation once the API direction is agreed upon. I have production experience implementing cooperative cancellation in a Java ADK-based agent platform (checkpoint-based, with checks at beforeModel/beforeTool/afterEvent boundaries).
- Lingua principale
- Java
- Stelle
- 1.7k
- Fork
- 421
- Merge medio
- 3g 8h
- PR unite (30g)
- 36
Guida per i contributori
Apri la guida per i contributori
Come iniziare
- Leggi tutta la issue e poi la guida ai contributi del progetto.
- Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
- Fai un fork del repository e lavora su un branch.
- Apri una pull request che faccia riferimento al numero della issue.
Altre issue di google/adk-java
-
needs review
Difficoltà 5/5 Più di una settimana Idoneità per principianti 35/100
-
needs review
-
needs review
-
needs review
-
needs review
Tutte le issue di google/adk-java
Issue simili
-
awaiting triage bug Causes friction Hop Gui P1 P2 Transforms
Difficoltà 2/5 1-3 ore Idoneità per principianti 75/100
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 75/100
apache/flink-agents#1152 ·
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 75/100
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 70/100
jenkinsci/blueocean-plugin#5417 ·
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 75/100
objectionary/eo-graphs#75 ·