From 06873011465c04709c5c922e50cd57364607faee Mon Sep 17 00:00:00 2001 From: Rob Clevenger Date: Wed, 2 Jul 2025 12:27:32 -0700 Subject: [PATCH] fix: Remove end_invocation check to ensure after_agent callbacks execute Removes the end_invocation check that was preventing after_agent callbacks from running. This ensures all registered callbacks are executed properly regardless of the invocation status. Fixes discussion: https://github.com/google/adk-python/discussions/1689 --- src/google/adk/agents/base_agent.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/google/adk/agents/base_agent.py b/src/google/adk/agents/base_agent.py index bdc10ac3a..a4946a0eb 100644 --- a/src/google/adk/agents/base_agent.py +++ b/src/google/adk/agents/base_agent.py @@ -147,8 +147,7 @@ async def run_async( async for event in self._run_async_impl(ctx): yield event - if ctx.end_invocation: - return + # Don't check for end_invocation. Let the afer agent callbacks get called. if event := await self.__handle_after_agent_callback(ctx): yield event