cognee-embedding 0.1.0

Embedding-engine abstraction (ONNX, OpenAI, Ollama) for the cognee pipeline.
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

use serde::{Deserialize, Serialize};

/// Selects the backend used to generate embeddings.
///
/// Matches the Python SDK `EMBEDDING_PROVIDER` env var values so that a shared
/// `.env` file works without modification across both SDKs.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
#[serde(rename_all = "snake_case")]
pub enum EmbeddingProvider {
    /// Local ONNX inference (existing engine). Equivalent to Python's "fastembed" provider.
    #[default]
    Onnx,
    /// Alias for Onnx — accepts Python env files using EMBEDDING_PROVIDER=fastembed unchanged.
    /// Deserialized as Onnx internally.
    #[serde(alias = "fastembed")]
    Fastembed,
    /// OpenAI API or any OpenAI-compatible server (llama.cpp, vLLM, TEI, etc.).
    /// In Python, EMBEDDING_PROVIDER=openai uses LiteLLMEmbeddingEngine; in Rust we use a
    /// direct-HTTP engine. Both "openai" and "openai_compatible" map to the same engine.
    #[serde(rename = "openai")]
    OpenAi,
    /// Explicit alias for self-hosted OpenAI-compatible servers.
    /// Identical engine to OpenAi; exists so Python "openai_compatible" env files work unchanged.
    #[serde(rename = "openai_compatible")]
    OpenAiCompatible,
    /// Ollama /api/embed endpoint (not OpenAI-compatible format).
    Ollama,
    /// Zero vectors; for testing. Activated by EMBEDDING_PROVIDER=mock or MOCK_EMBEDDING=true.
    Mock,
}