aljaz/lsp-ai
1
1
Fork 0
mirror of https://github.com/SilasMarvin/lsp-ai.git synced 2025-12-17 22:44:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Added todos and cleaned up tests

Browse Source
This commit is contained in:
SilasMarvin
2024-07-03 20:51:32 -07:00
parent c7182fffe7
commit 702ee61b13
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
crates/lsp-ai/src/memory_backends/vector_store.rs
View File

@@ -196,6 +196,7 @@ impl VS {
.par_values() .par_values()
.fold_with(BTreeMap::new(), |mut acc, chunks| { .fold_with(BTreeMap::new(), |mut acc, chunks| {
for chunk in chunks { for chunk in chunks {
// TODO: Review what type of similarity we want to use here
let score = match (&chunk.vec, &embedding) { let score = match (&chunk.vec, &embedding) {
(StoredChunkVec::F32(vec1), StoredChunkVec::F32(vec2)) => { (StoredChunkVec::F32(vec1), StoredChunkVec::F32(vec2)) => {
#[cfg(feature = "simsimd")] #[cfg(feature = "simsimd")]
@@ -204,6 +205,7 @@ impl VS {
SpatialSimilarity::cos(vec1, vec2).unwrap_or(0.) as f32 SpatialSimilarity::cos(vec1, vec2).unwrap_or(0.) as f32
) )
} }
// TODO: Add default similarity check
#[cfg(not(feature = "simsimd"))] #[cfg(not(feature = "simsimd"))]
{ {
0 0
@@ -216,6 +218,7 @@ impl VS {
BinarySimilarity::hamming(vec1, vec2).unwrap_or(0.) as f32, BinarySimilarity::hamming(vec1, vec2).unwrap_or(0.) as f32,
) )
} }
// TODO: Add default similarity check
#[cfg(not(feature = "simsimd"))] #[cfg(not(feature = "simsimd"))]
{ {
0 0
@@ -905,7 +908,7 @@ assert multiply_two_numbers(2, 3) == 6
println!("Insert took {} milliseconds.", elapsed_time.as_millis()); println!("Insert took {} milliseconds.", elapsed_time.as_millis());
// Time search // Time search
let now = std::time::Instant::now(); let now = std::time::Instant::now();
let results = vector_store.search(5, embedding, "", ByteRange::new(0, 0))?; vector_store.search(5, embedding, "", 0)?;
let elapsed_time = now.elapsed(); let elapsed_time = now.elapsed();
println!("Search took {} milliseconds.", elapsed_time.as_millis()); println!("Search took {} milliseconds.", elapsed_time.as_millis());
Ok(()) Ok(())
@@ -940,7 +943,7 @@ assert multiply_two_numbers(2, 3) == 6
println!("Insert took {} milliseconds.", elapsed_time.as_millis()); println!("Insert took {} milliseconds.", elapsed_time.as_millis());
// Time search // Time search
let now = std::time::Instant::now(); let now = std::time::Instant::now();
let results = vector_store.search(5, embedding, "", ByteRange::new(0, 0))?; vector_store.search(5, embedding, "", 0)?;
let elapsed_time = now.elapsed(); let elapsed_time = now.elapsed();
println!("Search took {} milliseconds.", elapsed_time.as_millis()); println!("Search took {} milliseconds.", elapsed_time.as_millis());
Ok(()) Ok(())