kernl 0.7.3 → 0.8.0

package/src/tool/sys/memory.ts

@@ -11,13 +11,11 @@ import { z } from "zod";
 import { tool } from "../tool";
 import { Toolkit } from "../toolkit";
 
-// --- Tools ---
-
 /**
  * Search memories for relevant information using natural language.
  */
 const search = tool({
-  id: "memories.search",
+  id: "search_memories",
   description:
     "Search your memories. " +
     "Use this to recall facts, preferences, or context you've previously stored.",
@@ -50,7 +48,7 @@ const search = tool({
  * Store a new memory to persist across conversations.
  */
 const create = tool({
-  id: "memories.create",
+  id: "create_memory",
   description:
     "Store a new memory. Use this to remember important facts, user preferences, " +
     "or context that should persist across conversations.",
@@ -59,13 +57,13 @@ const create = tool({
     collection: z
       .string()
       .optional()
-      .describe("Category for organizing memories (default: 'facts')"),
+      .describe("Category for organizing memories"),
   }),
   execute: async (ctx, { content, collection }) => {
     assert(ctx.agent, "ctx.agent required for memory tools");
 
     const mem = await ctx.agent.memories.create({
-      collection: collection ?? "facts",
+      collection,
       content: { text: content },
     });
 
@@ -73,11 +71,37 @@ const create = tool({
   },
 });
 
+/**
+ * Update an existing memory.
+ */
+const update = tool({
+  id: "update_memory",
+  description:
+    "Update an existing memory. Use this to correct or modify previously " +
+    "stored facts or preferences.",
+  parameters: z.object({
+    id: z.string().describe("ID of the memory to update"),
+    content: z.string().optional().describe("New text content"),
+    collection: z.string().optional().describe("New collection category"),
+  }),
+  execute: async (ctx, { id, content, collection }) => {
+    assert(ctx.agent, "ctx.agent required for memory tools");
+
+    const mem = await ctx.agent.memories.update({
+      id,
+      content: content ? { text: content } : undefined,
+      collection,
+    });
+
+    return { id: mem.id, updated: true };
+  },
+});
+
 /**
  * List stored memories, optionally filtered by collection.
  */
 const list = tool({
-  id: "memories.list",
+  id: "list_memories",
   description:
     "List your stored memories. Use this to see what you've remembered, " +
     "optionally filtered by collection.",
@@ -111,10 +135,10 @@ const list = tool({
 /**
  * Memory system toolkit.
  *
- * Provides memories.search, memories.create, and memories.list tools.
+ * Provides list_memories, create_memory, update_memory, and search_memories tools.
  */
 export const memory = new Toolkit({
   id: "sys.memory",
   description: "Tools for storing and retrieving agent memories",
-  tools: [list, create, search],
+  tools: [list, create, update, search],
 });