| 
19 | 19 | import org.elasticsearch.common.settings.Settings;  | 
20 | 20 | import org.elasticsearch.rest.RestStatus;  | 
21 | 21 | import org.elasticsearch.search.vectors.KnnVectorQueryBuilder;  | 
22 |  | -import org.elasticsearch.search.vectors.VectorData;  | 
23 | 22 | import org.elasticsearch.test.ESIntegTestCase;  | 
24 | 23 | import org.elasticsearch.xcontent.XContentBuilder;  | 
25 | 24 | import org.elasticsearch.xcontent.XContentFactory;  | 
@@ -131,19 +130,25 @@ public void testDenseVectorMappingUpdate() throws Exception {  | 
131 | 130 |         int expectedDocs = docsBefore + docsAfter;  | 
132 | 131 | 
 
  | 
133 | 132 |         // Count query  | 
134 |  | -        assertNoFailuresAndResponse(client().prepareSearch(INDEX_NAME).setSize(0).setTrackTotalHits(true).setSize(expectedDocs), response -> {  | 
135 |  | -            assertHitCount(response, expectedDocs);  | 
136 |  | -        });  | 
 | 133 | +        assertNoFailuresAndResponse(  | 
 | 134 | +            client().prepareSearch(INDEX_NAME).setSize(0).setTrackTotalHits(true).setSize(expectedDocs),  | 
 | 135 | +            response -> {  | 
 | 136 | +                assertHitCount(response, expectedDocs);  | 
 | 137 | +            }  | 
 | 138 | +        );  | 
137 | 139 | 
 
  | 
138 | 140 |         // KNN query  | 
139 | 141 |         float[] queryVector = new float[dimensions];  | 
140 | 142 |         for (int i = 0; i < queryVector.length; i++) {  | 
141 | 143 |             queryVector[i] = randomFloatBetween(-1, 1, true);  | 
142 | 144 |         }  | 
143 | 145 |         KnnVectorQueryBuilder queryBuilder = new KnnVectorQueryBuilder(VECTOR_FIELD, queryVector, null, null, null, null);  | 
144 |  | -        assertNoFailuresAndResponse(client().prepareSearch(INDEX_NAME).setQuery(queryBuilder).setTrackTotalHits(true).setSize(expectedDocs), response -> {  | 
145 |  | -            assertHitCount(response, expectedDocs);  | 
146 |  | -        });  | 
 | 146 | +        assertNoFailuresAndResponse(  | 
 | 147 | +            client().prepareSearch(INDEX_NAME).setQuery(queryBuilder).setTrackTotalHits(true).setSize(expectedDocs),  | 
 | 148 | +            response -> {  | 
 | 149 | +                assertHitCount(response, expectedDocs);  | 
 | 150 | +            }  | 
 | 151 | +        );  | 
147 | 152 |     }  | 
148 | 153 | 
 
  | 
149 | 154 |     private XContentBuilder updateMapping(int dimensions, String type) throws IOException {  | 
 | 
0 commit comments