feat: brute-force cosine vector index

This commit is contained in:
2026-05-18 18:50:20 +02:00
parent f19bd6e800
commit 34e2f70d91
2 changed files with 42 additions and 1 deletions
+14
View File
@@ -0,0 +1,14 @@
use alpha_id::vector::VectorIndex;
#[test]
fn cosine_topk_ranks_nearest_first() {
let vi = VectorIndex::from_vectors(vec![
vec![1.0, 0.0], // id 0
vec![0.0, 1.0], // id 1
vec![0.9, 0.1], // id 2
]);
let hits = vi.top_k(&[1.0, 0.0], 2);
assert_eq!(hits[0].0, 0);
assert_eq!(hits[1].0, 2);
assert!(hits[0].1 > hits[1].1);
}