diff --git a/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLFlowAgentRunner.java b/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLFlowAgentRunner.java
index 38ad680d99..7a7e657f34 100644
--- a/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLFlowAgentRunner.java
+++ b/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/MLFlowAgentRunner.java
@@ -293,6 +293,9 @@ Map<String, String> getToolExecuteParams(MLToolSpec toolSpec, Map<String, String
             StringSubstitutor substitutor = new StringSubstitutor(executeParams, "${parameters.", "}");
             input = substitutor.replace(input);
             executeParams.put("input", input);
+            // Remove the current input from params to avoid passing it to the following tools.
+            // Otherwise, it will override the default input of the following tools.
+            params.remove("input");
         }
         return executeParams;
     }