Runner drops callbackContextData written in onUserMessageCallback when it resumes an invocation

Ouverte Adaptée aux débutants
  • Quelque chose ne va pas ?
#1,695 1 commentaire 0 réactions 1 personne assignée Voir sur GitHub

Les mainteneurs répondent en général sous 1 jour

@hirematha y travaille déjà.

Depuis le 26/10/2026.

  • #1696 par @svetanis — ouverte

Évaluation

Difficulté
2/5
Temps estimé
1-3 heures
Accessibilité débutants
78/100
Type d'issue
Bug
Clarté
Clairement spécifiée
Activité
Active
Stack technique
java
Domaine
backend

Piste de recherche

Lisez runResumedAgent, resumeCore et runAgentWithUpdatedSession dans l’implémentation de Runner. Suivez la manière dont initialContext est construit et transmis aux callbacks, puis comparez le parcours de reprise au parcours d’une nouvelle invocation. Reproduisez le problème avec la reprise activée et un plugin qui stocke des données dans onUserMessageCallback ; le travail sera terminé lorsque les callbacks de l’agent repris verront ces données et qu’un test de régression couvrira le comportement de reprise existant.

Rédigé par le modèle d'indexation à partir du texte de l'issue.

Description

needs review

Please make sure you read the contribution guide and file the issues in the right place.
Contribution guide.

🔴 Required Information

Describe the Bug

InvocationContext.callbackContextData() is "a map for storing temporary context data that can be shared
between different parts of the invocation", and InvocationContext.Builder passes the same instance to every
copy "so that Plugins can access the same data … across all types of callbacks."

A new invocation keeps that promise. runNewInvocation runs onUserMessageCallback on initialContext, and
runAgentWithUpdatedSession builds the agent's context with initialContext.toBuilder(), so the map is shared.

A resumed invocation does not. resumeCore runs onUserMessageCallback on its own initialContext, but
runResumedAgent is not given that context and builds the agent's from a second builder:

// runResumedAgent
InvocationContext rootContext =
    newInvocationContextBuilder(session, rootAgent)
        .invocationId(resolvedInvocationId)
        .runConfig(runConfig)
        .userContent(userContentFor(session, resolvedInvocationId, userContent))
        .build();

newInvocationContextBuilder starts a new, empty callbackContextData, so whatever a plugin stored in
onUserMessageCallback is missing from beforeRunCallback and from every callback that runs with the agent.
The same builder also resets endInvocation: a plugin that ends the invocation in onUserMessageCallback stops
a new invocation, but a resumed one still calls the model. On every resume, onRunErrorCallback also runs on
initialContext, so it does not see what beforeRunCallback stored, even when no new message is sent.

Trigger: resumability enabled and an invocation paused (for example on a long-running tool) that is then
resumed, with or without a new message. New invocations are unaffected.

Steps to Reproduce
  1. A Runner with ResumabilityConfig.builder().resumable(true), an LlmAgent with a long-running tool, and a
    plugin whose onUserMessageCallback puts a value into invocationContext.callbackContextData() and whose
    beforeAgentCallback reads it back.
  2. Send a message. The model calls the tool and the invocation pauses.
  3. Send the tool's function response, which resumes the invocation.
Expected Behavior

On the resumed turn, beforeAgentCallback reads the value that turn's onUserMessageCallback stored, as it
does on the first turn.

Observed Behavior

The first turn reads its value, and the resumed turn reads null (the plugin stores message-1, then
message-2). beforeAgentCallback reads, per turn:

[agent:message-1, agent:null]

The same happens when the resume is routed to a sub-agent instead of the root.

Environment Details
  • ADK Library Version: main @ 84bae34a (pom 1.13.0)
  • OS: not OS-specific
  • TS Version: N/A (Java)
Model Information

N/A

🟡 Optional Information

Regression

No. Present since resuming an invocation was added (v1.11.0).

Willingness to contribute

Yes, a PR follows. runResumedAgent takes initialContext and builds the agent's context with
initialContext.toBuilder(), as runAgentWithUpdatedSession does:

InvocationContext rootContext =
    initialContext.toBuilder()
        .agent(rootAgent)
        .userContent(...)
        .build();
How often has this issue occurred?

Always (100%), on a resumed invocation.

Langage dominant
Java
Étoiles
1.8k
Forks
442
Merge moyen
3 j 13 h
PR mergées (30 j)
95

Préparer son environnement

Ouvrir dans Codespaces

Lance le conteneur de développement du projet dans votre navigateur, avec votre propre compte GitHub.

Par où commencer

  1. Lisez l'issue en entier, puis le guide de contribution du projet.
  2. Signalez en commentaire que vous la prenez — cela évite que deux personnes fassent le même travail.
  3. Forkez le dépôt et travaillez sur une branche.
  4. Ouvrez une pull request qui référence le numéro de l'issue.

Autres issues de google/adk-java

Toutes les issues de google/adk-java

Issues similaires

Plus d'issues Java

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.