diff --git a/apps/mcp/src/server/tools/add-memory.ts b/apps/mcp/src/server/tools/add-memory.ts index c708deb5..23c729dd 100644 --- a/apps/mcp/src/server/tools/add-memory.ts +++ b/apps/mcp/src/server/tools/add-memory.ts @@ -12,6 +12,7 @@ export function register(deps: ToolDeps) { .describe("The memory content to save or forget"), action: z.enum(["save", "forget"]).optional().default("save"), containerTag: optionalContainerTagSchema, + metadata: z.record(z.string(), z.unknown()).optional().describe("Optional metadata to attach to the saved memory"), }) deps.server.registerTool( @@ -42,7 +43,7 @@ export function register(deps: ToolDeps) { } } - const result = await client.createMemory(args.content) + const result = await client.createMemory(args.content, args.metadata) const message = `Memory saved (ID: ${result.id}, space: ${result.containerTag})` const structuredContent: AddMemoryOutput = { action: "save",