IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

fixed top_k spelling, added test for multiple vector query

Browse Source
This commit is contained in:
Mark Wolters 2023-06-01 14:35:50 +00:00
parent 6b63f9cc37
commit b6686ab955
3 changed files with 29 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
adapter-pinecone/src/main/java/io/nosqlbench/adapter/pinecone/opdispensers/PineconeQueryOpDispenser.java
View File

@ -77,7 +77,7 @@ public class PineconeQueryOpDispenser extends PineconeOpDispenser {
rFunc = l -> finalFunc.apply(l).setNamespace(af.apply(l));
}
Optional<LongFunction<Integer>> tFunc = op.getAsOptionalFunction("topk", Integer.class);
Optional<LongFunction<Integer>> tFunc = op.getAsOptionalFunction("top_k", Integer.class);
if (tFunc.isPresent()) {
LongFunction<QueryRequest.Builder> finalFunc = rFunc;
LongFunction<Integer> af = tFunc.get();

27
adapter-pinecone/src/main/resources/activities/pinecone-query-multiple.yaml Normal file
View File

@ -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"

2
adapter-pinecone/src/main/resources/activities/pinecone-query.yaml
View File

@ -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: