Skip to content

Commit 2bd7699

Browse files
committed
Vector similarity needs to wrap the new rescoring query and not the other way round
1 parent 28d929f commit 2bd7699

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2137,6 +2137,7 @@ && isNotUnitVector(squaredMagnitude)) {
21372137
name(),
21382138
queryVector,
21392139
similarity.vectorSimilarityFunction(indexVersionCreated, ElementType.FLOAT),
2140+
k,
21402141
knnQuery
21412142
);
21422143
}

server/src/test/java/org/elasticsearch/search/vectors/RescoreKnnVectorQueryTests.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ public void testRescoreDocs() throws Exception {
7777
FIELD_NAME,
7878
queryVector,
7979
VectorSimilarityFunction.COSINE,
80+
adjustedK,
8081
new MatchAllDocsQuery()
8182
);
8283

@@ -142,6 +143,7 @@ private void checkProfiling(float[] queryVector, IndexReader reader, Query inner
142143
FIELD_NAME,
143144
queryVector,
144145
VectorSimilarityFunction.COSINE,
146+
k,
145147
innerQuery
146148
);
147149
IndexSearcher searcher = newSearcher(reader, true, false);

0 commit comments

Comments
 (0)