diff --git a/src/main/java/org/opensearch/knn/index/query/KNNQueryBuilder.java b/src/main/java/org/opensearch/knn/index/query/KNNQueryBuilder.java index 1d706e0984..62400a5093 100644 --- a/src/main/java/org/opensearch/knn/index/query/KNNQueryBuilder.java +++ b/src/main/java/org/opensearch/knn/index/query/KNNQueryBuilder.java @@ -23,7 +23,6 @@ import org.opensearch.index.query.QueryBuilder; import org.opensearch.index.query.QueryRewriteContext; import org.opensearch.index.query.QueryShardContext; -import org.opensearch.index.query.RangeQueryBuilder; import org.opensearch.knn.index.engine.KNNMethodConfigContext; import org.opensearch.knn.index.engine.model.QueryContext; import org.opensearch.knn.index.engine.qframe.QuantizationConfig; @@ -661,22 +660,30 @@ public String getWriteableName() { return NAME; } - @Override protected QueryBuilder doRewrite(QueryRewriteContext queryShardContext) throws IOException { QueryBuilder rewrittenFilter; if (Objects.nonNull(filter)) { rewrittenFilter = filter.rewrite(queryShardContext); if (rewrittenFilter != filter) { - KNNQueryBuilder newKNNQuery = new KNNQueryBuilder(this.fieldName, this.vector, this.k, this.maxDistance, this.minScore, - this.methodParameters, rewrittenFilter, this.ignoreUnmapped, this.rescoreContext, this.expandNested); + KNNQueryBuilder newKNNQuery = new KNNQueryBuilder( + this.fieldName, + this.vector, + this.k, + this.maxDistance, + this.minScore, + this.methodParameters, + rewrittenFilter, + this.ignoreUnmapped, + this.rescoreContext, + this.expandNested + ); return newKNNQuery; } } return super.doRewrite(queryShardContext); } - @Getter @AllArgsConstructor private static class QueryConfigFromMapping {