synaptic-memory 0.4.0

Conversation memory: Buffer, Window, Summary, SummaryBuffer, TokenBuffer strategies
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

use std::sync::Arc;

use synaptic_core::{MemoryStore, Message};
use synaptic_memory::{ChatMessageHistory, ConversationBufferMemory};

fn new_store() -> Arc<ChatMessageHistory> {
    Arc::new(ChatMessageHistory::new(Arc::new(
        synaptic_store::InMemoryStore::new(),
    )))
}

#[tokio::test]
async fn buffer_stores_full_conversation() {
    let store = new_store();
    let buffer = ConversationBufferMemory::new(store);

    buffer.append("s1", Message::human("hello")).await.unwrap();
    buffer.append("s1", Message::ai("hi there")).await.unwrap();
    buffer
        .append("s1", Message::human("how are you?"))
        .await
        .unwrap();
    buffer
        .append("s1", Message::ai("doing well"))
        .await
        .unwrap();

    let loaded = buffer.load("s1").await.unwrap();
    assert_eq!(loaded.len(), 4);
    assert_eq!(loaded[0].content(), "hello");
    assert_eq!(loaded[1].content(), "hi there");
    assert_eq!(loaded[2].content(), "how are you?");
    assert_eq!(loaded[3].content(), "doing well");
}

#[tokio::test]
async fn buffer_clear() {
    let store = new_store();
    let buffer = ConversationBufferMemory::new(store);

    buffer.append("s1", Message::human("hello")).await.unwrap();
    buffer.clear("s1").await.unwrap();

    let loaded = buffer.load("s1").await.unwrap();
    assert!(loaded.is_empty());
}