rskit-agent 0.2.0-alpha.1

Agentic loop — Provider + Tools + Hooks in a turn-based execution engine
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

use std::collections::HashMap;

use async_trait::async_trait;
use parking_lot::RwLock;
use rskit_errors::AppError;
use rskit_llm::types::Message;

use super::Memory;

// ── InMemoryStore ───────────────────────────────────────────────────────────

/// A simple in-process memory store backed by a `RwLock<HashMap>`.
#[derive(Debug, Default)]
pub struct InMemoryStore {
    store: RwLock<HashMap<String, Vec<Message>>>,
}

impl InMemoryStore {
    /// Create an empty in-memory message store.
    pub fn new() -> Self {
        Self::default()
    }
}

#[async_trait]
impl Memory for InMemoryStore {
    async fn load(&self, session_id: &str) -> Result<Vec<Message>, AppError> {
        let guard = self.store.read();
        Ok(guard.get(session_id).cloned().unwrap_or_default())
    }

    async fn save(&self, session_id: &str, messages: &[Message]) -> Result<(), AppError> {
        let mut guard = self.store.write();
        guard.insert(session_id.to_string(), messages.to_vec());
        drop(guard);
        Ok(())
    }

    async fn append(&self, session_id: &str, messages: &[Message]) -> Result<(), AppError> {
        let mut guard = self.store.write();
        guard
            .entry(session_id.to_string())
            .or_default()
            .extend_from_slice(messages);
        drop(guard);
        Ok(())
    }

    async fn clear(&self, session_id: &str) -> Result<(), AppError> {
        let mut guard = self.store.write();
        guard.remove(session_id);
        drop(guard);
        Ok(())
    }
}