Remove the Vector variant and use the Vectors

This commit is contained in:
Clément Renault 2024-11-28 15:19:57 +01:00
parent 096a28656e
commit b57dd5c58e
No known key found for this signature in database
GPG key ID: F250A4C4E3AE5F5F
2 changed files with 4 additions and 141 deletions

View file

@ -16,7 +16,6 @@ use rand::SeedableRng as _;
use raw_collections::RawMap;
use time::OffsetDateTime;
pub use update_by_function::UpdateByFunction;
use {LargeEntry, LargeVector};
use super::channel::*;
use super::extract::*;
@ -430,14 +429,6 @@ where
}));
}
}
ReceiverAction::LargeVector(large_vector) => {
let embedding = large_vector.read_embedding();
let LargeVector { docid, embedder_id, .. } = large_vector;
let (_, _, writer, dimensions) =
arroy_writers.get(&embedder_id).expect("requested a missing embedder");
writer.del_items(wtxn, *dimensions, docid)?;
writer.add_item(wtxn, docid, embedding)?;
}
ReceiverAction::LargeVectors(large_vectors) => {
let LargeVectors { docid, embedder_id, .. } = large_vectors;
let (_, _, writer, dimensions) =
@ -594,16 +585,6 @@ fn write_from_bbqueue(
writer.del_items(wtxn, dimensions, docid)?;
}
}
EntryHeader::ArroySetVector(asv) => {
let ArroySetVector { docid, embedder_id, .. } = asv;
let frame = frame_with_header.frame();
let (_, _, writer, dimensions) =
arroy_writers.get(&embedder_id).expect("requested a missing embedder");
writer.del_items(wtxn, *dimensions, docid)?;
if let Some(embedding) = asv.read_embedding_into_vec(frame, aligned_embedding) {
writer.add_item(wtxn, docid, embedding)?;
}
}
EntryHeader::ArroySetVectors(asvs) => {
let ArroySetVectors { docid, embedder_id, .. } = asvs;
let frame = frame_with_header.frame();