chromadb 2.3.0

A Rust client library for the ChromaDB vector database.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

use super::commons::Embedding;
use anyhow::Result;
use async_trait::async_trait;

#[cfg(feature = "openai")]
pub mod openai;

#[async_trait]
pub trait EmbeddingFunction: Send + Sync {
    async fn embed(&self, docs: &[&str]) -> Result<Vec<Embedding>>;
}

#[derive(Clone)]
pub struct MockEmbeddingProvider;

#[async_trait]
impl EmbeddingFunction for MockEmbeddingProvider {
    async fn embed(&self, docs: &[&str]) -> Result<Vec<Embedding>> {
        Ok(docs.iter().map(|_| vec![0.0_f32; 768]).collect())
    }
}