fix no worker node error on multi-node cluster (#1487)

ylwu-amzn · web-flow · commit cea1cd675cd9 · 2023-10-11T10:40:11.000-07:00
Signed-off-by: Yaliang Wu &lt;ylwu@amazon.com&gt;
diff --git a/plugin/src/main/java/org/opensearch/ml/action/prediction/TransportPredictionTaskAction.java b/plugin/src/main/java/org/opensearch/ml/action/prediction/TransportPredictionTaskAction.java
@@ -121,9 +121,8 @@ public void onFailure(Exception e) {
 
             if (cachedMlModel != null) {
                 modelActionListener.onResponse(cachedMlModel);
-            } else if (modelAccessControlHelper.skipModelAccessControl(user)) {
-                executePredict(mlPredictionTaskRequest, wrappedListener, modelId);
             } else {
+                // For multi-node cluster, the function name is null in cache, so should always get model first.
                 mlModelManager.getModel(modelId, modelActionListener);
             }
         }

Original file line number	Diff line number	Diff line change
`@@ -121,9 +121,8 @@ public void onFailure(Exception e) {`
`121`	`121`
`122`	`122`	`if (cachedMlModel != null) {`
`123`	`123`	`modelActionListener.onResponse(cachedMlModel);`
`124`		`- } else if (modelAccessControlHelper.skipModelAccessControl(user)) {`
`125`		`- executePredict(mlPredictionTaskRequest, wrappedListener, modelId);`
`126`	`124`	`} else {`
	`125`	`+ // For multi-node cluster, the function name is null in cache, so should always get model first.`
`127`	`126`	`mlModelManager.getModel(modelId, modelActionListener);`
`128`	`127`	`}`
`129`	`128`	`}`