address comments III

Signed-off-by: Jing Zhang <[email protected]>
opensearch-project · Oct 21, 2024 · 8bca218 · 8bca218
1 parent cf68059
commit 8bca218
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/AgentUtils.java b/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/AgentUtils.java
@@ -480,6 +480,10 @@ public static Map<String, String> constructToolParams(
             StringSubstitutor substitutor = new StringSubstitutor(toolParams, "${parameters.", "}");
             input = substitutor.replace(input);
             toolParams.put("input", input);
+            if (isJson(input)) {
+                Map<String, String> params = getParameterMap(gson.fromJson(input, Map.class));
+                toolParams.putAll(params);
+            }
         } else {
             toolParams.put("input", actionInput);
             if (isJson(actionInput)) {