add logs

Signed-off-by: xinyual <xinyual@amazon.com>

Author: xinyual

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/AgentTool.java

@@ -24,6 +24,8 @@
 import lombok.Setter;
 import lombok.extern.log4j.Log4j2;
 
+import static org.opensearch.ml.common.utils.StringUtils.gson;
+
 /**
  * This tool supports running any Agent.
  */
@@ -51,6 +53,11 @@ public AgentTool(Client client, String agentId) {
 
     @Override
     public <T> void run(Map<String, String> parameters, ActionListener<T> listener) {
+        log.info("Agent tool before");
+        log.info(parameters);
+        parameters = extractFromChatParameters(parameters);
+        log.info("Agent tool after");
+        log.info(parameters);
         AgentMLInput agentMLInput = AgentMLInput
             .AgentMLInputBuilder()
             .agentId(agentId)
@@ -135,4 +142,16 @@ public String getDefaultVersion() {
             return null;
         }
     }
+
+    private Map<String, String> extractFromChatParameters(Map<String, String> parameters) {
+        if (parameters.containsKey("input")) {
+            try {
+                Map<String, String> chatParameters = gson.fromJson(parameters.get("input"), Map.class);
+                parameters.putAll(chatParameters);
+            } finally {
+                return parameters;
+            }
+        }
+        return parameters;
+    }
 }

ml-algorithms/src/main/java/org/opensearch/ml/engine/tools/MLModelTool.java

@@ -65,6 +65,8 @@ public MLModelTool(Client client, String modelId) {
 
     @Override
     public <T> void run(Map<String, String> parameters, ActionListener<T> listener) {
+        log.info("ML input");
+        log.info(parameters);
         RemoteInferenceInputDataSet inputDataSet = RemoteInferenceInputDataSet.builder().parameters(parameters).build();
         ActionRequest request = new MLPredictionTaskRequest(
             modelId,