engram-core 0.19.0

AI Memory Infrastructure - Persistent memory for AI agents with semantic search
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

//! Workspace tool handlers.

use serde_json::{json, Value};

use super::HandlerContext;

pub fn workspace_list(ctx: &HandlerContext, _params: Value) -> Value {
    use crate::storage::queries::list_workspaces;

    ctx.storage
        .with_connection(|conn| {
            let workspaces = list_workspaces(conn)?;
            Ok(json!({"count": workspaces.len(), "workspaces": workspaces}))
        })
        .unwrap_or_else(|e| json!({"error": e.to_string()}))
}

pub fn workspace_stats(ctx: &HandlerContext, params: Value) -> Value {
    use crate::storage::queries::get_workspace_stats;

    let workspace = match params.get("workspace").and_then(|v| v.as_str()) {
        Some(ws) => ws,
        None => return json!({"error": "workspace is required"}),
    };

    ctx.storage
        .with_connection(|conn| {
            let stats = get_workspace_stats(conn, workspace)?;
            Ok(json!(stats))
        })
        .unwrap_or_else(|e| json!({"error": e.to_string()}))
}

pub fn workspace_move(ctx: &HandlerContext, params: Value) -> Value {
    use crate::storage::queries::move_to_workspace;

    let id = match params.get("id").and_then(|v| v.as_i64()) {
        Some(id) => id,
        None => return json!({"error": "id is required"}),
    };
    let workspace = match params.get("workspace").and_then(|v| v.as_str()) {
        Some(ws) => ws,
        None => return json!({"error": "workspace is required"}),
    };

    ctx.storage
        .with_connection(|conn| {
            let memory = move_to_workspace(conn, id, workspace)?;
            Ok(json!({"success": true, "memory": memory}))
        })
        .unwrap_or_else(|e| json!({"error": e.to_string()}))
}

pub fn workspace_delete(ctx: &HandlerContext, params: Value) -> Value {
    use crate::storage::queries::delete_workspace;

    let workspace = match params.get("workspace").and_then(|v| v.as_str()) {
        Some(ws) => ws,
        None => return json!({"error": "workspace is required"}),
    };
    let move_to_default = params
        .get("move_to_default")
        .and_then(|v| v.as_bool())
        .unwrap_or(true);

    ctx.storage
        .with_connection(|conn| {
            let affected = delete_workspace(conn, workspace, move_to_default)?;
            Ok(json!({"success": true, "affected": affected, "move_to_default": move_to_default}))
        })
        .unwrap_or_else(|e| json!({"error": e.to_string()}))
}