From 1cfd2726e29c821ac5d3ccb0b22f7430b20d5d8c Mon Sep 17 00:00:00 2001 From: Mingshi Liu Date: Fri, 20 Feb 2026 17:10:18 -0800 Subject: [PATCH] fix failing IT Signed-off-by: Mingshi Liu --- .../ml/rest/RestMLInferenceSearchRequestProcessorIT.java | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLInferenceSearchRequestProcessorIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLInferenceSearchRequestProcessorIT.java index 98fe2b0bf7..03c59b9da4 100644 --- a/plugin/src/test/java/org/opensearch/ml/rest/RestMLInferenceSearchRequestProcessorIT.java +++ b/plugin/src/test/java/org/opensearch/ml/rest/RestMLInferenceSearchRequestProcessorIT.java @@ -101,9 +101,7 @@ public class RestMLInferenceSearchRequestProcessorIT extends MLCommonsRestTestCa + " \"content-type\": \"application/json\",\n" + " \"x-amz-content-sha256\": \"required\"\n" + " },\n" - + " \"request_body\": \"{ \\\"inputText\\\": \\\"${parameters.input}\\\" }\",\n" - + " \"pre_process_function\": \"connector.pre_process.bedrock.embedding\",\n" - + " \"post_process_function\": \"connector.post_process.bedrock.embedding\"\n" + + " \"request_body\": \"{ \\\"inputText\\\": \\\"${parameters.input}\\\" }\"\n" + " }\n" + " ]\n" + "}"; @@ -343,7 +341,7 @@ public void testMLInferenceProcessorRemoteModelRewriteQueryType() throws Excepti + " \"model_id\": \"" + this.bedrockEmbeddingModelId + "\",\n" - + " \"query_template\": \"{\\\"query\\\":{\\\"range\\\":{\\\"diary_embedding_size\\\":{\\\"lte\\\":${modelPrediction}}}}}\",\n" + + " \"query_template\": \"{\\\"query\\\":{\\\"range\\\":{\\\"diary_embedding_size_int\\\":{\\\"lte\\\":${modelPrediction}}}}}\",\n" + " \"input_map\": [\n" + " {\n" + " \"input\": \"query.term.diary_embedding_size.value\"\n"