Skip to content

Commit

Permalink
spotless
Browse files Browse the repository at this point in the history
Signed-off-by: Sahil Buddharaju <[email protected]>
  • Loading branch information
Sahil Buddharaju committed Dec 30, 2024
1 parent 008919a commit 9d01854
Showing 1 changed file with 12 additions and 5 deletions.
17 changes: 12 additions & 5 deletions src/main/java/org/opensearch/knn/index/query/KNNQueryBuilder.java
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down Expand Up @@ -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 {
Expand Down

0 comments on commit 9d01854

Please sign in to comment.