diff --git a/vertexai/_genai/_evals_common.py b/vertexai/_genai/_evals_common.py index 8388784272..f8850d621f 100644 --- a/vertexai/_genai/_evals_common.py +++ b/vertexai/_genai/_evals_common.py @@ -278,7 +278,8 @@ def agent_run_wrapper( and type(agent_engine).__name__ == "AgentEngine" ): agent_engine_instance = agent_engine - return asyncio.run( + loop = asyncio.get_event_loop() + return loop.run_until_complete( inference_fn_arg( row=row_arg, contents=contents_arg,