kyma-memory 0.0.1

kyma agentic memory: bi-temporal knowledge graph with hybrid (vector + keyword + graph) recall.
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
35
36
37
38

//! Embedding-backend construction. Memory is the first server-side consumer of
//! `kyma-embed`; the backend (an ONNX model for fastembed) is built once and
//! shared process-wide via a `OnceCell` so the model loads at most once.

use std::sync::Arc;

use kyma_embed::{EmbeddingBackend, EmbeddingConfig, EmbeddingProvider};
use tokio::sync::OnceCell;

use crate::error::{MemoryError, Result};

static SHARED: OnceCell<Arc<dyn EmbeddingBackend>> = OnceCell::const_new();

/// The process-wide embedding backend, lazily built from
/// [`EmbeddingConfig::from_env`] on first use.
pub async fn shared_embedding() -> Result<Arc<dyn EmbeddingBackend>> {
    SHARED
        .get_or_try_init(|| async { build_embedding_backend(&EmbeddingConfig::from_env()).await })
        .await
        .map(|b| b.clone())
}

/// Construct an embedding backend from config. Only providers whose
/// `kyma-embed` feature is enabled are available; others return an error.
pub async fn build_embedding_backend(cfg: &EmbeddingConfig) -> Result<Arc<dyn EmbeddingBackend>> {
    match &cfg.provider {
        EmbeddingProvider::Fastembed { id, model_path } => {
            let b = kyma_embed::FastembedBackend::new(id, model_path.as_deref())
                .await
                .map_err(|e| MemoryError::Embed(e.to_string()))?;
            Ok(Arc::new(b))
        }
        other => Err(MemoryError::Embed(format!(
            "embedding provider not enabled in this build: {other:?}; \
             rebuild kyma-memory with the matching kyma-embed feature"
        ))),
    }
}