@@ -2199,9 +2199,11 @@ private Query createKnnBitQuery(
21992199 BitSetProducer parentFilter
22002200 ) {
22012201 elementType .checkDimensions (dims , queryVector .length );
2202- Query knnQuery = PatienceKnnVectorQuery .fromByteQuery (parentFilter != null
2203- ? new ESDiversifyingChildrenByteKnnVectorQuery (name (), queryVector , filter , k , numCands , parentFilter )
2204- : new ESKnnByteVectorQuery (name (), queryVector , k , numCands , filter ));
2202+ Query knnQuery = PatienceKnnVectorQuery .fromByteQuery (
2203+ parentFilter != null
2204+ ? new ESDiversifyingChildrenByteKnnVectorQuery (name (), queryVector , filter , k , numCands , parentFilter )
2205+ : new ESKnnByteVectorQuery (name (), queryVector , k , numCands , filter )
2206+ );
22052207 if (similarityThreshold != null ) {
22062208 knnQuery = new VectorSimilarityQuery (
22072209 knnQuery ,
@@ -2226,9 +2228,11 @@ private Query createKnnByteQuery(
22262228 float squaredMagnitude = VectorUtil .dotProduct (queryVector , queryVector );
22272229 elementType .checkVectorMagnitude (similarity , ElementType .errorByteElementsAppender (queryVector ), squaredMagnitude );
22282230 }
2229- Query knnQuery = PatienceKnnVectorQuery .fromByteQuery (parentFilter != null
2230- ? new ESDiversifyingChildrenByteKnnVectorQuery (name (), queryVector , filter , k , numCands , parentFilter )
2231- : new ESKnnByteVectorQuery (name (), queryVector , k , numCands , filter ));
2231+ Query knnQuery = PatienceKnnVectorQuery .fromByteQuery (
2232+ parentFilter != null
2233+ ? new ESDiversifyingChildrenByteKnnVectorQuery (name (), queryVector , filter , k , numCands , parentFilter )
2234+ : new ESKnnByteVectorQuery (name (), queryVector , k , numCands , filter )
2235+ );
22322236 if (similarityThreshold != null ) {
22332237 knnQuery = new VectorSimilarityQuery (
22342238 knnQuery ,
@@ -2279,9 +2283,11 @@ && isNotUnitVector(squaredMagnitude)) {
22792283 adjustedK = Math .min ((int ) Math .ceil (k * oversample ), OVERSAMPLE_LIMIT );
22802284 numCands = Math .max (adjustedK , numCands );
22812285 }
2282- Query knnQuery = PatienceKnnVectorQuery .fromFloatQuery (parentFilter != null
2283- ? new ESDiversifyingChildrenFloatKnnVectorQuery (name (), queryVector , filter , adjustedK , numCands , parentFilter )
2284- : new ESKnnFloatVectorQuery (name (), queryVector , adjustedK , numCands , filter ));
2286+ Query knnQuery = PatienceKnnVectorQuery .fromFloatQuery (
2287+ parentFilter != null
2288+ ? new ESDiversifyingChildrenFloatKnnVectorQuery (name (), queryVector , filter , adjustedK , numCands , parentFilter )
2289+ : new ESKnnFloatVectorQuery (name (), queryVector , adjustedK , numCands , filter )
2290+ );
22852291 if (rescore ) {
22862292 knnQuery = new RescoreKnnVectorQuery (
22872293 name (),
0 commit comments