jVector changes are backwards compatible with integ tests passing

opensearch-project · Feb 7, 2025 · 3e86bfc · 3e86bfc
1 parent efdc583
commit 3e86bfc
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/src/main/java/org/opensearch/knn/index/codec/jvector/JVectorReader.java b/src/main/java/org/opensearch/knn/index/codec/jvector/JVectorReader.java
@@ -179,9 +179,24 @@ public FieldEntry(
             final Path indexFilePath;
             final long sliceOffset;
             if (state.segmentInfo.getUseCompoundFile()) {
-                JVectorCompoundFormat.JVectorCompoundReader jVectorCompoundReader = (JVectorCompoundFormat.JVectorCompoundReader) directory;
-                sliceOffset = jVectorCompoundReader.getOffsetInCompoundFile(originalIndexFileName);
-                indexFilePath = jVectorCompoundReader.getCompoundFilePath();
+                if (directory instanceof JVectorCompoundFormat.JVectorCompoundReader) {
+                    JVectorCompoundFormat.JVectorCompoundReader jVectorCompoundReader = (JVectorCompoundFormat.JVectorCompoundReader) directory;
+                    sliceOffset = jVectorCompoundReader.getOffsetInCompoundFile(originalIndexFileName);
+                    indexFilePath = jVectorCompoundReader.getCompoundFilePath();
+                } else if (directory instanceof KNN80CompoundDirectory) {
+                    KNN80CompoundDirectory knn80CompoundDirectory = (KNN80CompoundDirectory) directory;
+                    JVectorCompoundFormat.JVectorCompoundReader jVectorCompoundReader = new JVectorCompoundFormat.JVectorCompoundReader(
+                            knn80CompoundDirectory.getDelegate(),
+                            knn80CompoundDirectory.getDir(),
+                            state.segmentInfo,
+                            state.context
+                    );
+                    sliceOffset = jVectorCompoundReader.getOffsetInCompoundFile(originalIndexFileName);
+                    indexFilePath = jVectorCompoundReader.getCompoundFilePath();
+                } else {
+                    throw new IllegalArgumentException("directory must be JVectorCompoundFormat or KNN80CompoundDirectory but instead had type: " + directory.getClass().getName());
+                }
+
             } else {
                 sliceOffset = 0;
                 indexFilePath = expectedIndexFilePath;