Skip to content

Commit 4b9afec

Browse files
committed
fix predict issue
Signed-off-by: Jing Zhang <[email protected]>
1 parent 17a9037 commit 4b9afec

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

plugin/src/main/java/org/opensearch/ml/rest/RestMLPredictionAction.java

+6
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,12 @@ public RestChannelConsumer prepareRequest(RestRequest request, NodeClient client
8888
String modelId = getParameterId(request, PARAMETER_MODEL_ID);
8989
Optional<FunctionName> functionName = modelManager.getOptionalModelFunctionName(modelId);
9090

91+
if (userAlgorithm != null) {
92+
MLPredictionTaskRequest mlPredictionTaskRequest = getRequest(modelId, "", userAlgorithm, request);
93+
return channel -> client
94+
.execute(MLPredictionTaskAction.INSTANCE, mlPredictionTaskRequest, new RestToXContentListener<>(channel));
95+
}
96+
9197
// check if the model is in cache
9298
if (functionName.isPresent()) {
9399
MLPredictionTaskRequest predictionRequest = getRequest(

0 commit comments

Comments
 (0)