Fix agent param validation

Signed-off-by: Nathalie Jonathan <[email protected]>

Author: nathaliellenaa

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/agent/AgentUtils.java

@@ -996,7 +996,10 @@ public static Tool createTool(Map<String, Tool.Factory> toolFactories, Map<Strin
             throw new IllegalArgumentException("Tool not found: " + toolSpec.getType());
         }
         Map<String, Object> toolParams = new HashMap<>();
-        toolParams.putAll(executeParams);
+        // Parse JSON strings back to original type since we need to validate each parameter type when creating tool
+        for (Map.Entry<String, String> entry : executeParams.entrySet()) {
+            toolParams.put(entry.getKey(), parseValue(entry.getValue()));
+        }
         Map<String, Object> runtimeResources = toolSpec.getRuntimeResources();
         if (runtimeResources != null) {
             toolParams.putAll(runtimeResources);
@@ -1014,4 +1017,32 @@ public static Tool createTool(Map<String, Tool.Factory> toolFactories, Map<Strin
 
         return tool;
     }
+
+    private static Object parseValue(String value) {
+        if (value == null || "null".equals(value)) {
+            return null;
+        }
+        String v = value.trim();
+
+        // Try JSON array
+        if (v.startsWith("[") && v.endsWith("]")) {
+            try {
+                return gson.fromJson(v, List.class);
+            } catch (Exception e) {
+                return value;
+            }
+        }
+
+        // Try boolean
+        if ("true".equalsIgnoreCase(v) || "false".equalsIgnoreCase(v)) {
+            return Boolean.parseBoolean(v);
+        }
+
+        // Try integer
+        try {
+            return Integer.parseInt(v);
+        } catch (NumberFormatException e) {
+            return value;
+        }
+    }
 }