diff --git a/adapter-pinecone/src/main/java/io/nosqlbench/adapter/pinecone/opdispensers/PineconeQueryOpDispenser.java b/adapter-pinecone/src/main/java/io/nosqlbench/adapter/pinecone/opdispensers/PineconeQueryOpDispenser.java index 68c2d5108..3c7412c1d 100644 --- a/adapter-pinecone/src/main/java/io/nosqlbench/adapter/pinecone/opdispensers/PineconeQueryOpDispenser.java +++ b/adapter-pinecone/src/main/java/io/nosqlbench/adapter/pinecone/opdispensers/PineconeQueryOpDispenser.java @@ -77,7 +77,7 @@ public class PineconeQueryOpDispenser extends PineconeOpDispenser { rFunc = l -> finalFunc.apply(l).setNamespace(af.apply(l)); } - Optional> tFunc = op.getAsOptionalFunction("topk", Integer.class); + Optional> tFunc = op.getAsOptionalFunction("top_k", Integer.class); if (tFunc.isPresent()) { LongFunction finalFunc = rFunc; LongFunction af = tFunc.get(); diff --git a/adapter-pinecone/src/main/resources/activities/pinecone-query-multiple.yaml b/adapter-pinecone/src/main/resources/activities/pinecone-query-multiple.yaml new file mode 100644 index 000000000..185deab5b --- /dev/null +++ b/adapter-pinecone/src/main/resources/activities/pinecone-query-multiple.yaml @@ -0,0 +1,27 @@ +scenarios: + default: + query: run driver=pinecone tags==blocks:query cycles=10 + +blocks: + query: + ops: + op1: + query: "test-index" + top_k: 3 + query_vectors: + - id: "1" + values: "0.8602578079921012,0.12103044768221516,0.7737329191858439,0.4521093269320254,0.29351661477669416,0.4261807015226558,0.14131665592103335,0.882370813029422,0.4412833140430886,0.9916525700115515" + namespace: "example_namespace" + top_k: 3 + filter: + filterfield: "category" + operator: "$eq" + comparator: "movies" + - id: "2" + values: "0.8602578079921012,0.12103044768221516,0.7737329191858439,0.4521093269320254,0.29351661477669416,0.4261807015226558,0.14131665592103335,0.882370813029422,0.4412833140430886,0.9916525700115515" + namespace: "example_namespace" + top_k: 3 + filter: + filterfield: "category" + operator: "$eq" + comparator: "sports" diff --git a/adapter-pinecone/src/main/resources/activities/pinecone-query.yaml b/adapter-pinecone/src/main/resources/activities/pinecone-query.yaml index a4a42149b..b76462818 100644 --- a/adapter-pinecone/src/main/resources/activities/pinecone-query.yaml +++ b/adapter-pinecone/src/main/resources/activities/pinecone-query.yaml @@ -9,7 +9,7 @@ blocks: query: "test-index" vector: "0.8602578079921012,0.12103044768221516,0.7737329191858439,0.4521093269320254,0.29351661477669416,0.4261807015226558,0.14131665592103335,0.882370813029422,0.4412833140430886,0.9916525700115515" namespace: "example_namespace" - topk: 10 + top_k: 10 include_values: true include_metadata: true filter: