use filtered output in streaming (opensearch-project#4335)

nathaliellenaa · web-flow · commit ec105585c5fd · 2025-10-28T21:03:22.000-07:00
Signed-off-by: Nathalie Jonathan &lt;nathhjo@amazon.com&gt;
diff --git a/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLChatAgentRunner.java b/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLChatAgentRunner.java
@@ -486,7 +486,7 @@ private void runReAct(
                     }
 
                     sessionMsgAnswerBuilder.append(outputToOutputString(filteredOutput));
-                    streamingWrapper.sendToolResponse(outputToOutputString(output), sessionId, parentInteractionId);
+                    streamingWrapper.sendToolResponse(outputToOutputString(filteredOutput), sessionId, parentInteractionId);
                     traceTensors
                         .add(
                             ModelTensors

Original file line number	Diff line number	Diff line change
`@@ -486,7 +486,7 @@ private void runReAct(`
`486`	`486`	`}`
`487`	`487`
`488`	`488`	`sessionMsgAnswerBuilder.append(outputToOutputString(filteredOutput));`
`489`		`- streamingWrapper.sendToolResponse(outputToOutputString(output), sessionId, parentInteractionId);`
	`489`	`+ streamingWrapper.sendToolResponse(outputToOutputString(filteredOutput), sessionId, parentInteractionId);`
`490`	`490`	`traceTensors`
`491`	`491`	`.add(`
`492`	`492`	`ModelTensors`