fix: populate index cache at the end of loading (#2123)

lancedb · Mar 28, 2024 · 7e114ee · 7e114ee
1 parent 951a95e
commit 7e114ee
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/rust/lance/src/index.rs b/rust/lance/src/index.rs
@@ -482,6 +482,7 @@ impl DatasetIndexInternalExt for Dataset {
 
     async fn open_vector_index(&self, column: &str, uuid: &str) -> Result<Arc<dyn VectorIndex>> {
         if let Some(index) = self.session.index_cache.get_vector(uuid) {
+            log::debug!("Found vector index in cache uuid: {}", uuid);
             return Ok(index);
         }
 

diff --git a/rust/lance/src/index/vector.rs b/rust/lance/src/index/vector.rs
@@ -439,5 +439,10 @@ pub(crate) async fn open_vector_index_v2(
         }
     };
 
+    dataset
+        .session
+        .index_cache
+        .insert_vector(uuid, index.clone());
+
     Ok(index)
 }