alaya 0.4.8

A memory engine for conversational AI agents, inspired by neuroscience and Buddhist psychology
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

use alaya::{Alaya, EpisodeContext, NewEpisode, NewSemanticNode, Role, SemanticType};

/// Create an empty in-memory store for testing.
pub fn empty_store() -> Alaya {
    Alaya::open_in_memory().unwrap()
}

/// Create a store populated with sample episodes.
#[allow(dead_code)]
pub fn populated_store() -> Alaya {
    let store = empty_store();
    for i in 0..5 {
        store
            .episodes()
            .store(&make_episode(
                &format!("Test message {i}"),
                "user",
                "session-1",
                1700000000 + i * 60,
            ))
            .unwrap();
    }
    store
}

/// Create a store with a specific number of episodes.
#[allow(dead_code)]
pub fn store_with_episodes(n: i64) -> Alaya {
    let store = empty_store();
    for i in 0..n {
        store
            .episodes()
            .store(&make_episode(
                &format!("Episode {i}"),
                "user",
                "session-1",
                1700000000 + i * 60,
            ))
            .unwrap();
    }
    store
}

/// Build a NewEpisode with sensible defaults.
#[allow(dead_code)]
pub fn make_episode(content: &str, role: &str, session_id: &str, timestamp: i64) -> NewEpisode {
    let role = match role {
        "assistant" => Role::Assistant,
        "system" => Role::System,
        _ => Role::User,
    };
    NewEpisode {
        content: content.to_string(),
        role,
        session_id: session_id.to_string(),
        timestamp,
        context: EpisodeContext::default(),
        embedding: None,
    }
}

/// Build a NewSemanticNode with sensible defaults.
#[allow(dead_code)]
pub fn make_semantic_node(content: &str, node_type: SemanticType) -> NewSemanticNode {
    NewSemanticNode {
        content: content.to_string(),
        node_type,
        confidence: 0.95,
        source_episodes: vec![],
        embedding: None,
    }
}