bug fix for similarity function

nosqlbench · May 10, 2024 · 031fd5f · 031fd5f
1 parent 101e9d9
commit 031fd5f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...n/java/io/nosqlbench/adapter/dataapi/opdispensers/DataApiCreateCollectionOpDispenser.java b/...n/java/io/nosqlbench/adapter/dataapi/opdispensers/DataApiCreateCollectionOpDispenser.java
@@ -51,7 +51,7 @@ private LongFunction<DataApiCreateCollectionOp> createOpFunction(ParsedOp op) {
             Optional<LongFunction<String>> simFunc = op.getAsOptionalFunction("similarity", String.class);
             if (simFunc.isPresent()) {
                 LongFunction<String> sf = simFunc.get();
-                optionsBldr.vectorSimilarity(SimilarityMetric.valueOf(sf.apply(l)));
+                optionsBldr.vectorSimilarity(SimilarityMetric.fromValue(sf.apply(l)));
             }
 
             DataApiCreateCollectionOp dataApiCreateCollectionOp =

diff --git a/nb-adapters/adapter-dataapi/src/main/resources/activities/create_collection.yaml b/nb-adapters/adapter-dataapi/src/main/resources/activities/create_collection.yaml
@@ -8,4 +8,4 @@ blocks:
       op1:
         create_collection: "collectionName"
         dimensions: 10
-        similarity: "cosine"
+        similarity: "COSINE"