persisted-memory 1.0.0

package/dist/server.js

@@ -0,0 +1,399 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { z } from "zod";
+import { randomUUID } from "node:crypto";
+import { join, dirname } from "node:path";
+import { fileURLToPath } from "node:url";
+import { spawn } from "node:child_process";
+import { MEMORY_TYPES, IMPORTANCE_LEVELS } from "./storage/types.js";
+import * as lanceStore from "./storage/lance-store.js";
+import * as mdStore from "./storage/markdown-store.js";
+import * as kgStore from "./storage/knowledge-graph.js";
+import { embed, isAvailable } from "./embeddings/ollama.js";
+import { hybridSearch } from "./search/hybrid.js";
+import { chunkText } from "./utils/chunking.js";
+import { getMemoryDir, getProjectName } from "./utils/project.js";
+import { stripPrivate, hasPrivateContent } from "./utils/privacy.js";
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+const SUMMARY_TRUNCATE_LENGTH = 200;
+// Lazy initialization cache per memory dir
+const initPromises = new Map();
+async function ensureInit() {
+    const memoryDir = getMemoryDir();
+    if (!initPromises.has(memoryDir)) {
+        initPromises.set(memoryDir, (async () => {
+            mdStore.ensureDirs(memoryDir);
+            await lanceStore.init(memoryDir);
+            return memoryDir;
+        })());
+    }
+    return initPromises.get(memoryDir);
+}
+let graphMutex = Promise.resolve();
+async function withGraphLock(fn) {
+    const prev = graphMutex;
+    let resolve;
+    graphMutex = new Promise((r) => { resolve = r; });
+    await prev;
+    try {
+        return fn();
+    }
+    finally {
+        resolve();
+    }
+}
+export function createServer() {
+    const server = new McpServer({
+        name: "persisted-memory",
+        version: "1.0.0",
+    });
+    // --- memory_store ---
+    server.tool("memory_store", "Store a memory entry. Wrap sensitive content in <private>...</private> tags to exclude it from summaries and public views. Dual-writes to both LanceDB (vector index) and Markdown files (source of truth).", {
+        text: z.string().describe("The memory content to store"),
+        type: z.enum(MEMORY_TYPES).describe("Category of memory"),
+        importance: z.enum(IMPORTANCE_LEVELS).describe("How important this memory is"),
+        file_paths: z.array(z.string()).optional().describe("Related file paths"),
+        tags: z.array(z.string()).optional().describe("Free-form tags"),
+        session_id: z.string().optional().describe("Claude Code session ID"),
+    }, async (params) => {
+        const memoryDir = await ensureInit();
+        const entry = {
+            id: randomUUID(),
+            text: params.text,
+            type: params.type,
+            importance: params.importance,
+            timestamp: Date.now(),
+            session_id: params.session_id,
+            file_paths: params.file_paths,
+            tags: params.tags,
+        };
+        // Get embedding
+        const chunks = chunkText(entry.text);
+        const vector = await embed(chunks[0]);
+        // Store in LanceDB
+        await lanceStore.store(entry, vector || new Array(768).fill(0));
+        // If text is long, store additional chunks
+        for (let i = 1; i < chunks.length; i++) {
+            const chunkEntry = {
+                ...entry,
+                id: randomUUID(),
+                text: chunks[i],
+            };
+            const chunkVec = await embed(chunks[i]);
+            await lanceStore.store(chunkEntry, chunkVec || new Array(768).fill(0));
+        }
+        // Dual-write to Markdown
+        mdStore.appendDaily(memoryDir, `**[${entry.type}]** (${entry.importance})\n\n${entry.text}`);
+        if (entry.type === "decision") {
+            mdStore.appendDecision(memoryDir, entry.text);
+        }
+        // Update knowledge graph with extracted entities
+        await withGraphLock(() => {
+            const graph = kgStore.loadGraph(memoryDir);
+            kgStore.updateGraphFromMemory(graph, entry.id, entry.text, entry.tags, entry.file_paths);
+            kgStore.saveGraph(memoryDir, graph);
+        });
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        id: entry.id,
+                        stored: true,
+                        chunks: chunks.length,
+                        project: getProjectName(),
+                    }),
+                },
+            ],
+        };
+    });
+    // --- memory_search ---
+    server.tool("memory_search", "Search memories using hybrid search (70% semantic + 30% keyword). Falls back to keyword-only if Ollama is unavailable.", {
+        query: z.string().describe("Search query"),
+        limit: z.number().optional().default(10).describe("Max results to return"),
+        type: z.enum(MEMORY_TYPES).optional().describe("Filter by memory type"),
+        importance: z.enum(IMPORTANCE_LEVELS).optional().describe("Filter by importance"),
+    }, async (params) => {
+        await ensureInit();
+        const results = await hybridSearch(params.query, params.limit, {
+            type: params.type,
+            importance: params.importance,
+        });
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        results: results.map((r) => ({
+                            id: r.id,
+                            text: stripPrivate(r.text),
+                            type: r.type,
+                            importance: r.importance,
+                            score: Math.round(r.score * 1000) / 1000,
+                            timestamp: r.timestamp,
+                            file_paths: r.file_paths,
+                            tags: r.tags,
+                        })),
+                        count: results.length,
+                        project: getProjectName(),
+                    }),
+                },
+            ],
+        };
+    });
+    // --- memory_recall ---
+    server.tool("memory_recall", "Recall memories about a specific topic. Returns formatted, human-readable results optimized for context injection. Supports progressive disclosure via detail_level to save tokens.", {
+        topic: z.string().describe("Topic to recall memories about"),
+        limit: z.number().optional().default(5).describe("Max results"),
+        detail_level: z.enum(["summary", "detailed", "ids_only"]).optional().default("summary").describe("Level of detail: 'summary' (default, token-efficient), 'detailed' (full text), 'ids_only' (minimal)"),
+    }, async (params) => {
+        await ensureInit();
+        const results = await hybridSearch(params.topic, params.limit);
+        if (results.length === 0) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: `No memories found for topic: "${params.topic}"`,
+                    },
+                ],
+            };
+        }
+        const detailLevel = params.detail_level ?? "summary";
+        if (detailLevel === "ids_only") {
+            const items = results.map((r) => ({
+                id: r.id,
+                type: r.type,
+                importance: r.importance,
+                score: Math.round(r.score * 1000) / 1000,
+            }));
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            results: items,
+                            count: results.length,
+                            project: getProjectName(),
+                            hint: `Use detail_level: "detailed" for full content, or memory_search with specific IDs`,
+                        }),
+                    },
+                ],
+            };
+        }
+        if (detailLevel === "detailed") {
+            const formatted = results
+                .map((r, i) => {
+                const date = new Date(r.timestamp).toISOString().slice(0, 16).replace("T", " ");
+                const tags = r.tags?.length ? ` [${r.tags.join(", ")}]` : "";
+                const displayText = stripPrivate(r.text);
+                const privateFlag = hasPrivateContent(r.text) ? " [private]" : "";
+                return `${i + 1}. **${r.type}** (${r.importance}) — ${date}${tags}${privateFlag}\n   ${displayText}`;
+            })
+                .join("\n\n");
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: `## Memories: ${params.topic}\n\n${formatted}`,
+                    },
+                ],
+            };
+        }
+        // Default: summary — truncated text for token efficiency
+        const formatted = results
+            .map((r, i) => {
+            const date = new Date(r.timestamp).toISOString().slice(0, 16).replace("T", " ");
+            const tags = r.tags?.length ? ` [${r.tags.join(", ")}]` : "";
+            const displayText = stripPrivate(r.text);
+            const privateFlag = hasPrivateContent(r.text) ? " [private]" : "";
+            const truncatedText = displayText.length > SUMMARY_TRUNCATE_LENGTH ? displayText.slice(0, SUMMARY_TRUNCATE_LENGTH) + "..." : displayText;
+            return `${i + 1}. **${r.type}** (${r.importance}) — ${date}${tags}${privateFlag}\n   ${truncatedText}`;
+        })
+            .join("\n\n");
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `## Memories: ${params.topic}\n\n${formatted}\n\n_Use detail_level: "detailed" for full content, or memory_search with specific IDs_`,
+                },
+            ],
+        };
+    });
+    // --- memory_list ---
+    server.tool("memory_list", "List stored memories with pagination. Returns entries sorted by timestamp (newest first).", {
+        limit: z.number().optional().default(20).describe("Max entries to return"),
+        offset: z.number().optional().default(0).describe("Offset for pagination"),
+        type: z.enum(MEMORY_TYPES).optional().describe("Filter by memory type"),
+    }, async (params) => {
+        await ensureInit();
+        const entries = await lanceStore.list(params.limit, params.offset, params.type);
+        const total = await lanceStore.count();
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        entries: entries.map((e) => {
+                            const displayText = stripPrivate(e.text);
+                            return {
+                                id: e.id,
+                                text: displayText.slice(0, SUMMARY_TRUNCATE_LENGTH) + (displayText.length > SUMMARY_TRUNCATE_LENGTH ? "..." : ""),
+                                type: e.type,
+                                importance: e.importance,
+                                timestamp: e.timestamp,
+                            };
+                        }),
+                        total,
+                        project: getProjectName(),
+                    }),
+                },
+            ],
+        };
+    });
+    // --- memory_status ---
+    server.tool("memory_status", "Get memory system status: total entries, entries by type, Ollama availability, and memory directory path.", {}, async () => {
+        const memoryDir = await ensureInit();
+        const [total, byType, ollamaAvailable] = await Promise.all([
+            lanceStore.count(),
+            lanceStore.countByType(),
+            isAvailable(),
+        ]);
+        const summary = mdStore.readSummary(memoryDir);
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        total,
+                        by_type: byType,
+                        ollama_available: ollamaAvailable,
+                        memory_dir: memoryDir,
+                        project: getProjectName(),
+                        has_summary: summary !== null,
+                    }),
+                },
+            ],
+        };
+    });
+    // --- memory_forget ---
+    server.tool("memory_forget", "Delete a specific memory entry by ID from the vector index.", {
+        id: z.string().describe("Memory entry ID to delete"),
+    }, async (params) => {
+        const memoryDir = await ensureInit();
+        await lanceStore.deleteEntry(params.id);
+        // Prune knowledge graph of references to the deleted memory
+        await withGraphLock(() => {
+            const graph = kgStore.loadGraph(memoryDir);
+            kgStore.removeMemoryFromGraph(graph, params.id);
+            kgStore.saveGraph(memoryDir, graph);
+        });
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({ deleted: true, id: params.id }),
+                },
+            ],
+        };
+    });
+    // --- memory_consolidate ---
+    server.tool("memory_consolidate", "Consolidate memories by finding and removing near-duplicate entries (cosine similarity > 0.95). Keeps the entry with higher importance or newer timestamp.", {}, async () => {
+        await ensureInit();
+        const result = await lanceStore.consolidate(embed);
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        merged: result.merged,
+                        removed: result.removed,
+                        project: getProjectName(),
+                    }),
+                },
+            ],
+        };
+    });
+    // --- memory_graph ---
+    server.tool("memory_graph", "Explore the knowledge graph. Find entities and their relationships to understand how concepts connect across memories.", {
+        entity: z.string().optional().describe("Entity name to explore connections for"),
+        list_entities: z.boolean().optional().default(false).describe("List all known entities"),
+    }, async (params) => {
+        const memoryDir = await ensureInit();
+        const graph = kgStore.loadGraph(memoryDir);
+        if (params.entity) {
+            const result = kgStore.findRelated(graph, params.entity);
+            const target = graph.entities[params.entity.toLowerCase().trim()];
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            entity: target || null,
+                            connected_entities: result.entities,
+                            relations: result.relations,
+                            project: getProjectName(),
+                        }),
+                    },
+                ],
+            };
+        }
+        if (params.list_entities) {
+            const entities = Object.values(graph.entities).sort((a, b) => b.last_seen - a.last_seen);
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: JSON.stringify({
+                            entities,
+                            total: entities.length,
+                            project: getProjectName(),
+                        }),
+                    },
+                ],
+            };
+        }
+        // Default: return graph summary
+        const entityCount = Object.keys(graph.entities).length;
+        const relationCount = graph.relations.length;
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        entity_count: entityCount,
+                        relation_count: relationCount,
+                        project: getProjectName(),
+                        hint: "Use entity param to explore connections, or list_entities: true to see all entities.",
+                    }),
+                },
+            ],
+        };
+    });
+    // --- memory_viewer ---
+    server.tool("memory_viewer", "Launch the web-based memory viewer for visual exploration of stored memories.", {
+        port: z.number().optional().default(3777).describe("Port to serve the viewer on"),
+    }, async (params) => {
+        const viewerScript = join(__dirname, "cli", "viewer.js");
+        const child = spawn("node", [viewerScript], {
+            env: { ...process.env, PORT: String(params.port) },
+            stdio: "ignore",
+            detached: true,
+        });
+        child.unref();
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: JSON.stringify({
+                        launched: true,
+                        url: `http://localhost:${params.port}`,
+                        note: "Memory viewer launched. Open the URL in your browser.",
+                    }),
+                },
+            ],
+        };
+    });
+    return server;
+}
+//# sourceMappingURL=server.js.map

package/dist/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAG3C,OAAO,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACrE,OAAO,KAAK,UAAU,MAAM,0BAA0B,CAAC;AACvD,OAAO,KAAK,OAAO,MAAM,6BAA6B,CAAC;AACvD,OAAO,KAAK,OAAO,MAAM,8BAA8B,CAAC;AACxD,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAClE,OAAO,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAErE,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAEtC,MAAM,uBAAuB,GAAG,GAAG,CAAC;AAEpC,2CAA2C;AAC3C,MAAM,YAAY,GAAG,IAAI,GAAG,EAA2B,CAAC;AAExD,KAAK,UAAU,UAAU;IACvB,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;IACjC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;QACjC,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,KAAK,IAAI,EAAE;YACtC,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YAC9B,MAAM,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACjC,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,EAAE,CAAC,CAAC;IACR,CAAC;IACD,OAAO,YAAY,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;AACtC,CAAC;AAED,IAAI,UAAU,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;AAElD,KAAK,UAAU,aAAa,CAAI,EAAW;IACzC,MAAM,IAAI,GAAG,UAAU,CAAC;IACxB,IAAI,OAAoB,CAAC;IACzB,UAAU,GAAG,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE,GAAG,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,MAAM,IAAI,CAAC;IACX,IAAI,CAAC;QACH,OAAO,EAAE,EAAE,CAAC;IACd,CAAC;YAAS,CAAC;QACT,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,YAAY;IAC1B,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;QAC3B,IAAI,EAAE,kBAAkB;QACxB,OAAO,EAAE,OAAO;KACjB,CAAC,CAAC;IAEH,uBAAuB;IACvB,MAAM,CAAC,IAAI,CACT,cAAc,EACd,6MAA6M,EAC7M;QACE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;QACxD,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,QAAQ,CAAC,oBAAoB,CAAC;QACzD,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,QAAQ,CAAC,8BAA8B,CAAC;QAC9E,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC;QACzE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QAC/D,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;KACrE,EACD,KAAK,EAAE,MAAM,EAAE,EAAE;QACf,MAAM,SAAS,GAAG,MAAM,UAAU,EAAE,CAAC;QAErC,MAAM,KAAK,GAAgB;YACzB,EAAE,EAAE,UAAU,EAAE;YAChB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,MAAM,CAAC,IAAkB;YAC/B,UAAU,EAAE,MAAM,CAAC,UAAwB;YAC3C,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,IAAI,EAAE,MAAM,CAAC,IAAI;SAClB,CAAC;QAEF,gBAAgB;QAChB,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAEtC,mBAAmB;QACnB,MAAM,UAAU,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,IAAI,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAEhE,2CAA2C;QAC3C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,MAAM,UAAU,GAAgB;gBAC9B,GAAG,KAAK;gBACR,EAAE,EAAE,UAAU,EAAE;gBAChB,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;aAChB,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,UAAU,CAAC,KAAK,CAAC,UAAU,EAAE,QAAQ,IAAI,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACzE,CAAC;QAED,yBAAyB;QACzB,OAAO,CAAC,WAAW,CAAC,SAAS,EAAE,MAAM,KAAK,CAAC,IAAI,QAAQ,KAAK,CAAC,UAAU,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAE7F,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC9B,OAAO,CAAC,cAAc,CAAC,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC;QAED,iDAAiD;QACjD,MAAM,aAAa,CAAC,GAAG,EAAE;YACvB,MAAM,KAAK,GAAG,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC3C,OAAO,CAAC,qBAAqB,CAC3B,KAAK,EACL,KAAK,CAAC,EAAE,EACR,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,UAAU,CACjB,CAAC;YACF,OAAO,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,EAAE,EAAE,KAAK,CAAC,EAAE;wBACZ,MAAM,EAAE,IAAI;wBACZ,MAAM,EAAE,MAAM,CAAC,MAAM;wBACrB,OAAO,EAAE,cAAc,EAAE;qBAC1B,CAAC;iBACH;aACF;SACF,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,wBAAwB;IACxB,MAAM,CAAC,IAAI,CACT,eAAe,EACf,wHAAwH,EACxH;QACE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC1C,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QAC1E,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QACvE,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sBAAsB,CAAC;KAClF,EACD,KAAK,EAAE,MAAM,EAAE,EAAE;QACf,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE;YAC7D,IAAI,EAAE,MAAM,CAAC,IAA8B;YAC3C,UAAU,EAAE,MAAM,CAAC,UAAoC;SACxD,CAAC,CAAC;QAEH,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;4BAC3B,EAAE,EAAE,CAAC,CAAC,EAAE;4BACR,IAAI,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC;4BAC1B,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,UAAU,EAAE,CAAC,CAAC,UAAU;4BACxB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,IAAI;4BACxC,SAAS,EAAE,CAAC,CAAC,SAAS;4BACtB,UAAU,EAAE,CAAC,CAAC,UAAU;4BACxB,IAAI,EAAE,CAAC,CAAC,IAAI;yBACb,CAAC,CAAC;wBACH,KAAK,EAAE,OAAO,CAAC,MAAM;wBACrB,OAAO,EAAE,cAAc,EAAE;qBAC1B,CAAC;iBACH;aACF;SACF,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,wBAAwB;IACxB,MAAM,CAAC,IAAI,CACT,eAAe,EACf,qLAAqL,EACrL;QACE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;QAC5D,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC;QAC/D,YAAY,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,qGAAqG,CAAC;KACxM,EACD,KAAK,EAAE,MAAM,EAAE,EAAE;QACf,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAE/D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,iCAAiC,MAAM,CAAC,KAAK,GAAG;qBACvD;iBACF;aACF,CAAC;QACJ,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,CAAC,YAAY,IAAI,SAAS,CAAC;QAErD,IAAI,WAAW,KAAK,UAAU,EAAE,CAAC;YAC/B,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAChC,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,UAAU,EAAE,CAAC,CAAC,UAAU;gBACxB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,IAAI;aACzC,CAAC,CAAC,CAAC;YAEJ,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;4BACnB,OAAO,EAAE,KAAK;4BACd,KAAK,EAAE,OAAO,CAAC,MAAM;4BACrB,OAAO,EAAE,cAAc,EAAE;4BACzB,IAAI,EAAE,mFAAmF;yBAC1F,CAAC;qBACH;iBACF;aACF,CAAC;QACJ,CAAC;QAED,IAAI,WAAW,KAAK,UAAU,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAG,OAAO;iBACtB,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;gBACZ,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;gBAChF,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7D,MAAM,WAAW,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACzC,MAAM,WAAW,GAAG,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClE,OAAO,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,UAAU,OAAO,IAAI,GAAG,IAAI,GAAG,WAAW,QAAQ,WAAW,EAAE,CAAC;YACvG,CAAC,CAAC;iBACD,IAAI,CAAC,MAAM,CAAC,CAAC;YAEhB,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,gBAAgB,MAAM,CAAC,KAAK,OAAO,SAAS,EAAE;qBACrD;iBACF;aACF,CAAC;QACJ,CAAC;QAED,yDAAyD;QACzD,MAAM,SAAS,GAAG,OAAO;aACtB,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACZ,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAChF,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7D,MAAM,WAAW,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,WAAW,GAAG,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC;YAClE,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,GAAG,uBAAuB,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,uBAAuB,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC;YACzI,OAAO,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,UAAU,OAAO,IAAI,GAAG,IAAI,GAAG,WAAW,QAAQ,aAAa,EAAE,CAAC;QACzG,CAAC,CAAC;aACD,IAAI,CAAC,MAAM,CAAC,CAAC;QAEhB,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,gBAAgB,MAAM,CAAC,KAAK,OAAO,SAAS,yFAAyF;iBAC5I;aACF;SACF,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,sBAAsB;IACtB,MAAM,CAAC,IAAI,CACT,aAAa,EACb,2FAA2F,EAC3F;QACE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QAC1E,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QAC1E,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;KACxE,EACD,KAAK,EAAE,MAAM,EAAE,EAAE;QACf,MAAM,UAAU,EAAE,CAAC;QAEnB,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,IAAI,CACnC,MAAM,CAAC,KAAK,EACZ,MAAM,CAAC,MAAM,EACb,MAAM,CAAC,IAA8B,CACtC,CAAC;QACF,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;QAEvC,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;4BACzB,MAAM,WAAW,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;4BACzC,OAAO;gCACL,EAAE,EAAE,CAAC,CAAC,EAAE;gCACR,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,uBAAuB,CAAC,GAAG,CAAC,WAAW,CAAC,MAAM,GAAG,uBAAuB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gCACjH,IAAI,EAAE,CAAC,CAAC,IAAI;gCACZ,UAAU,EAAE,CAAC,CAAC,UAAU;gCACxB,SAAS,EAAE,CAAC,CAAC,SAAS;6BACvB,CAAC;wBACJ,CAAC,CAAC;wBACF,KAAK;wBACL,OAAO,EAAE,cAAc,EAAE;qBAC1B,CAAC;iBACH;aACF;SACF,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,wBAAwB;IACxB,MAAM,CAAC,IAAI,CACT,eAAe,EACf,2GAA2G,EAC3G,EAAE,EACF,KAAK,IAAI,EAAE;QACT,MAAM,SAAS,GAAG,MAAM,UAAU,EAAE,CAAC;QAErC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,eAAe,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACzD,UAAU,CAAC,KAAK,EAAE;YAClB,UAAU,CAAC,WAAW,EAAE;YACxB,WAAW,EAAE;SACd,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAE/C,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,KAAK;wBACL,OAAO,EAAE,MAAM;wBACf,gBAAgB,EAAE,eAAe;wBACjC,UAAU,EAAE,SAAS;wBACrB,OAAO,EAAE,cAAc,EAAE;wBACzB,WAAW,EAAE,OAAO,KAAK,IAAI;qBAC9B,CAAC;iBACH;aACF;SACF,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,wBAAwB;IACxB,MAAM,CAAC,IAAI,CACT,eAAe,EACf,6DAA6D,EAC7D;QACE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;KACrD,EACD,KAAK,EAAE,MAAM,EAAE,EAAE;QACf,MAAM,SAAS,GAAG,MAAM,UAAU,EAAE,CAAC;QACrC,MAAM,UAAU,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAExC,4DAA4D;QAC5D,MAAM,aAAa,CAAC,GAAG,EAAE;YACvB,MAAM,KAAK,GAAG,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC3C,OAAO,CAAC,qBAAqB,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;YAChD,OAAO,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC;iBACvD;aACF;SACF,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,6BAA6B;IAC7B,MAAM,CAAC,IAAI,CACT,oBAAoB,EACpB,4JAA4J,EAC5J,EAAE,EACF,KAAK,IAAI,EAAE;QACT,MAAM,UAAU,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAEnD,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,MAAM,EAAE,MAAM,CAAC,MAAM;wBACrB,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,OAAO,EAAE,cAAc,EAAE;qBAC1B,CAAC;iBACH;aACF;SACF,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,uBAAuB;IACvB,MAAM,CAAC,IAAI,CACT,cAAc,EACd,wHAAwH,EACxH;QACE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wCAAwC,CAAC;QAChF,aAAa,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,yBAAyB,CAAC;KACzF,EACD,KAAK,EAAE,MAAM,EAAE,EAAE;QACf,MAAM,SAAS,GAAG,MAAM,UAAU,EAAE,CAAC;QACrC,MAAM,KAAK,GAAG,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAE3C,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YACzD,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YAElE,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;4BACnB,MAAM,EAAE,MAAM,IAAI,IAAI;4BACtB,kBAAkB,EAAE,MAAM,CAAC,QAAQ;4BACnC,SAAS,EAAE,MAAM,CAAC,SAAS;4BAC3B,OAAO,EAAE,cAAc,EAAE;yBAC1B,CAAC;qBACH;iBACF;aACF,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CACjD,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CACpC,CAAC;YAEF,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;4BACnB,QAAQ;4BACR,KAAK,EAAE,QAAQ,CAAC,MAAM;4BACtB,OAAO,EAAE,cAAc,EAAE;yBAC1B,CAAC;qBACH;iBACF;aACF,CAAC;QACJ,CAAC;QAED,gCAAgC;QAChC,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;QACvD,MAAM,aAAa,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC;QAE7C,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,YAAY,EAAE,WAAW;wBACzB,cAAc,EAAE,aAAa;wBAC7B,OAAO,EAAE,cAAc,EAAE;wBACzB,IAAI,EAAE,sFAAsF;qBAC7F,CAAC;iBACH;aACF;SACF,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,wBAAwB;IACxB,MAAM,CAAC,IAAI,CACT,eAAe,EACf,+EAA+E,EAC/E;QACE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,6BAA6B,CAAC;KAClF,EACD,KAAK,EAAE,MAAM,EAAE,EAAE;QACf,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,YAAY,CAAC,EAAE;YAC1C,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAClD,KAAK,EAAE,QAAQ;YACf,QAAQ,EAAE,IAAI;SACf,CAAC,CAAC;QACH,KAAK,CAAC,KAAK,EAAE,CAAC;QAEd,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,QAAQ,EAAE,IAAI;wBACd,GAAG,EAAE,oBAAoB,MAAM,CAAC,IAAI,EAAE;wBACtC,IAAI,EAAE,uDAAuD;qBAC9D,CAAC;iBACH;aACF;SACF,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/storage/knowledge-graph.d.ts

@@ -0,0 +1,32 @@
+export interface Entity {
+    name: string;
+    type: "concept" | "file" | "tool" | "person" | "project" | "technology";
+    memory_ids: string[];
+    first_seen: number;
+    last_seen: number;
+}
+export interface Relation {
+    from: string;
+    to: string;
+    type: "relates_to" | "depends_on" | "replaces" | "contradicts" | "part_of";
+    memory_id: string;
+    timestamp: number;
+}
+export interface KnowledgeGraph {
+    entities: Record<string, Entity>;
+    relations: Relation[];
+}
+export declare function loadGraph(memoryDir: string): KnowledgeGraph;
+export declare function saveGraph(memoryDir: string, graph: KnowledgeGraph): void;
+export declare function addEntity(graph: KnowledgeGraph, name: string, type: Entity["type"], memoryId: string): void;
+export declare function addRelation(graph: KnowledgeGraph, from: string, to: string, type: Relation["type"], memoryId: string): void;
+export declare function findRelated(graph: KnowledgeGraph, entityName: string): {
+    entities: Entity[];
+    relations: Relation[];
+};
+export declare function extractEntities(text: string, tags?: string[], filePaths?: string[]): Array<{
+    name: string;
+    type: Entity["type"];
+}>;
+export declare function updateGraphFromMemory(graph: KnowledgeGraph, memoryId: string, text: string, tags?: string[], filePaths?: string[]): void;
+export declare function removeMemoryFromGraph(graph: KnowledgeGraph, memoryId: string): void;