TerribleDev/ente
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

vector db api ensure never duplicate keys

Browse Source
This commit is contained in:
laurenspriem
2025-05-08 14:30:51 +05:30
parent d87069eb4c
commit 6478d438b5
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
mobile/rust/src/api/usearch_api.rs
View File

@@ -61,7 +61,11 @@ impl VectorDB {
}
pub fn add_vector(&self, key: u64, vector: &Vec<f32>) {
self.ensure_capacity(1);
if self.contains_vector(key) {
self.remove_vector(key);
} else {
self.ensure_capacity(1);
}
self.index.add(key, vector).expect("Failed to add vector");
self.save_index();
}
@@ -69,6 +73,9 @@ impl VectorDB {
pub fn bulk_add_vectors(&self, keys: Vec<u64>, vectors: &Vec<Vec<f32>>) {
self.ensure_capacity(keys.len());
for (key, vector) in keys.iter().zip(vectors.iter()) {
if self.contains_vector(*key) {
self.remove_vector(*key);
}
self.index
.add(*key, vector)
.expect("Failed to (bulk) add vector");