add cellulite to the index

This commit is contained in:
Tamo
2025-07-15 23:14:06 +02:00
parent df2e7cde53
commit 73eb64242d
3 changed files with 17 additions and 1 deletions

View File

@ -114,6 +114,7 @@ pub mod db_name {
pub const FIELD_ID_DOCID_FACET_STRINGS: &str = "field-id-docid-facet-strings";
pub const VECTOR_EMBEDDER_CATEGORY_ID: &str = "vector-embedder-category-id";
pub const VECTOR_ARROY: &str = "vector-arroy";
pub const CELLULITE: &str = "cellulite";
pub const DOCUMENTS: &str = "documents";
}
const NUMBER_OF_DBS: u32 = 25;
@ -182,6 +183,9 @@ pub struct Index {
/// Vector store based on arroy™.
pub vector_arroy: arroy::Database<Unspecified>,
/// Geo store based on cellulite™.
pub cellulite: cellulite::Database,
/// Maps the document id to the document as an obkv store.
pub(crate) documents: Database<BEU32, ObkvCodec>,
}
@ -238,6 +242,7 @@ impl Index {
let embedder_category_id =
env.create_database(&mut wtxn, Some(VECTOR_EMBEDDER_CATEGORY_ID))?;
let vector_arroy = env.create_database(&mut wtxn, Some(VECTOR_ARROY))?;
let cellulite = env.create_database(&mut wtxn, Some(CELLULITE))?;
let documents = env.create_database(&mut wtxn, Some(DOCUMENTS))?;
@ -266,6 +271,7 @@ impl Index {
field_id_docid_facet_strings,
vector_arroy,
embedder_category_id,
cellulite,
documents,
};
if this.get_version(&wtxn)?.is_none() && creation {
@ -1023,6 +1029,13 @@ impl Index {
Ok(geo_filter)
}
/// Returns true if the geo sorting feature is enabled.
pub fn is_geojson_enabled(&self, rtxn: &RoTxn<'_>) -> Result<bool> {
let geojson_filter =
self.filterable_attributes_rules(rtxn)?.iter().any(|field| field.has_geojson());
Ok(geojson_filter)
}
pub fn asc_desc_fields(&self, rtxn: &RoTxn<'_>) -> Result<HashSet<String>> {
let asc_desc_fields = self
.criteria(rtxn)?
@ -1842,6 +1855,7 @@ impl Index {
field_id_docid_facet_strings,
vector_arroy,
embedder_category_id,
cellulite: _,
documents,
} = self;