fix: Use await to wait for the full list rather than async for which only works for async iterable.

DeanChensj · copybara-github · commit f96cdc675cea · 2025-05-06T15:10:14.000-07:00
PiperOrigin-RevId: 755541004
diff --git a/src/google/adk/tools/agent_tool.py b/src/google/adk/tools/agent_tool.py
@@ -146,11 +146,12 @@ async def run_async(
 
     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,
