From f1369e43cc585f0003de0696a8a3a2dab2f207ea Mon Sep 17 00:00:00 2001
From: Google Team Member <genai-sdk-bot@google.com>
Date: Fri, 18 Jul 2025 00:04:41 -0700
Subject: [PATCH] fix: Callbacks registered to run after agent processing
 should run independent of 'end_invocation' value

PiperOrigin-RevId: 784467112
---
 src/google/adk/agents/base_agent.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/src/google/adk/agents/base_agent.py b/src/google/adk/agents/base_agent.py
index c90cc86a0..db6e900d0 100644
--- a/src/google/adk/agents/base_agent.py
+++ b/src/google/adk/agents/base_agent.py
@@ -208,9 +208,6 @@ async def run_async(
       async for event in self._run_async_impl(ctx):
         yield event
 
-      if ctx.end_invocation:
-        return
-
       if event := await self.__handle_after_agent_callback(ctx):
         yield event