harald/zeroclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

feat(tools): expose custom memory categories in memory_store tool

Browse source 
The MemoryCategory::Custom variant already exists in the memory backend
but the memory_store tool only accepted core/daily/conversation. Now any
string is accepted as a category, passing through to Custom(name) for
non-builtin values.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Edvard 2026-02-17 18:40:01 -05:00 • committed by Chummy
parent b2993d68c4
commit c04f2855e4
1 changed files with 21 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

25
src/tools/memory_store.rs
View file

@ -22,7 +22,7 @@ impl Tool for MemoryStoreTool {
} }
fn description(&self) -> &str { fn description(&self) -> &str {
"Store a fact, preference, or note in long-term memory. Use category 'core' for permanent facts, 'daily' for session notes, 'conversation' for chat context." "Store a fact, preference, or note in long-term memory. Use category 'core' for permanent facts, 'daily' for session notes, 'conversation' for chat context, or a custom category name."
} }
fn parameters_schema(&self) -> serde_json::Value { fn parameters_schema(&self) -> serde_json::Value {
@ -39,8 +39,7 @@ impl Tool for MemoryStoreTool {
}, },
"category": { "category": {
"type": "string", "type": "string",
"enum": ["core", "daily", "conversation"], "description": "Memory category: 'core' (permanent), 'daily' (session), 'conversation' (chat), or a custom category name. Defaults to 'core'."
"description": "Memory category: core (permanent), daily (session), conversation (chat)"
} }
}, },
"required": ["key", "content"] "required": ["key", "content"]
@ -59,9 +58,10 @@ impl Tool for MemoryStoreTool {
.ok_or_else(|| anyhow::anyhow!("Missing 'content' parameter"))?; .ok_or_else(|| anyhow::anyhow!("Missing 'content' parameter"))?;
let category = match args.get("category").and_then(|v| v.as_str()) { let category = match args.get("category").and_then(|v| v.as_str()) {
Some("core") | None => MemoryCategory::Core,
Some("daily") => MemoryCategory::Daily, Some("daily") => MemoryCategory::Daily,
Some("conversation") => MemoryCategory::Conversation, Some("conversation") => MemoryCategory::Conversation,
_ => MemoryCategory::Core, Some(other) => MemoryCategory::Custom(other.to_string()),
}; };
match self.memory.store(key, content, category, None).await { match self.memory.store(key, content, category, None).await {
@ -128,6 +128,23 @@ mod tests {
assert!(result.success); assert!(result.success);
} }
#[tokio::test]
async fn store_with_custom_category() {
let (_tmp, mem) = test_mem();
let tool = MemoryStoreTool::new(mem.clone());
let result = tool
.execute(
json!({"key": "proj_note", "content": "Uses async runtime", "category": "project"}),
)
.await
.unwrap();
assert!(result.success);
let entry = mem.get("proj_note").await.unwrap().unwrap();
assert_eq!(entry.content, "Uses async runtime");
assert_eq!(entry.category, MemoryCategory::Custom("project".into()));
}
#[tokio::test] #[tokio::test]
async fn store_missing_key() { async fn store_missing_key() {
let (_tmp, mem) = test_mem(); let (_tmp, mem) = test_mem();