Change multimodal connector name to bedrock multimodal connector

Signed-off-by: zane-neo <zaniu@amazon.com>
zane-neo · Jul 18, 2024 · aff666a · aff666a
1 parent be8171f
commit aff666a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
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
@@ -23,7 +23,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 TEXT_DOCS_TO_BEDROCK_EMBEDDING_INPUT = "connector.pre_process.bedrock.embedding";
-    public static final String TEXT_IMAGE_TO_BEDROCK_EMBEDDING_INPUT = "connector.pre_process.multimodal.embedding";
+    public static final String TEXT_IMAGE_TO_BEDROCK_EMBEDDING_INPUT = "connector.pre_process.bedrock.multimodal_embedding";
     public static final String TEXT_DOCS_TO_DEFAULT_EMBEDDING_INPUT = "connector.pre_process.default.embedding";
     public static final String TEXT_SIMILARITY_TO_COHERE_RERANK_INPUT = "connector.pre_process.cohere.rerank";
     public static final String TEXT_SIMILARITY_TO_DEFAULT_INPUT = "connector.pre_process.default.rerank";

diff --git a/...src/test/resources/org/opensearch/ml/rest/templates/BedRockMultiModalConnectorBodies.json b/...src/test/resources/org/opensearch/ml/rest/templates/BedRockMultiModalConnectorBodies.json
@@ -24,7 +24,7 @@
           "x-amz-content-sha256": "required"
         },
         "request_body": "{ \"inputText\": \"${parameters.inputText}\", \"inputImage\": \"${parameters.inputImage:-null}\" }",
-        "pre_process_function": "connector.pre_process.multimodal.embedding",
+        "pre_process_function": "connector.pre_process.bedrock.multimodal_embedding",
         "post_process_function": "connector.post_process.bedrock.embedding"
       }
     ]
@@ -55,7 +55,7 @@
           "x-amz-content-sha256": "required"
         },
         "request_body": "{ \"inputText\": \"${parameters.inputText}\", \"inputImage\": \"${parameters.inputImage:-null}\" }",
-        "pre_process_function": "connector.pre_process.multimodal.embedding",
+        "pre_process_function": "connector.pre_process.bedrock.multimodal_embedding",
         "post_process_function": "connector.post_process.bedrock.embedding"
       }
     ]