Skip to content

Commit b44ec48

Browse files
committed
Fix knn query usage in other modules
1 parent 6c2c1be commit b44ec48

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

modules/percolator/src/internalClusterTest/java/org/elasticsearch/percolator/PercolatorQuerySearchIT.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1359,7 +1359,7 @@ public void testKnnQueryNotSupportedInPercolator() throws IOException {
13591359
""");
13601360
indicesAdmin().prepareCreate("index1").setMapping(mappings).get();
13611361
ensureGreen();
1362-
QueryBuilder knnVectorQueryBuilder = new KnnVectorQueryBuilder("my_vector", new float[] { 1, 1, 1, 1, 1 }, 10, 10, null);
1362+
QueryBuilder knnVectorQueryBuilder = new KnnVectorQueryBuilder("my_vector", new float[] { 1, 1, 1, 1, 1 }, 10, 10, null, null);
13631363

13641364
IndexRequestBuilder indexRequestBuilder = prepareIndex("index1").setId("knn_query1")
13651365
.setSource(jsonBuilder().startObject().field("my_query", knnVectorQueryBuilder).endObject());

x-pack/plugin/rank-rrf/src/internalClusterTest/java/org/elasticsearch/xpack/rank/rrf/RRFRetrieverTelemetryIT.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ public void testTelemetryForRRFRetriever() throws IOException {
117117
{
118118
performSearch(
119119
new SearchSourceBuilder().retriever(
120-
new StandardRetrieverBuilder(new KnnVectorQueryBuilder("vector", new float[] { 1.0f }, 10, 15, null))
120+
new StandardRetrieverBuilder(new KnnVectorQueryBuilder("vector", new float[] { 1.0f }, 10, 15, null, null))
121121
)
122122
);
123123
}

x-pack/plugin/security/src/internalClusterTest/java/org/elasticsearch/integration/DocumentLevelSecurityTests.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -884,7 +884,7 @@ public void testKnnSearch() throws Exception {
884884
// Since there's no kNN search action at the transport layer, we just emulate
885885
// how the action works (it builds a kNN query under the hood)
886886
float[] queryVector = new float[] { 0.0f, 0.0f, 0.0f };
887-
KnnVectorQueryBuilder query = new KnnVectorQueryBuilder("vector", queryVector, 50, 50, null);
887+
KnnVectorQueryBuilder query = new KnnVectorQueryBuilder("vector", queryVector, 50, 50, null, null);
888888

889889
if (randomBoolean()) {
890890
query.addFilterQuery(new WildcardQueryBuilder("other", "value*"));

x-pack/plugin/security/src/internalClusterTest/java/org/elasticsearch/integration/FieldLevelSecurityTests.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -441,7 +441,7 @@ public void testKnnSearch() throws IOException {
441441
// Since there's no kNN search action at the transport layer, we just emulate
442442
// how the action works (it builds a kNN query under the hood)
443443
float[] queryVector = new float[] { 0.0f, 0.0f, 0.0f };
444-
KnnVectorQueryBuilder query = new KnnVectorQueryBuilder("vector", queryVector, 10, 10, null);
444+
KnnVectorQueryBuilder query = new KnnVectorQueryBuilder("vector", queryVector, 10, 10, null, null);
445445

446446
// user1 has access to vector field, so the query should match with the document:
447447
assertResponse(
@@ -475,7 +475,7 @@ public void testKnnSearch() throws IOException {
475475
}
476476
);
477477
// user1 can access field1, so the filtered query should match with the document:
478-
KnnVectorQueryBuilder filterQuery1 = new KnnVectorQueryBuilder("vector", queryVector, 10, 10, null).addFilterQuery(
478+
KnnVectorQueryBuilder filterQuery1 = new KnnVectorQueryBuilder("vector", queryVector, 10, 10, null, null).addFilterQuery(
479479
QueryBuilders.matchQuery("field1", "value1")
480480
);
481481
assertHitCount(
@@ -486,7 +486,7 @@ public void testKnnSearch() throws IOException {
486486
);
487487

488488
// user1 cannot access field2, so the filtered query should not match with the document:
489-
KnnVectorQueryBuilder filterQuery2 = new KnnVectorQueryBuilder("vector", queryVector, 10, 10, null).addFilterQuery(
489+
KnnVectorQueryBuilder filterQuery2 = new KnnVectorQueryBuilder("vector", queryVector, 10, 10, null, null).addFilterQuery(
490490
QueryBuilders.matchQuery("field2", "value2")
491491
);
492492
assertHitCount(

0 commit comments

Comments
 (0)