snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2025-09-05 20:26:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use a version of hannoy with basic filtering

Browse Source
This commit is contained in:
Kerollmops
2025-07-24 12:20:01 +02:00
parent bc1aa5af07
commit 5d22377c2f
3 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
crates/milli/Cargo.toml
View File

@ -87,7 +87,7 @@ rhai = { version = "1.22.2", features = [
"no_time",
"sync",
] }
hannoy = { git = "https://github.com/kerollmops/hannoy", branch = "expose-private-types" }
hannoy = { git = "https://github.com/kerollmops/hannoy", branch = "basic-and-ugly-filtering" }
rand = "0.8.5"
tracing = "0.1.41"
ureq = { version = "2.12.1", features = ["json"] }

10
crates/milli/src/vector/mod.rs
View File

@ -461,13 +461,12 @@ impl HannoyWrapper {
for reader in self.readers(rtxn, db) {
let reader = reader?;
let searcher = reader.nns(limit, limit * 2); // TODO find better ef
let mut searcher = reader.nns(limit, limit * 2); // TODO find better ef
if let Some(filter) = filter {
if reader.item_ids().is_disjoint(filter) {
continue;
}
tracing::error!("Hannoy doesn't support filtering");
// searcher.candidates(filter);
searcher.candidates(filter);
}
if let Some(mut ret) = searcher.by_item(rtxn, item)? {
@ -504,13 +503,12 @@ impl HannoyWrapper {
for reader in self.readers(rtxn, db) {
let reader = reader?;
let searcher = reader.nns(limit, limit * 2); // TODO find better ef
let mut searcher = reader.nns(limit, limit * 2); // TODO find better ef
if let Some(filter) = filter {
if reader.item_ids().is_disjoint(filter) {
continue;
}
tracing::error!("Hannoy doesn't support filtering");
// searcher.candidates(filter);
searcher.candidates(filter);
}
results.append(&mut searcher.by_vector(rtxn, vector)?);