@echomem/mcp 1.0.0 → 1.0.3

package/dist/index.js

@@ -6,6 +6,7 @@ import axios from "axios";
 import { ZodError } from "zod";
 import { canonicalToolNames, keywordsSchema, listToolSpecs, othersSchema, resolveCanonicalToolName, saveConversationSchema, searchMemoriesSchema, timeRangeSchema, } from "./v1-contract.js";
 import { KeyStore } from "./keystore.js";
+import { randomUUID } from "node:crypto";
 import { fetchEncryptionConfig, decryptMemoryFields } from "./encryption.js";
 import { runCli } from "./setup.js";
 const ECHO_API_BASE_URL = process.env.ECHO_API_BASE_URL || "https://echo-mem-chrome.vercel.app";
@@ -79,6 +80,8 @@ class EchoMemApiClient {
     store;
     axios;
     whoamiCache = null;
+    /** One id per bridge process — groups all saves from this coding session under a single EchoMem context. */
+    sessionId = randomUUID();
     encConfigPromise = null;
     constructor(store) {
         this.store = store;
@@ -101,6 +104,27 @@ class EchoMemApiClient {
     hasToken() {
         return !!this.store.getToken();
     }
+    /**
+     * Compact topic map of the user's memory for the search-tool description — a cheap "what's in here"
+     * index built from memory keys so the agent knows the boundary up front and recalls proactively.
+     * Best-effort: returns undefined on any failure (locked vault, network) rather than blocking tools.
+     */
+    async fetchMemoryMap() {
+        try {
+            const enc = await this.encState(); // unencrypted → {enabled:false}; encrypted+locked → throws
+            const today = new Date().toISOString().slice(0, 10);
+            const response = await this.axios.post("/api/extension/memories/time-range", { startDate: "2000-01-01", endDate: today, limit: 200 }, { timeout: 6000 });
+            const data = enc.enabled ? await this.decryptResult(response.data, enc.key) : response.data;
+            const memories = Array.isArray(data?.memories) ? data.memories : [];
+            const keys = [...new Set(memories.map((m) => String(m?.keys || "").trim()).filter(Boolean))];
+            if (!keys.length)
+                return undefined;
+            return keys.slice(0, 40).map((k) => `- ${k}`).join("\n");
+        }
+        catch {
+            return undefined;
+        }
+    }
     /** Encryption config for the account, fetched once and cached. Failures are not cached. */
     async getEncryptionConfig() {
         if (!this.encConfigPromise) {
@@ -241,6 +265,8 @@ class EchoMemApiClient {
             sourceUrl: parsed.url,
             source: parsed.source || "mcp_server",
             title: parsed.title,
+            // Stable per-session id so multiple saves in this coding session group under one context.
+            conversationKey: this.sessionId,
         }, config);
         return response.data;
     }
@@ -286,6 +312,7 @@ class EchoMemApiClient {
 class EchoMemMCPServer {
     server;
     client;
+    mapCache = null;
     constructor(store) {
         this.server = new Server({
             name: "echomem-mcp",
@@ -304,7 +331,14 @@ class EchoMemMCPServer {
         });
     }
     setupToolHandlers() {
-        this.server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: listToolSpecs() }));
+        this.server.setRequestHandler(ListToolsRequestSchema, async () => {
+            // Inject a compact topic map of the user's memory into the search-tool description so the agent
+            // knows the boundary up front and recalls proactively (cached; best-effort — no map on failure).
+            if (this.client.hasToken() && !this.mapCache)
+                this.mapCache = this.client.fetchMemoryMap();
+            const map = this.mapCache ? await this.mapCache : undefined;
+            return { tools: listToolSpecs({ map }) };
+        });
         this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
             const canonicalName = resolveCanonicalToolName(request.params.name);
             try {

package/dist/setup.js

@@ -39,15 +39,17 @@ export function knownClients() {
         { id: "windsurf", label: "Windsurf", kind: "json", configPath: home(".codeium", "windsurf", "mcp_config.json") },
         { id: "claude-desktop", label: "Claude Desktop", kind: "json", configPath: path.join(appSupport, "Claude", "claude_desktop_config.json") },
         { id: "claude-code", label: "Claude Code", kind: "snippet", note: "run: claude mcp add-json echomem '<entry>'  (or add to .mcp.json)" },
-        { id: "codex", label: "Codex", kind: "snippet", note: "add to ~/.codex/config.toml under [mcp_servers.echomem]" },
+        { id: "codex", label: "Codex", kind: "command", detectDir: home(".codex"), configPath: home(".codex", "config.toml"), note: "add to ~/.codex/config.toml under [mcp_servers.echomem]" },
     ];
 }
 /** A client is "present" if its config dir already exists (JSON) — a cheap heuristic for detection. */
 export function detectClients() {
     return knownClients().filter((c) => {
-        if (c.kind !== "json")
-            return false;
-        return fs.existsSync(path.dirname(c.configPath));
+        if (c.kind === "json")
+            return fs.existsSync(path.dirname(c.configPath));
+        if (c.kind === "command")
+            return fs.existsSync(c.detectDir);
+        return false;
     });
 }
 /**
@@ -61,6 +63,28 @@ export function buildServerEntry(opts = {}) {
     }
     return { command: "npx", args: ["-y", "@echomem/mcp"] };
 }
+/** The TOML block EchoMem adds to ~/.codex/config.toml. No secret — the bridge reads the keystore. */
+export function codexTomlBlock(entry) {
+    const command = JSON.stringify(String(entry.command));
+    const args = (Array.isArray(entry.args) ? entry.args : []).map((a) => JSON.stringify(String(a))).join(", ");
+    return `[mcp_servers.echomem]\ncommand = ${command}\nargs = [${args}]\n`;
+}
+/** Write/merge the EchoMem entry straight into Codex's config.toml — no `codex` CLI needed. Idempotent. */
+function writeCodexConfig(configPath, entry) {
+    let content = "";
+    try {
+        content = fs.readFileSync(configPath, "utf8");
+    }
+    catch {
+        /* fresh config */
+    }
+    if (/^\s*\[mcp_servers\.echomem\]/m.test(content))
+        return "exists";
+    fs.mkdirSync(path.dirname(configPath), { recursive: true });
+    const sep = content ? (content.endsWith("\n") ? "\n" : "\n\n") : "";
+    fs.appendFileSync(configPath, sep + codexTomlBlock(entry));
+    return "wrote";
+}
 /** Merge the EchoMem entry into a JSON client's `mcpServers` map without clobbering siblings. */
 export function writeJsonClientConfig(configPath, entry) {
     let config = {};
@@ -243,9 +267,9 @@ async function cmdSetup(flags) {
     const requested = typeof flags.client === "string" ? flags.client : undefined;
     const targets = (requested ? knownClients().filter((c) => c.id === requested) : detectClients());
     if (targets.length === 0) {
-        console.log("No JSON-config client auto-detected. Add this MCP server entry manually:\n");
+        console.log("No client auto-detected. Add this MCP server entry manually:\n");
         console.log(JSON.stringify({ echomem: entry }, null, 2));
-        console.log("\n(Or re-run with --client cursor|windsurf|claude-desktop.)");
+        console.log("\n(Or re-run with --client cursor|windsurf|claude-desktop|codex.)");
     }
     else {
         for (const c of targets) {
@@ -253,6 +277,13 @@ async function cmdSetup(flags) {
                 writeJsonClientConfig(c.configPath, entry);
                 console.log(`✅ Wrote EchoMem MCP entry to ${c.label}: ${c.configPath}`);
             }
+            else if (c.kind === "command") {
+                const result = writeCodexConfig(c.configPath, entry);
+                if (result === "wrote")
+                    console.log(`✅ Wrote EchoMem MCP entry to ${c.label}: ${c.configPath} — start a new Codex session to load it.`);
+                else
+                    console.log(`✅ ${c.label} already has the EchoMem MCP entry: ${c.configPath}`);
+            }
             else {
                 console.log(`ℹ️  ${c.label}: ${c.note}\n   entry: ${JSON.stringify(entry)}`);
             }

package/dist/v1-contract.js

@@ -45,12 +45,16 @@ export const keywordsSchema = z.object({
 export const othersSchema = z.object({
     query: z.string(),
 });
-export function listToolSpecs() {
+export function listToolSpecs(opts = {}) {
     const currentTime = new Date().toISOString();
+    const map = opts.map?.trim();
+    const mapSection = map
+        ? `\n\nThis user's EchoMem currently covers these topics (a relevance guide — recall when the task relates to one of them):\n${map}\n`
+        : "";
     return [
         {
             name: canonicalToolNames.search,
-            description: `Recall the user's prior decisions, preferences, constraints, and context from EchoMem — their long-term memory spanning ALL their AI tools (Claude.ai, ChatGPT, other agents), not just this session. CALL THIS AT THE START of any non-trivial task, before planning or writing code, to avoid re-deriving things the user already decided. Also call it whenever the user refers to past work ("what did we decide", "like before", "the usual"). Returns ranked memories with provenance. Current time: ${currentTime}.`,
+            description: `Recall the user's prior decisions, preferences, constraints, and project context from EchoMem — their long-term memory across ALL their AI tools (Claude.ai, ChatGPT, other agents), not just this session.${mapSection}\nCall this when the task plausibly relates to that remembered context — a topic above, or when the user refers to past work ("what did we decide", "like before", "the usual") — so you don't re-derive or re-ask what they already settled. Skip it for self-contained tasks with no link to their history (e.g. a generic algorithm question). Returns ranked memories with provenance. Current time: ${currentTime}.`,
             inputSchema: {
                 type: "object",
                 properties: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@echomem/mcp",
-  "version": "1.0.0",
+  "version": "1.0.3",
   "description": "EchoMem Cloud-First MCP Server",
   "main": "dist/index.js",
   "type": "module",