debugging

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

Author: b4sjoo

plugin/src/main/java/org/opensearch/ml/action/prediction/TransportPredictionTaskAction.java

@@ -156,9 +156,14 @@ public void onResponse(MLModel mlModel) {
                                                     )
                                                 );
                                         } else {
+                                            log.info("ModelId " + modelId + " is enabled for prediction in local layer.");
+                                            log.info(mlPredictionTaskRequest.getMlInput().toXContent(XContentFactory.jsonBuilder(), ToXContent.EMPTY_PARAMS).toString());
+                                            validateInputSchema(modelId, mlPredictionTaskRequest.getMlInput());
                                             executePredict(mlPredictionTaskRequest, wrappedListener, modelId);
                                         }
                                     } else {
+                                        log.info("ModelId " + modelId + " is enabled for prediction in remote layer.");
+                                        log.info(mlPredictionTaskRequest.getMlInput().toXContent(XContentFactory.jsonBuilder(), ToXContent.EMPTY_PARAMS).toString());
                                         validateInputSchema(modelId, mlPredictionTaskRequest.getMlInput());
                                         executePredict(mlPredictionTaskRequest, wrappedListener, modelId);
                                     }
@@ -234,8 +239,10 @@ private void executePredict(
     }
 
     private void validateInputSchema(String modelId, MLInput mlInput) {
+        log.info("Went to validate input schema");
         if (modelCacheHelper.getModelInterface(modelId) != null && modelCacheHelper.getModelInterface(modelId).get("input") != null) {
             String inputSchemaString = modelCacheHelper.getModelInterface(modelId).get("input");
+            log.info(inputSchemaString);
             try {
                 MLNodeUtils
                     .validateSchema(

plugin/src/main/java/org/opensearch/ml/task/MLPredictTaskRunner.java

@@ -380,12 +380,15 @@ private void runPredict(
                             ((MLPredictionOutput) output).setStatus(MLTaskState.COMPLETED.name());
                         }
                         if (output instanceof ModelTensorOutput) {
+                            log.info("ModelId " + modelId + " is completed for prediction.");
                             // pick the first output tensor to validate the schema
                             if (((ModelTensorOutput) output).getMlModelOutputs() != null
                                 && !((ModelTensorOutput) output).getMlModelOutputs().isEmpty()) {
                                 ModelTensors modelTensors = ((ModelTensorOutput) output).getMlModelOutputs().get(0);
+                                log.info(modelTensors.toXContent(XContentFactory.jsonBuilder(), ToXContent.EMPTY_PARAMS).toString());
                                 if (modelTensors.getMlModelTensors() != null && !modelTensors.getMlModelTensors().isEmpty()) {
                                     ModelTensor modelTensorOutput = modelTensors.getMlModelTensors().get(0);
+                                    log.info(modelTensorOutput.toXContent(XContentFactory.jsonBuilder(), ToXContent.EMPTY_PARAMS).toString());
                                     validateOutputSchema(modelId, modelTensorOutput);
                                 }
                             }
@@ -447,8 +450,10 @@ private void handlePredictFailure(
     }
 
     private void validateOutputSchema(String modelId, ModelTensor modelTensor) {
+        log.info("Went to validate output schema");
         if (mlModelManager.getModelInterface(modelId) != null && mlModelManager.getModelInterface(modelId).get("output") != null) {
             String outputSchemaString = mlModelManager.getModelInterface(modelId).get("output");
+            log.info(outputSchemaString);
             try {
                 MLNodeUtils
                     .validateSchema(

plugin/src/main/java/org/opensearch/ml/utils/MLNodeUtils.java

@@ -67,11 +67,9 @@ public static <T> void parseField(XContentParser parser, Set<T> set, Function<St
 
     public static void validateSchema(String schemaString, String instanceString) throws IOException {
         ObjectMapper mapper = new ObjectMapper();
-        System.out.println(schemaString);
         // parse the schema JSON as string
         JsonNode schemaNode = mapper.readTree(schemaString);
         JsonSchema schema = JsonSchemaFactory.getInstance(VersionFlag.V202012).getSchema(schemaNode);
-        System.out.println(instanceString);
         // JSON data to validate
         JsonNode jsonNode = mapper.readTree(instanceString);