do-memory-core 0.1.31

Core episodic learning system for AI agents with pattern extraction, reward scoring, and dual storage backend
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

//! Supported embedding providers

use serde::{Deserialize, Serialize};

/// Supported embedding providers
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub enum EmbeddingProvider {
    /// Local embedding using sentence transformers
    Local,
    /// `OpenAI`'s text embedding models
    OpenAI,
    /// Mistral AI's embedding models
    Mistral,
    /// Azure `OpenAI` Service
    AzureOpenAI,
    /// Custom provider implementation
    Custom(String),
}

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

    #[test]
    fn test_embedding_provider_variants() {
        // Test Local provider
        let local = EmbeddingProvider::Local;
        assert_eq!(local, EmbeddingProvider::Local);

        // Test OpenAI provider
        let openai = EmbeddingProvider::OpenAI;
        assert_eq!(openai, EmbeddingProvider::OpenAI);

        // Test Mistral provider
        let mistral = EmbeddingProvider::Mistral;
        assert_eq!(mistral, EmbeddingProvider::Mistral);

        // Test Azure OpenAI provider
        let azure = EmbeddingProvider::AzureOpenAI;
        assert_eq!(azure, EmbeddingProvider::AzureOpenAI);

        // Test Custom provider
        let custom = EmbeddingProvider::Custom("custom-provider".to_string());
        assert_eq!(
            custom,
            EmbeddingProvider::Custom("custom-provider".to_string())
        );

        // Test equality/inequality
        assert_ne!(local, openai);
        assert_ne!(openai, mistral);
    }
}