Fix compilation on lucene_snapshot branch after recent Lucene upgrade.

jpountz · Apr 11, 2024 · 6ca913d · 6ca913d
1 parent f637420
commit 6ca913d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/server/src/main/java/org/elasticsearch/index/codec/vectors/ES813Int8FlatVectorFormat.java b/server/src/main/java/org/elasticsearch/index/codec/vectors/ES813Int8FlatVectorFormat.java
@@ -45,7 +45,9 @@ public ES813Int8FlatVectorFormat() {
      */
     public ES813Int8FlatVectorFormat(Float confidenceInterval) {
         super(NAME);
-        this.format = new Lucene99ScalarQuantizedVectorsFormat(confidenceInterval);
+        int bits = 7;
+        boolean compress = false;
+        this.format = new Lucene99ScalarQuantizedVectorsFormat(confidenceInterval, bits, compress);
     }
 
     @Override

diff --git a/server/src/main/java/org/elasticsearch/index/mapper/vectors/DenseVectorFieldMapper.java b/server/src/main/java/org/elasticsearch/index/mapper/vectors/DenseVectorFieldMapper.java
@@ -996,7 +996,9 @@ private Int8HnswIndexOptions(int m, int efConstruction, Float confidenceInterval
 
         @Override
         public KnnVectorsFormat getVectorsFormat() {
-            return new Lucene99HnswScalarQuantizedVectorsFormat(m, efConstruction, 1, confidenceInterval, null);
+            int bits = 7;
+            boolean compress = false;
+            return new Lucene99HnswScalarQuantizedVectorsFormat(m, efConstruction, 1, bits, compress, confidenceInterval, null);
         }
 
         @Override

diff --git a/server/src/test/java/org/elasticsearch/index/mapper/vectors/DenseVectorFieldMapperTests.java b/server/src/test/java/org/elasticsearch/index/mapper/vectors/DenseVectorFieldMapperTests.java
@@ -1155,6 +1155,7 @@ public void testKnnQuantizedHNSWVectorsFormat() throws IOException {
             + ", flatVectorFormat=Lucene99ScalarQuantizedVectorsFormat("
             + "name=Lucene99ScalarQuantizedVectorsFormat, confidenceInterval="
             + (setConfidenceInterval ? confidenceInterval : null)
+            + ", bits=7, compress=false"
             + ", rawVectorFormat=Lucene99FlatVectorsFormat()"
             + "))";
         assertEquals(expectedString, knnVectorsFormat.toString());