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

@@ -311,6 +311,8 @@ public static JSONObject wrapProperties(JSONObject jsonObject) {
                 // Check if the nested object has any schema keyword field
                 if (nestedObject.has("type") || nestedObject.has("properties") || nestedObject.has("description")) {
                     newObject.put(key, nestedObject); // Leave as is, because it's likely a schema definition
+                } else if (key.equals("properties")) {
+                    newObject.put(key, wrapProperties(nestedObject));
                 } else {
                     // Recurse to handle nested objects and wrap them with "properties"
                     newObject.put(key, new JSONObject().put("properties", wrapProperties(nestedObject)));
@@ -330,7 +332,7 @@ private static JSONArray wrapPropertiesInArray(JSONArray jsonArray) {
         for (int i = 0; i < jsonArray.length(); i++) {
             Object item = jsonArray.get(i);
             if (item instanceof JSONObject) {
-                newArray.put(processJsonObject((JSONObject) item));
+                newArray.put(wrapProperties((JSONObject) item));
             } else if (item instanceof JSONArray) {
                 newArray.put(wrapPropertiesInArray((JSONArray) item));
             } else {

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

@@ -168,6 +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.getParameters().toXContent(XContentFactory.jsonBuilder(), ToXContent.EMPTY_PARAMS).toString());
+                System.out.println(mlInput.getInputDataset().toString());
                 ConnectorUtils.validateSchema(inputSchemaObject.toString(), mlInput.toXContent(XContentFactory.jsonBuilder(), ToXContent.EMPTY_PARAMS).toString());
             } catch (IOException | JSONException e) {
                 throw new IllegalArgumentException("Error validating input schema: " + e.getMessage());