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

adding Doc type for results

Browse Source
This commit is contained in:
Mark Wolters 2024-02-08 17:21:22 -04:00
parent b7b790766e
commit b16b04f466
2 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
adapter-aws-opensearch/src/main/java/io/nosqlbench/adapter/opensearch/dispensers/KnnSearchOpDispenser.java
View File

@ -52,7 +52,7 @@ public class KnnSearchOpDispenser extends BaseOpenSearchOpDispenser {
.index(targetF.apply(1)) .index(targetF.apply(1))
.query(new Query.Builder().knn(finalKnnfunc.apply(l).build()).build()); .query(new Query.Builder().knn(finalKnnfunc.apply(l).build()).build());
return (long l) -> new KnnSearchOp(clientF.apply(l), bfunc.apply(l).build(), Object.class); return (long l) -> new KnnSearchOp(clientF.apply(l), bfunc.apply(l).build(), Doc.class);
} }
private KnnQuery.Builder convertVector(KnnQuery.Builder builder, List list) { private KnnQuery.Builder convertVector(KnnQuery.Builder builder, List list) {
@ -63,5 +63,22 @@ public class KnnSearchOpDispenser extends BaseOpenSearchOpDispenser {
return builder.vector(vector); return builder.vector(vector);
} }
public static class Doc {
private float[] values;
public Doc() {}
public Doc(float[] values) {
this.values = values;
}
public float[] getValues() {
return values;
}
public void setValues(float[] values) {
this.values = values;
}
@Override
public String toString() {
return "{" + "values=" + values + "}";
}
}
} }

1
adapter-aws-opensearch/src/main/resources/activities/osvectors.yaml
View File

@ -21,3 +21,4 @@ blocks:
knn_search: TEMPLATE(indexname,vectors_index) knn_search: TEMPLATE(indexname,vectors_index)
k: 100 k: 100
vector: [1f,2f,3f,4f,5f,6f,7f,8f,9f,10f,11f,12f,13f,14f,15f,16f,17f,18f,19f,20f,21f,22f,23f,24f,25f] vector: [1f,2f,3f,4f,5f,6f,7f,8f,9f,10f,11f,12f,13f,14f,15f,16f,17f,18f,19f,20f,21f,22f,23f,24f,25f]
field: value