snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-22 10:07:40 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

documents! macro accepts a single object again

Browse Source
This commit is contained in:
Louis Dureuil 2024-11-19 12:02:26 +01:00 committed by Clément Renault
parent 32d0e50a75
commit 1aef0e4037
No known key found for this signature in database
GPG Key ID: F250A4C4E3AE5F5F
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
crates/milli/src/documents/mod.rs
View File

@ -151,8 +151,17 @@ macro_rules! documents {
($data:tt) => {{ ($data:tt) => {{
let documents = serde_json::json!($data); let documents = serde_json::json!($data);
let mut file = tempfile::tempfile().unwrap(); let mut file = tempfile::tempfile().unwrap();
for document in documents.as_array().unwrap() {
serde_json::to_writer(&mut file, &document).unwrap(); match documents {
serde_json::Value::Array(vec) => {
for document in vec {
serde_json::to_writer(&mut file, &document).unwrap();
}
}
serde_json::Value::Object(document) => {
serde_json::to_writer(&mut file, &document).unwrap();
}
_ => unimplemented!("The `documents!` macro only support Objects and Array"),
} }
file.sync_all().unwrap(); file.sync_all().unwrap();
unsafe { memmap2::Mmap::map(&file).unwrap() } unsafe { memmap2::Mmap::map(&file).unwrap() }

3
crates/milli/src/index.rs
View File

@ -1749,7 +1749,8 @@ pub(crate) mod tests {
let db_fields_ids_map = self.inner.fields_ids_map(&rtxn)?; let db_fields_ids_map = self.inner.fields_ids_map(&rtxn)?;
let mut new_fields_ids_map = db_fields_ids_map.clone(); let mut new_fields_ids_map = db_fields_ids_map.clone();
let embedders = EmbeddingConfigs::default(); /// TODO: fetch configs from the index let embedders = EmbeddingConfigs::default();
/// TODO: fetch configs from the index
let mut indexer = let mut indexer =
indexer::DocumentOperation::new(IndexDocumentsMethod::ReplaceDocuments); indexer::DocumentOperation::new(IndexDocumentsMethod::ReplaceDocuments);
indexer.add_documents(&documents).unwrap(); indexer.add_documents(&documents).unwrap();