|
23 | 23 | import org.apache.lucene.analysis.standard.StandardAnalyzer; |
24 | 24 | import org.apache.lucene.document.Document; |
25 | 25 | import org.apache.lucene.document.Field; |
26 | | -import org.apache.lucene.document.KnnVectorField; |
| 26 | +import org.apache.lucene.document.KnnFloatVectorField; |
27 | 27 | import org.apache.lucene.document.StoredField; |
28 | 28 | import org.apache.lucene.document.TextField; |
29 | 29 | import org.apache.lucene.index.DirectoryReader; |
30 | 30 | import org.apache.lucene.index.IndexReader; |
31 | 31 | import org.apache.lucene.index.IndexWriter; |
32 | 32 | import org.apache.lucene.index.IndexWriterConfig; |
33 | 33 | import org.apache.lucene.search.IndexSearcher; |
34 | | -import org.apache.lucene.search.KnnVectorQuery; |
| 34 | +import org.apache.lucene.search.KnnFloatVectorQuery; |
35 | 35 | import org.apache.lucene.search.TopDocs; |
36 | 36 | import org.apache.lucene.store.ByteBuffersDirectory; |
37 | 37 | import org.apache.lucene.store.Directory; |
@@ -107,7 +107,7 @@ public void addVectorIndex(boolean isVertex, K key, String fieldName, float[] ve |
107 | 107 | } |
108 | 108 |
|
109 | 109 | // Add vector field |
110 | | - doc.add(new KnnVectorField(fieldName, vector)); |
| 110 | + doc.add(new KnnFloatVectorField(fieldName, vector)); |
111 | 111 |
|
112 | 112 | // Add document to index |
113 | 113 | writer.addDocument(doc); |
@@ -140,12 +140,12 @@ public K searchVectorIndex(boolean isVertex, String fieldName, float[] vector, i |
140 | 140 | IndexSearcher searcher = new IndexSearcher(reader); |
141 | 141 |
|
142 | 142 | // Create KNN vector query |
143 | | - KnnVectorQuery knnQuery = new KnnVectorQuery(fieldName, vector, topK); |
| 143 | + KnnFloatVectorQuery knnQuery = new KnnFloatVectorQuery(fieldName, vector, topK); |
144 | 144 |
|
145 | 145 | // Execute search |
146 | 146 | TopDocs topDocs = searcher.search(knnQuery, topK); |
147 | 147 |
|
148 | | - Document firstDoc = searcher.doc(topDocs.scoreDocs[0].doc); |
| 148 | + Document firstDoc = searcher.storedFields().document(topDocs.scoreDocs[0].doc); |
149 | 149 |
|
150 | 150 | K result; |
151 | 151 | if (keyClass == String.class) { |
|
0 commit comments