Add pre and post process functions for Bedrock Rerank API opensearch-…

…project#3254 Signed-off-by: tkykenmt <[email protected]>
tkykenmt · Jan 6, 2025 · 3a4ae65 · 3a4ae65
1 parent a1de787
commit 3a4ae65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...pensearch/ml/common/connector/functions/postprocess/BedrockRerankPostProcessFunction.java b/...pensearch/ml/common/connector/functions/postprocess/BedrockRerankPostProcessFunction.java
@@ -52,7 +52,7 @@ public List<ModelTensor> process(List<Map<String, Object>> rerankResults) {
                             .name("similarity")
                             .shape(new long[] { 1 })
                             .data(new Number[] { scores[i].doubleValue() })
-                            .dataType(MLResultDataType.FLOAT32)
+                            .dataType(MLResultDataType.FLOAT64)
                             .build()
                     );
             }