Properly track state update in callbacks.

google-genai-bot · copybara-github · commit 86656374430e · 2025-06-17T13:17:38.000-07:00
PiperOrigin-RevId: 772593312
diff --git a/core/src/main/java/com/google/adk/agents/BaseAgent.java b/core/src/main/java/com/google/adk/agents/BaseAgent.java
@@ -266,9 +266,7 @@ private Single<Optional<Event>> callCallback(
         .switchIfEmpty(
             Single.defer(
                 () -> {
-                  boolean hasStateDelta = !callbackContext.eventActions().stateDelta().isEmpty();
-
-                  if (hasStateDelta) {
+                  if (callbackContext.state().hasDelta()) {
                     Event.Builder eventBuilder =
                         Event.builder()
                             .id(Event.generateEventId())
diff --git a/core/src/main/java/com/google/adk/sessions/State.java b/core/src/main/java/com/google/adk/sessions/State.java
@@ -154,4 +154,8 @@ public int size() {
   public Collection<Object> values() {
     return state.values();
   }
+
+  public boolean hasDelta() {
+    return !delta.isEmpty();
+  }
 }

Original file line number	Diff line number	Diff line change
`@@ -154,4 +154,8 @@ public int size() {`
`154`	`154`	`public Collection<Object> values() {`
`155`	`155`	`return state.values();`
`156`	`156`	`}`
	`157`	`+`
	`158`	`+ public boolean hasDelta() {`
	`159`	`+ return !delta.isEmpty();`
	`160`	`+ }`
`157`	`161`	`}`