From 5322aee018c84e8d0608820f0d02b91099367061 Mon Sep 17 00:00:00 2001 From: Mubelotix Date: Mon, 18 Aug 2025 14:30:20 +0200 Subject: [PATCH] Add cellulite database sizes --- crates/milli/src/index.rs | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/crates/milli/src/index.rs b/crates/milli/src/index.rs index cd68ee118..afca6429b 100644 --- a/crates/milli/src/index.rs +++ b/crates/milli/src/index.rs @@ -1932,6 +1932,16 @@ impl Index { sizes.insert("embedder_category_id", embedder_category_id.stat(rtxn).map(compute_size)?); sizes.insert("documents", documents.stat(rtxn).map(compute_size)?); + // Cellulite + const _CELLULITE_DB_CHECK: () = { + if Cellulite::nb_dbs() != 3 { + panic!("Cellulite database count has changed, please update the code accordingly.") + } + }; + sizes.insert("cellulite_item", self.cellulite.item_db_stats(rtxn).map(compute_size)?); + sizes.insert("cellulite_cell", self.cellulite.cell_db_stats(rtxn).map(compute_size)?); + sizes.insert("cellulite_update", self.cellulite.update_db_stats(rtxn).map(compute_size)?); + Ok(sizes) } }