Skip to content

Commit a0634d4

Browse files
author
elasticsearchmachine
committed
[CI] Auto commit changes from spotless
1 parent e893304 commit a0634d4

File tree

1 file changed

+15
-9
lines changed

1 file changed

+15
-9
lines changed

server/src/main/java/org/elasticsearch/index/mapper/vectors/DenseVectorFieldMapper.java

Lines changed: 15 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)