Rebased and fixed escape JSON

Signed-off-by: Arjun kumar Giri <arjung@amazon.com>
arjunkumargiri · Dec 5, 2023 · 6dd5242 · 6dd5242
1 parent 01480ba
commit 6dd5242
Showing 1 changed file with 3 additions and 3 deletions.
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
@@ -117,7 +117,7 @@ public void run(MLAgent mlAgent, Map<String, String> params, ActionListener<Obje
                     }
 
                     String outputResponse = parseResponse(output);
-                    params.put(outputKey, outputResponse);
+                    params.put(outputKey, escapeJson(outputResponse));
                     additionalInfo.put(outputKey, outputResponse);
 
                     if (finalI == toolSpecs.size()) {
@@ -175,12 +175,12 @@ private String parseResponse(Object output) {
             return response + "";
 
         } else if (output instanceof ModelTensor) {
-            return escapeJson(toJson(((ModelTensor) output).getDataAsMap()));
+            return toJson(((ModelTensor) output).getDataAsMap());
         } else {
             if (output instanceof String) {
                 return (String) output;
             } else {
-                return escapeJson(toJson(output));
+                return toJson(output);
             }
         }
     }