diff --git a/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ReActAgentTest.java b/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ReActAgentTest.java index 7bff4550b..38eb800bc 100644 --- a/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ReActAgentTest.java +++ b/e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ReActAgentTest.java @@ -215,9 +215,14 @@ public void testReActAgentNoOutputSchema() throws Exception { .apply(agent) .toDataStream(); - out.print(); + CloseableIterator results = out.collectAsync(); env.execute(); + + Assertions.assertTrue(results.hasNext(), "The agent should emit a result."); + Assertions.assertTrue( + results.next() instanceof String, + "The result should be a String when no output schema is provided."); } // create ReAct agent; pass false to skip the output schema.