h/silos
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: v1 api: better error handling

Browse Source
This commit is contained in:
Himadri Bhattacharjee
2025-07-02 09:53:57 +05:30
parent cae538f27b
commit 5dadaca767
1 changed files with 10 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/v1/api.rs
View File

@@ -1,4 +1,4 @@
use hora::core::ann_index::ANNIndex;
use hora::core::{ann_index::ANNIndex, metrics::Metric::Euclidean};
use std::collections::HashMap;
use super::errors::Error;
@@ -63,21 +63,19 @@ pub(crate) async fn add_snippet(
let Ok(mut appstate) = data.inner.lock() else {
return Err(Error::Busy);
};
let Ok(embedding) = appstate.embed.embed(&snippet.desc) else {
return Err(Error::EmbedFailed);
};
let embedding = appstate
.embed
.embed(&snippet.desc)
.map_err(|_| Error::EmbedFailed)?;
let index = appstate
.v1
.dict
.entry(snippet.lang.clone())
.or_insert_with(|| {
let dimension = 384;
let params = hora::index::hnsw_params::HNSWParams::<f32>::default();
HNSWIndex::<f32, String>::new(dimension, &params)
});
index.add(&embedding, snippet.body.clone()).unwrap();
index.build(hora::core::metrics::Metric::Euclidean).unwrap();
.or_insert_with(|| HNSWIndex::<f32, String>::new(384, &Default::default()));
index
.add(&embedding, snippet.body.clone())
.map_err(|_| Error::EmbedFailed)?;
index.build(Euclidean).map_err(|_| Error::EmbedFailed)?;
Ok(format!(
"{} {} {}",