Skip to content

Commit 1071f1e

Browse files
google-genai-botcopybara-github
authored andcommitted
fix: remove is a state mutation operation and should also be captured in the delta
PiperOrigin-RevId: 785569259
1 parent 5ede49c commit 1071f1e

File tree

1 file changed

+8
-1
lines changed
  • core/src/main/java/com/google/adk/sessions

1 file changed

+8
-1
lines changed

core/src/main/java/com/google/adk/sessions/State.java

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,12 +119,19 @@ public void putAll(Map<? extends String, ? extends Object> m) {
119119

120120
@Override
121121
public Object remove(Object key) {
122+
if (state.containsKey(key)) {
123+
delta.put((String) key, null);
124+
}
122125
return state.remove(key);
123126
}
124127

125128
@Override
126129
public boolean remove(Object key, Object value) {
127-
return state.remove(key, value);
130+
boolean removed = state.remove(key, value);
131+
if (removed) {
132+
delta.put((String) key, null);
133+
}
134+
return removed;
128135
}
129136

130137
@Override

0 commit comments

Comments
 (0)