Skip to content

Commit ccd64ae

Browse files
fix boosting for knn
1 parent 6370d60 commit ccd64ae

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/queries/SemanticKnnVectorQueryRewriteInterceptor.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,8 @@ private KnnVectorQueryBuilder addIndexFilterToKnnVectorQuery(Collection<String>
148148
);
149149
}
150150

151+
copy.boost(original.boost());
152+
copy.queryName(original.queryName());
151153
copy.addFilterQueries(original.filterQueries());
152154
copy.addFilterQuery(new TermsQueryBuilder(IndexFieldMapper.NAME, indices));
153155
return copy;
@@ -187,6 +189,8 @@ private KnnVectorQueryBuilder buildNewKnnVectorQuery(
187189
);
188190
}
189191

192+
newQueryBuilder.boost(original.boost());
193+
newQueryBuilder.queryName(original.queryName());
190194
newQueryBuilder.addFilterQueries(original.filterQueries());
191195
return newQueryBuilder;
192196
}

0 commit comments

Comments
 (0)