Fix parser poc

Signed-off-by: Sicheng Song <sicheng.song@outlook.com>

Author: b4sjoo

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/ConnectorUtils.java

@@ -18,12 +18,15 @@
 import java.util.*;
 import java.util.function.Function;
 
+import org.json.JSONException;
 import org.json.JSONObject;
 import org.json.JSONArray;
 import org.apache.commons.lang3.StringUtils;
 import org.apache.commons.text.StringSubstitutor;
 import org.opensearch.OpenSearchParseException;
 import org.opensearch.OpenSearchStatusException;
+import org.opensearch.common.xcontent.XContentFactory;
+import org.opensearch.core.xcontent.ToXContent;
 import org.opensearch.core.xcontent.XContent;
 import org.opensearch.core.xcontent.XContentBuilder;
 import org.opensearch.ml.common.connector.Connector;
@@ -265,6 +268,16 @@ public static SdkHttpFullRequest signRequest(
         return signer.sign(request, params);
     }
 
+    public static JSONObject parametersObjectExtractor(MLInput mlInput) throws IOException {
+            XContentBuilder mlInputXContent = mlInput.toXContent(XContentFactory.jsonBuilder(), ToXContent.EMPTY_PARAMS);
+        try {
+            JSONObject mlInputObject = new JSONObject(mlInputXContent.toString());
+            return (JSONObject) mlInputObject.get("parameters");
+        } catch (JSONException e) {
+            throw new OpenSearchParseException("Failed to extract parameters object from MLInput", e.getMessage());
+        }
+    }
+
     public static void validateSchema(String schemaString, String instanceString) {
         try {
             // parse the schema JSON as string

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/RemoteConnectorExecutor.java

@@ -168,8 +168,8 @@ private void validateInputSchema(MLInput mlInput) {
             String inputSchemaString = getConnector().getModelInterface().get("input");
             try {
                 JSONObject inputSchemaObject = ConnectorUtils.processJsonObject(new JSONObject(inputSchemaString));
-                System.out.println(mlInput.getInputDataset().toString());
-                ConnectorUtils.validateSchema(inputSchemaObject.toString(), mlInput.toXContent(XContentFactory.jsonBuilder(), ToXContent.EMPTY_PARAMS).toString());
+                JSONObject parametersObject = ConnectorUtils.parametersObjectExtractor(mlInput);
+                ConnectorUtils.validateSchema(inputSchemaObject.toString(), parametersObject.toString());
             } catch (IOException | JSONException e) {
                 throw new IllegalArgumentException("Error validating input schema: " + e.getMessage());
             }