From f96cdc675cea2f75f3d9e89203353ead68c70579 Mon Sep 17 00:00:00 2001 From: Shangjie Chen Date: Tue, 6 May 2025 15:09:09 -0700 Subject: [PATCH] fix: Use await to wait for the full list rather than async for which only works for async iterable. PiperOrigin-RevId: 755541004 --- src/google/adk/tools/agent_tool.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/google/adk/tools/agent_tool.py b/src/google/adk/tools/agent_tool.py index aab9ae2..f55906c 100644 --- a/src/google/adk/tools/agent_tool.py +++ b/src/google/adk/tools/agent_tool.py @@ -146,11 +146,12 @@ class AgentTool(BaseTool): if runner.artifact_service: # Forward all artifacts to parent session. - async for artifact_name in runner.artifact_service.list_artifact_keys( + artifact_names = await runner.artifact_service.list_artifact_keys( app_name=session.app_name, user_id=session.user_id, session_id=session.id, - ): + ) + for artifact_name in artifact_names: if artifact := await runner.artifact_service.load_artifact( app_name=session.app_name, user_id=session.user_id,