From 8fe439fedcaabbf6421898b0b5ad056bd0ae8501 Mon Sep 17 00:00:00 2001
From: Yaliang Wu <ylwu@amazon.com>
Date: Tue, 13 Feb 2024 11:18:14 -0800
Subject: [PATCH] test

Signed-off-by: Yaliang Wu <ylwu@amazon.com>
---
 .../ml/engine/algorithms/agent/MLChatAgentRunner.java           | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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
index 9479d8d958..efbf8c2f88 100644
--- 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
@@ -309,7 +309,6 @@ private void runReAct(
                         );
                     } else {
                         String res = String.format(Locale.ROOT, "Failed to run the tool %s which is unsupported.", action);
-                        ((ActionListener<Object>) nextStepListener).onResponse(res);
                         StringSubstitutor substitutor = new StringSubstitutor(
                             Map.of(SCRATCHPAD, scratchpadBuilder.toString()),
                             "${parameters.",
@@ -317,6 +316,7 @@ private void runReAct(
                         );
                         newPrompt.set(substitutor.replace(finalPrompt));
                         tmpParameters.put(PROMPT, newPrompt.get());
+                        ((ActionListener<Object>) nextStepListener).onResponse(res);
                     }
                 } else {
                     addToolOutputToAddtionalInfo(toolSpecMap, lastAction, additionalInfo, output);