@s_s/mnemo 1.0.0

package/build/tools/search.d.ts

@@ -0,0 +1,5 @@
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+/**
+ * Register the memory_search tool
+ */
+export declare function registerSearchTool(server: McpServer): void;

package/build/tools/search.js

@@ -0,0 +1,107 @@
+import { z } from 'zod';
+import { searchNotes, isEmbeddingReady } from '../core/embedding.js';
+import { readNote } from '../core/notes.js';
+/**
+ * Register the memory_search tool
+ */
+export function registerSearchTool(server) {
+    server.registerTool('memory_search', {
+        title: 'Search Memory',
+        description: 'Search through persistent memories using semantic similarity. Use this at the start of conversations to load relevant context, when the user references past discussions, or when you need background information for a task.',
+        inputSchema: {
+            query: z
+                .string()
+                .describe("Natural language description of what you're looking for. E.g., 'architecture decisions for the mnemo project', 'user preferences about code style'"),
+            top_k: z
+                .number()
+                .int()
+                .min(1)
+                .max(20)
+                .default(5)
+                .optional()
+                .describe('Maximum number of results to return (default: 5)'),
+            source_filter: z
+                .string()
+                .optional()
+                .describe("Filter results by source agent tool, e.g., 'opencode', 'claude-code'"),
+            tag_filter: z
+                .array(z.string())
+                .optional()
+                .describe('Filter results to only include notes that have ALL of the specified tags. E.g., ["architecture", "decision"]'),
+        },
+    }, async ({ query, top_k, source_filter, tag_filter }) => {
+        try {
+            if (!isEmbeddingReady()) {
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: 'Mnemo is still loading the embedding model. Please try again in a few seconds.',
+                        },
+                    ],
+                };
+            }
+            // When tag_filter is used, fetch more results to compensate for post-filtering
+            const fetchK = tag_filter && tag_filter.length > 0 ? (top_k || 5) * 3 : top_k || 5;
+            let results = await searchNotes(query, fetchK, source_filter);
+            // Post-filter by tags if specified
+            if (tag_filter && tag_filter.length > 0) {
+                results = results.filter((r) => {
+                    const noteTags = r.tags.split(',').filter(Boolean);
+                    return tag_filter.every((t) => noteTags.includes(t));
+                });
+                results = results.slice(0, top_k || 5);
+            }
+            if (results.length === 0) {
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: 'No relevant memories found.',
+                        },
+                    ],
+                };
+            }
+            // For each result, load the full note content
+            const enrichedResults = await Promise.all(results.map(async (r) => {
+                const note = await readNote(r.id);
+                return {
+                    id: r.id,
+                    score: r.score,
+                    content: note?.content || r.text,
+                    tags: note?.meta.tags || r.tags.split(',').filter(Boolean),
+                    source: r.source,
+                    created: r.created,
+                };
+            }));
+            const output = enrichedResults
+                .map((r, i) => `### Memory ${i + 1} (relevance: ${(r.score * 100).toFixed(1)}%)\n` +
+                `- **ID:** ${r.id}\n` +
+                `- **Tags:** [${Array.isArray(r.tags) ? r.tags.join(', ') : r.tags}]\n` +
+                `- **Source:** ${r.source}\n` +
+                `- **Created:** ${r.created}\n\n` +
+                `${r.content}`)
+                .join('\n\n---\n\n');
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Found ${results.length} relevant memories:\n\n${output}`,
+                    },
+                ],
+            };
+        }
+        catch (error) {
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Failed to search memories: ${error instanceof Error ? error.message : String(error)}`,
+                    },
+                ],
+                isError: true,
+            };
+        }
+    });
+}
+//# sourceMappingURL=search.js.map

package/build/tools/search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.js","sourceRoot":"","sources":["../../src/tools/search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAE5C;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,MAAiB;IAChD,MAAM,CAAC,YAAY,CACf,eAAe,EACf;QACI,KAAK,EAAE,eAAe;QACtB,WAAW,EACP,+NAA+N;QACnO,WAAW,EAAE;YACT,KAAK,EAAE,CAAC;iBACH,MAAM,EAAE;iBACR,QAAQ,CACL,oJAAoJ,CACvJ;YACL,KAAK,EAAE,CAAC;iBACH,MAAM,EAAE;iBACR,GAAG,EAAE;iBACL,GAAG,CAAC,CAAC,CAAC;iBACN,GAAG,CAAC,EAAE,CAAC;iBACP,OAAO,CAAC,CAAC,CAAC;iBACV,QAAQ,EAAE;iBACV,QAAQ,CAAC,kDAAkD,CAAC;YACjE,aAAa,EAAE,CAAC;iBACX,MAAM,EAAE;iBACR,QAAQ,EAAE;iBACV,QAAQ,CAAC,sEAAsE,CAAC;YACrF,UAAU,EAAE,CAAC;iBACR,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;iBACjB,QAAQ,EAAE;iBACV,QAAQ,CACL,8GAA8G,CACjH;SACR;KACJ,EACD,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,EAAE,EAAE;QAClD,IAAI,CAAC;YACD,IAAI,CAAC,gBAAgB,EAAE,EAAE,CAAC;gBACtB,OAAO;oBACH,OAAO,EAAE;wBACL;4BACI,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,gFAAgF;yBACzF;qBACJ;iBACJ,CAAC;YACN,CAAC;YAED,+EAA+E;YAC/E,MAAM,MAAM,GAAG,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;YACnF,IAAI,OAAO,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;YAE9D,mCAAmC;YACnC,IAAI,UAAU,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtC,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC3B,MAAM,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;oBACnD,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;gBACzD,CAAC,CAAC,CAAC;gBACH,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;YAC3C,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,OAAO;oBACH,OAAO,EAAE;wBACL;4BACI,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,6BAA6B;yBACtC;qBACJ;iBACJ,CAAC;YACN,CAAC;YAED,8CAA8C;YAC9C,MAAM,eAAe,GAAG,MAAM,OAAO,CAAC,GAAG,CACrC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;gBACpB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAClC,OAAO;oBACH,EAAE,EAAE,CAAC,CAAC,EAAE;oBACR,KAAK,EAAE,CAAC,CAAC,KAAK;oBACd,OAAO,EAAE,IAAI,EAAE,OAAO,IAAI,CAAC,CAAC,IAAI;oBAChC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;oBAC1D,MAAM,EAAE,CAAC,CAAC,MAAM;oBAChB,OAAO,EAAE,CAAC,CAAC,OAAO;iBACrB,CAAC;YACN,CAAC,CAAC,CACL,CAAC;YAEF,MAAM,MAAM,GAAG,eAAe;iBACzB,GAAG,CACA,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACL,cAAc,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;gBACnE,aAAa,CAAC,CAAC,EAAE,IAAI;gBACrB,gBAAgB,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK;gBACvE,iBAAiB,CAAC,CAAC,MAAM,IAAI;gBAC7B,kBAAkB,CAAC,CAAC,OAAO,MAAM;gBACjC,GAAG,CAAC,CAAC,OAAO,EAAE,CACrB;iBACA,IAAI,CAAC,aAAa,CAAC,CAAC;YAEzB,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,SAAS,OAAO,CAAC,MAAM,0BAA0B,MAAM,EAAE;qBAClE;iBACJ;aACJ,CAAC;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,8BAA8B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;qBAC/F;iBACJ;gBACD,OAAO,EAAE,IAAI;aAChB,CAAC;QACN,CAAC;IACL,CAAC,CACJ,CAAC;AACN,CAAC"}

package/build/tools/setup.d.ts

@@ -0,0 +1,5 @@
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+/**
+ * Register the memory_setup tool
+ */
+export declare function registerSetupTool(server: McpServer): void;

package/build/tools/setup.js

@@ -0,0 +1,120 @@
+import fs from 'node:fs/promises';
+import os from 'node:os';
+import { z } from 'zod';
+import { AGENT_TYPES } from '../core/config.js';
+import { getAgentConfig, injectPrompt, hasPromptInjected } from '../prompts/templates.js';
+/**
+ * Detect which agent tool is being used by checking config file existence
+ */
+async function detectAgentType(cwd) {
+    const home = os.homedir();
+    // Check for agent-specific config files
+    const checks = [
+        {
+            type: 'opencode',
+            paths: [`${cwd}/opencode.json`, `${cwd}/opencode.jsonc`, `${home}/.config/opencode/opencode.json`],
+        },
+        {
+            type: 'claude-code',
+            paths: [`${cwd}/CLAUDE.md`, `${home}/.claude/CLAUDE.md`],
+        },
+        {
+            type: 'openclaw',
+            paths: [`${home}/.openclaw/openclaw.json`],
+        },
+        {
+            type: 'codex',
+            paths: [`${cwd}/.codex/config.toml`, `${home}/.codex/config.toml`],
+        },
+    ];
+    for (const check of checks) {
+        for (const p of check.paths) {
+            try {
+                await fs.access(p);
+                return check.type;
+            }
+            catch {
+                continue;
+            }
+        }
+    }
+    return null;
+}
+/**
+ * Register the memory_setup tool
+ */
+export function registerSetupTool(server) {
+    server.registerTool('memory_setup', {
+        title: 'Memory Setup',
+        description: "Initialize Mnemo memory management. Writes memory management instructions into the agent's configuration file (e.g., AGENTS.md for OpenCode). Should be called once when setting up Mnemo for the first time.",
+        inputSchema: {
+            agent_type: z
+                .enum(AGENT_TYPES)
+                .optional()
+                .describe('Agent tool type. If not specified, will try to auto-detect. Valid values: opencode, claude-code, openclaw, codex'),
+            scope: z
+                .enum(['project', 'global'])
+                .default('project')
+                .describe("Where to write the prompt: 'project' for current project, 'global' for user-level config"),
+        },
+    }, async ({ agent_type, scope }) => {
+        const cwd = process.cwd();
+        const home = os.homedir();
+        // Determine agent type
+        let agentType = agent_type;
+        if (!agentType) {
+            const detected = await detectAgentType(cwd);
+            if (!detected) {
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: `Could not auto-detect agent type. Please specify agent_type parameter. Valid values: ${AGENT_TYPES.join(', ')}`,
+                        },
+                    ],
+                    isError: true,
+                };
+            }
+            agentType = detected;
+        }
+        const config = getAgentConfig(agentType);
+        const targetPath = scope === 'global' ? config.globalPath(home) : config.projectPath(cwd);
+        // Read existing content or start fresh
+        let existingContent = '';
+        try {
+            existingContent = await fs.readFile(targetPath, 'utf-8');
+        }
+        catch {
+            // File doesn't exist yet, that's fine
+        }
+        // Check if already installed
+        if (hasPromptInjected(existingContent)) {
+            // Update in place
+            const updated = injectPrompt(existingContent);
+            await fs.writeFile(targetPath, updated, 'utf-8');
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Mnemo memory instructions updated in ${targetPath}`,
+                    },
+                ],
+            };
+        }
+        // Inject prompt
+        const updated = injectPrompt(existingContent);
+        // Ensure parent directory exists
+        const dir = targetPath.substring(0, targetPath.lastIndexOf('/'));
+        await fs.mkdir(dir, { recursive: true });
+        await fs.writeFile(targetPath, updated, 'utf-8');
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: `Mnemo memory management initialized successfully.\n\nAgent type: ${agentType}\nConfig file: ${targetPath}\n\nMnemo is now ready to use.`,
+                },
+            ],
+        };
+    });
+}
+//# sourceMappingURL=setup.js.map

package/build/tools/setup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup.js","sourceRoot":"","sources":["../../src/tools/setup.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,WAAW,EAAkB,MAAM,mBAAmB,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAE1F;;GAEG;AACH,KAAK,UAAU,eAAe,CAAC,GAAW;IACtC,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;IAE1B,wCAAwC;IACxC,MAAM,MAAM,GAAgD;QACxD;YACI,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,CAAC,GAAG,GAAG,gBAAgB,EAAE,GAAG,GAAG,iBAAiB,EAAE,GAAG,IAAI,iCAAiC,CAAC;SACrG;QACD;YACI,IAAI,EAAE,aAAa;YACnB,KAAK,EAAE,CAAC,GAAG,GAAG,YAAY,EAAE,GAAG,IAAI,oBAAoB,CAAC;SAC3D;QACD;YACI,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,CAAC,GAAG,IAAI,0BAA0B,CAAC;SAC7C;QACD;YACI,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,CAAC,GAAG,GAAG,qBAAqB,EAAE,GAAG,IAAI,qBAAqB,CAAC;SACrE;KACJ,CAAC;IAEF,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QACzB,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAC1B,IAAI,CAAC;gBACD,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBACnB,OAAO,KAAK,CAAC,IAAI,CAAC;YACtB,CAAC;YAAC,MAAM,CAAC;gBACL,SAAS;YACb,CAAC;QACL,CAAC;IACL,CAAC;IAED,OAAO,IAAI,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,MAAiB;IAC/C,MAAM,CAAC,YAAY,CACf,cAAc,EACd;QACI,KAAK,EAAE,cAAc;QACrB,WAAW,EACP,+MAA+M;QACnN,WAAW,EAAE;YACT,UAAU,EAAE,CAAC;iBACR,IAAI,CAAC,WAAW,CAAC;iBACjB,QAAQ,EAAE;iBACV,QAAQ,CACL,kHAAkH,CACrH;YACL,KAAK,EAAE,CAAC;iBACH,IAAI,CAAC,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;iBAC3B,OAAO,CAAC,SAAS,CAAC;iBAClB,QAAQ,CACL,0FAA0F,CAC7F;SACR;KACJ,EACD,KAAK,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;QAC5B,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;QAC1B,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;QAE1B,uBAAuB;QACvB,IAAI,SAAS,GAA0B,UAAU,CAAC;QAClD,IAAI,CAAC,SAAS,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,GAAG,CAAC,CAAC;YAC5C,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACZ,OAAO;oBACH,OAAO,EAAE;wBACL;4BACI,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,wFAAwF,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;yBACzH;qBACJ;oBACD,OAAO,EAAE,IAAI;iBAChB,CAAC;YACN,CAAC;YACD,SAAS,GAAG,QAAQ,CAAC;QACzB,CAAC;QAED,MAAM,MAAM,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC;QACzC,MAAM,UAAU,GAAG,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAE1F,uCAAuC;QACvC,IAAI,eAAe,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC;YACD,eAAe,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC7D,CAAC;QAAC,MAAM,CAAC;YACL,sCAAsC;QAC1C,CAAC;QAED,6BAA6B;QAC7B,IAAI,iBAAiB,CAAC,eAAe,CAAC,EAAE,CAAC;YACrC,kBAAkB;YAClB,MAAM,OAAO,GAAG,YAAY,CAAC,eAAe,CAAC,CAAC;YAC9C,MAAM,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YACjD,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,wCAAwC,UAAU,EAAE;qBAC7D;iBACJ;aACJ,CAAC;QACN,CAAC;QAED,gBAAgB;QAChB,MAAM,OAAO,GAAG,YAAY,CAAC,eAAe,CAAC,CAAC;QAE9C,iCAAiC;QACjC,MAAM,GAAG,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,UAAU,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;QACjE,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAEzC,MAAM,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAEjD,OAAO;YACH,OAAO,EAAE;gBACL;oBACI,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,oEAAoE,SAAS,kBAAkB,UAAU,gCAAgC;iBAClJ;aACJ;SACJ,CAAC;IACN,CAAC,CACJ,CAAC;AACN,CAAC"}

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "@s_s/mnemo",
+  "version": "1.0.0",
+  "description": "Memory management MCP server for AI coding assistants",
+  "type": "module",
+  "main": "./build/index.js",
+  "bin": {
+    "mnemo": "build/index.js"
+  },
+  "files": [
+    "build"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "start": "node build/index.js",
+    "prettier": "prettier --check \"**/*\"",
+    "prettier:fix": "prettier -c --write \"**/*\"",
+    "prettier:staged": "prettier --write",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "release": "bash scripts/release.sh",
+    "prepare": "husky"
+  },
+  "keywords": [
+    "mcp",
+    "memory",
+    "agent"
+  ],
+  "author": "",
+  "license": "MIT",
+  "dependencies": {
+    "@huggingface/transformers": "^3.8.1",
+    "@modelcontextprotocol/sdk": "^1.27.1",
+    "vectra": "^0.12.3",
+    "zod": "^4.3.6"
+  },
+  "devDependencies": {
+    "@types/node": "^25.3.3",
+    "husky": "^9.1.7",
+    "lint-staged": "^16.3.2",
+    "prettier": "^3.8.1",
+    "typescript": "^5.9.3",
+    "vitest": "^4.0.18"
+  },
+  "lint-staged": {
+    "**/*.{js,ts,md}": "npm run prettier:staged --"
+  }
+}