add MLInferenceSearchResponseProcessor

Signed-off-by: Mingshi Liu <[email protected]>
opensearch-project · Jul 20, 2024 · bf4eda2 · bf4eda2
1 parent 6ff84c1
commit bf4eda2
Show file tree

Hide file tree

Showing 6 changed files with 1,237 additions and 1 deletion.
diff --git a/plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java b/plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java
@@ -213,6 +213,7 @@
 import org.opensearch.ml.model.MLModelCacheHelper;
 import org.opensearch.ml.model.MLModelManager;
 import org.opensearch.ml.processor.MLInferenceIngestProcessor;
+import org.opensearch.ml.processor.MLInferenceSearchResponseProcessor;
 import org.opensearch.ml.repackage.com.google.common.collect.ImmutableList;
 import org.opensearch.ml.rest.RestMLCreateConnectorAction;
 import org.opensearch.ml.rest.RestMLCreateControllerAction;
@@ -990,6 +991,12 @@ public Map<String, Processor.Factory<SearchResponseProcessor>> getResponseProces
                 new GenerativeQAResponseProcessor.Factory(this.client, () -> this.ragSearchPipelineEnabled)
             );
 
+        responseProcessors
+            .put(
+                MLInferenceSearchResponseProcessor.TYPE,
+                new MLInferenceSearchResponseProcessor.Factory(parameters.client, parameters.namedXContentRegistry)
+            );
+
         return responseProcessors;
     }