Get max dimensions from codec

Yelp · Mar 1, 2024 · 127d3c1 · 127d3c1
1 parent be9afbb
commit 127d3c1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main/java/com/yelp/nrtsearch/server/luceneserver/field/VectorFieldDef.java b/src/main/java/com/yelp/nrtsearch/server/luceneserver/field/VectorFieldDef.java
@@ -34,7 +34,6 @@
 import org.apache.lucene.index.BinaryDocValues;
 import org.apache.lucene.index.DocValues;
 import org.apache.lucene.index.DocValuesType;
-import org.apache.lucene.index.FloatVectorValues;
 import org.apache.lucene.index.LeafReaderContext;
 import org.apache.lucene.index.VectorSimilarityFunction;
 import org.apache.lucene.util.BytesRef;
@@ -135,9 +134,11 @@ protected void validateRequest(Field requestField) {
     }
 
     if (requestField.getSearch()) {
-      if (requestField.getVectorDimensions() > FloatVectorValues.MAX_DIMENSIONS) {
+      if (requestField.getVectorDimensions() > Lucene95HnswVectorsFormat.DEFAULT_MAX_DIMENSIONS) {
         throw new IllegalArgumentException(
-            "Vector dimension must be <= " + FloatVectorValues.MAX_DIMENSIONS + " for search");
+            "Vector dimension must be <= "
+                + Lucene95HnswVectorsFormat.DEFAULT_MAX_DIMENSIONS
+                + " for search");
       }
     }
   }