You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
Repository navigation
AgentEngine async_stream_query not working as expected. Only returns first few events. #3003
** Please make sure you read the contribution guide and file the issues in the right place. ** Contribution guide.
Describe the bug
I'm trying to invoke the AgentEngine using vertexAI sdk for python. My root agent calls multiple agents in sequence. When I invoke async_stream_query passing the input, it returns only the first 3 events out of 27 events.
To Reproduce
agent = agent_engines.get("projects/PROJECT_ID/locations/LOCATION/reasoningEngines/RESOURCE_ID")
async for response in agent.async_stream_query(
input="case_id"
):
print(response)
Expected behavior
This method should return all events asynchronously
Desktop (please complete the following information):
Activity