diff --git a/Cargo.lock b/Cargo.lock index 3a5e02eb2..b77c2ced3 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -2603,8 +2603,8 @@ dependencies = [ [[package]] name = "hannoy" -version = "0.0.1" -source = "git+https://github.com/Kerollmops/hannoy?branch=release-0.0.2#d734d5ab13ae34b3ee4d34b9eeb3ed8843cd11c2" +version = "0.0.2" +source = "git+https://github.com/nnethercott/hannoy?rev=cab22c0#cab22c01dce4c825ae4d0d8824ba51ed4cef57e1" dependencies = [ "bytemuck", "byteorder", diff --git a/crates/milli/Cargo.toml b/crates/milli/Cargo.toml index 3ff7cb323..12dacf1b1 100644 --- a/crates/milli/Cargo.toml +++ b/crates/milli/Cargo.toml @@ -88,7 +88,7 @@ rhai = { version = "1.22.2", features = [ "sync", ] } arroy = "0.6.1" -hannoy = { git = "https://github.com/Kerollmops/hannoy", branch = "release-0.0.2" } +hannoy = { git = "https://github.com/nnethercott/hannoy", rev = "cab22c0" } rand = "0.8.5" tracing = "0.1.41" ureq = { version = "2.12.1", features = ["json"] } diff --git a/crates/milli/src/vector/mod.rs b/crates/milli/src/vector/mod.rs index 6c016eccd..5ff4f658e 100644 --- a/crates/milli/src/vector/mod.rs +++ b/crates/milli/src/vector/mod.rs @@ -468,7 +468,8 @@ impl VectorStore { for reader in self.readers(rtxn, db) { let reader = reader?; - let mut searcher = reader.nns(limit, limit * 10); // TODO find better ef + let mut searcher = reader.nns(limit); + searcher.ef_search(limit * 10); // TODO find better ef if let Some(filter) = filter { if reader.item_ids().is_disjoint(filter) { continue; @@ -510,7 +511,8 @@ impl VectorStore { for reader in self.readers(rtxn, db) { let reader = reader?; - let mut searcher = reader.nns(limit, limit * 10); // TODO find better ef + let mut searcher = reader.nns(limit); + searcher.ef_search(limit * 10); // TODO find better ef if let Some(filter) = filter { if reader.item_ids().is_disjoint(filter) { continue;