spire-ai 0.1.3

AI-native SDK for SpireDB — RAG, code search, agents
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

//! Remember tool — stores information in agent memory.

use async_trait::async_trait;
use serde_json::Value;

use crate::agent::AgentMemory;
use crate::error::Result;
use crate::llm::types::ToolDef;
use crate::tool::{Tool, ToolResult};

/// Store information in persistent agent memory.
pub struct RememberTool {
    memory: AgentMemory,
}

impl RememberTool {
    pub fn new(memory: AgentMemory) -> Self {
        Self { memory }
    }
}

#[async_trait]
impl Tool for RememberTool {
    fn name(&self) -> &str {
        "remember"
    }

    fn def(&self) -> ToolDef {
        ToolDef {
            name: "remember".into(),
            description: "Store important information in persistent memory for later recall."
                .into(),
            parameters: serde_json::json!({
                "type": "object",
                "properties": {
                    "content": {
                        "type": "string",
                        "description": "The information to remember"
                    },
                    "importance": {
                        "type": "string",
                        "enum": ["low", "normal", "high", "critical"],
                        "description": "Importance level (default: normal)"
                    }
                },
                "required": ["content"]
            }),
        }
    }

    async fn call(&self, args: Value) -> Result<ToolResult> {
        let content = args["content"]
            .as_str()
            .ok_or_else(|| crate::error::Error::Tool("remember: missing 'content'".into()))?;

        let importance = match args["importance"].as_str() {
            Some("low") => crate::agent::Importance::Low,
            Some("high") => crate::agent::Importance::High,
            Some("critical") => crate::agent::Importance::Critical,
            _ => crate::agent::Importance::Normal,
        };

        let id = self.memory.remember_with(content, importance).await?;
        Ok(ToolResult::ok(format!("Remembered (id: {id})")))
    }
}