Skip to main content

synaptic_embeddings/
lib.rs

1mod cached;
2mod fake;
3mod ollama;
4mod openai;
5
6pub use cached::CacheBackedEmbeddings;
7pub use fake::FakeEmbeddings;
8pub use ollama::{OllamaEmbeddings, OllamaEmbeddingsConfig};
9pub use openai::{OpenAiEmbeddings, OpenAiEmbeddingsConfig};
10
11use async_trait::async_trait;
12use synaptic_core::SynapseError;
13
14/// Trait for embedding text into vectors.
15#[async_trait]
16pub trait Embeddings: Send + Sync {
17    /// Embed multiple texts (for batch document embedding).
18    async fn embed_documents(&self, texts: &[&str]) -> Result<Vec<Vec<f32>>, SynapseError>;
19
20    /// Embed a single query text.
21    async fn embed_query(&self, text: &str) -> Result<Vec<f32>, SynapseError>;
22}