Change default pre/post process function name

Signed-off-by: zane-neo <zaniu@amazon.com>
zane-neo · Sep 15, 2023 · a26c686 · a26c686
1 parent fe25fcf
commit a26c686
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/common/src/main/java/org/opensearch/ml/common/connector/MLPostProcessFunction.java b/common/src/main/java/org/opensearch/ml/common/connector/MLPostProcessFunction.java
@@ -19,7 +19,7 @@ public class MLPostProcessFunction {
     public static final String COHERE_EMBEDDING = "connector.post_process.cohere.embedding";
     public static final String OPENAI_EMBEDDING = "connector.post_process.openai.embedding";
 
-    public static final String NEURAL_SEARCH_EMBEDDING = "connector.post_process.neural_search.text_embedding";
+    public static final String NEURAL_SEARCH_EMBEDDING = "connector.post_process.default";
 
     private static final Map<String, String> JSON_PATH_EXPRESSION = new HashMap<>();
 

diff --git a/common/src/main/java/org/opensearch/ml/common/connector/MLPreProcessFunction.java b/common/src/main/java/org/opensearch/ml/common/connector/MLPreProcessFunction.java
@@ -16,7 +16,7 @@ public class MLPreProcessFunction {
     public static final String TEXT_DOCS_TO_COHERE_EMBEDDING_INPUT = "connector.pre_process.cohere.embedding";
     public static final String TEXT_DOCS_TO_OPENAI_EMBEDDING_INPUT = "connector.pre_process.openai.embedding";
 
-    public static final String NEURAL_SEARCH_EMBEDDING_INPUT = "connector.pre_process.neural_search.text_embedding";
+    public static final String NEURAL_SEARCH_EMBEDDING_INPUT = "connector.pre_process.default";
 
     private static Function<List<String>, Map<String, Object>> cohereTextEmbeddingPreProcess() {
         return inputs -> Map.of("parameters", Map.of("texts", inputs));