put listener to last

Signed-off-by: Yaliang Wu <[email protected]>
ylwu-amzn · Feb 13, 2024 · 5d5f261 · 5d5f261
1 parent 3fc5b30
commit 5d5f261
Showing 1 changed file with 1 addition and 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
@@ -309,14 +309,14 @@ 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.",
                             "}"
                         );
                         newPrompt.set(substitutor.replace(finalPrompt));
                         tmpParameters.put(PROMPT, newPrompt.get());
+                        ((ActionListener<Object>) nextStepListener).onResponse(res);
                     }
                 } else {
                     addToolOutputToAddtionalInfo(toolSpecMap, lastAction, additionalInfo, output);