asobi 0.4.1

A persistent, project-local knowledge graph CLI for AI agents.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

use anyhow::Result;

#[allow(async_fn_in_trait)]
pub trait EmbeddingProvider: Send + Sync {
    async fn embed(&self, texts: &[String]) -> Result<Vec<Vec<f32>>>;
    fn dim(&self) -> usize;
}

pub mod fastembed_provider;
pub use fastembed_provider::FastEmbedProvider;

#[cfg(test)]
mod tests {
    use super::*;

    struct DummyProvider;

    impl EmbeddingProvider for DummyProvider {
        async fn embed(&self, texts: &[String]) -> Result<Vec<Vec<f32>>> {
            Ok(texts.iter().map(|_| vec![0.0f32; 4]).collect())
        }
        fn dim(&self) -> usize {
            4
        }
    }

    #[tokio::test]
    async fn test_provider_trait() {
        let p = DummyProvider;
        let result = p.embed(&["hello".to_string()]).await.unwrap();
        assert_eq!(result.len(), 1);
        assert_eq!(result[0].len(), 4);
    }
}