NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-14 01:41:25 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source 
PiperOrigin-RevId: 755541004
This commit is contained in:
Shangjie Chen 2025-05-06 15:09:09 -07:00 committed by Copybara-Service
parent 1393965720
commit f96cdc675c
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/google/adk/tools/agent_tool.py
View File

@ -146,11 +146,12 @@ class AgentTool(BaseTool):
if runner.artifact_service: if runner.artifact_service:
# Forward all artifacts to parent session. # 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, app_name=session.app_name,
user_id=session.user_id, user_id=session.user_id,
session_id=session.id, session_id=session.id,
): )
for artifact_name in artifact_names:
if artifact := await runner.artifact_service.load_artifact( if artifact := await runner.artifact_service.load_artifact(
app_name=session.app_name, app_name=session.app_name,
user_id=session.user_id, user_id=session.user_id,