diff --git a/core/src/main/java/com/google/adk/agents/BaseAgent.java b/core/src/main/java/com/google/adk/agents/BaseAgent.java index ed6631c50..95fe838cc 100644 --- a/core/src/main/java/com/google/adk/agents/BaseAgent.java +++ b/core/src/main/java/com/google/adk/agents/BaseAgent.java @@ -529,7 +529,8 @@ public B beforeAgentCallback(BeforeAgentCallback beforeAgentCallback) { @CanIgnoreReturnValue public B beforeAgentCallback(List beforeAgentCallback) { - this.beforeAgentCallback = CallbackUtil.getBeforeAgentCallbacks(beforeAgentCallback); + this.beforeAgentCallback = + ImmutableList.copyOf(CallbackUtil.getBeforeAgentCallbacks(beforeAgentCallback)); return self(); } @@ -541,7 +542,8 @@ public B afterAgentCallback(AfterAgentCallback afterAgentCallback) { @CanIgnoreReturnValue public B afterAgentCallback(List afterAgentCallback) { - this.afterAgentCallback = CallbackUtil.getAfterAgentCallbacks(afterAgentCallback); + this.afterAgentCallback = + ImmutableList.copyOf(CallbackUtil.getAfterAgentCallbacks(afterAgentCallback)); return self(); } diff --git a/core/src/main/java/com/google/adk/agents/CallbackUtil.java b/core/src/main/java/com/google/adk/agents/CallbackUtil.java index 11740ae9c..4eb8704b6 100644 --- a/core/src/main/java/com/google/adk/agents/CallbackUtil.java +++ b/core/src/main/java/com/google/adk/agents/CallbackUtil.java @@ -42,7 +42,7 @@ public final class CallbackUtil { * @return normalized async callbacks, or empty list if input is null. */ @CanIgnoreReturnValue - public static ImmutableList getBeforeAgentCallbacks( + public static List getBeforeAgentCallbacks( List beforeAgentCallbacks) { return getCallbacks( beforeAgentCallbacks, @@ -59,7 +59,7 @@ public static ImmutableList getBeforeAgentCallbacks( * @return normalized async callbacks, or empty list if input is null. */ @CanIgnoreReturnValue - public static ImmutableList getAfterAgentCallbacks( + public static List getAfterAgentCallbacks( List afterAgentCallback) { return getCallbacks( afterAgentCallback,