snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2025-07-27 16:51:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Send owned metadata and clear inputs in case of error

Browse Source
This commit is contained in:
Louis Dureuil
2025-07-03 10:32:57 +02:00
parent 3740755d9c
commit 735634e998
3 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
crates/milli/src/vector/session.rs
View File

@ -28,7 +28,7 @@ pub trait OnEmbed<'doc> {
error: EmbedError,
embedder_name: &'doc str,
unused_vectors_distribution: &Self::ErrorMetadata,
metadata: &[Metadata<'doc>],
metadata: BVec<'doc, Metadata<'doc>>,
) -> crate::Error;
}
@ -143,12 +143,19 @@ impl<'doc, C: OnEmbed<'doc>, I: Input> EmbedSession<'doc, C, I> {
Ok(())
}
Err(error) => {
// reset metadata and inputs, and send metadata to the error processing.
let doc_alloc = self.metadata.bump();
let metadata = std::mem::replace(
&mut self.metadata,
BVec::with_capacity_in(self.inputs.capacity(), doc_alloc),
);
self.inputs.clear();
return Err(self.on_embed.process_embedding_error(
error,
self.embedder_name,
unused_vectors_distribution,
&self.metadata,
))
metadata,
));
}
};
self.inputs.clear();