"Refactor loadIndex method to directly assign loaded index items to i…

…ndexItems and print the count and path of the loaded items."
ZachNagengast · Feb 14, 2024 · b8d134a · b8d134a
1 parent eef5ab0
commit b8d134a
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/Sources/SimilaritySearchKit/Core/Index/SimilarityIndex.swift b/Sources/SimilaritySearchKit/Core/Index/SimilarityIndex.swift
@@ -326,11 +326,9 @@ extension SimilarityIndex {
 
     public func loadIndex(fromDirectory path: URL? = nil, name: String? = nil) throws -> [IndexItem]? {
         if let indexPath = try getIndexPath(fromDirectory: path, name: name) {
-            let loadedIndexItems = try vectorStore.loadIndex(from: indexPath)
-            addItems(loadedIndexItems) {[self] in
-                print("Loaded \(indexItems.count) index items from \(indexPath.absoluteString)")
-            }
-            return loadedIndexItems
+            indexItems = try vectorStore.loadIndex(from: indexPath)
+            print("Loaded \(indexItems.count) index items from \(indexPath.absoluteString)")
+            return indexItems
         }
 
         return nil