@naveenadi/mnemonic 0.1.0

package/dist/mcp/server.js

@@ -0,0 +1,360 @@
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
+import { MnemonicDB } from '../store/database.js';
+import { DocumentStore } from '../store/documents.js';
+import { CollectionStore } from '../store/collections.js';
+import { SearchPipeline } from '../search/pipeline.js';
+import { detectLLMBackend } from '../llm/factory.js';
+import { join } from 'node:path';
+import { homedir } from 'node:os';
+import { mkdirSync, writeFileSync } from 'node:fs';
+let mcpCtx = { dbPath: '' };
+export async function startMcpServer(dbPath, args) {
+    mcpCtx.dbPath = dbPath;
+    const httpPort = getArg(args, '--http') ? parseInt(getArg(args, '--port') ?? '8181', 10) : null;
+    const daemon = hasArg(args, '--daemon');
+    // Initialize database
+    const db = new MnemonicDB(dbPath);
+    db.init();
+    await db.loadVectors();
+    const docs = new DocumentStore(db);
+    const collections = new CollectionStore(db);
+    const pipeline = new SearchPipeline(db);
+    mcpCtx.db = db;
+    mcpCtx.docs = docs;
+    mcpCtx.collections = collections;
+    mcpCtx.pipeline = pipeline;
+    // Try to initialize LLM (non-blocking — will use available backend)
+    try {
+        const { backend } = await detectLLMBackend();
+        mcpCtx.llm = backend;
+        pipeline.setLLM(backend);
+    }
+    catch {
+        // No LLM available, search will fall back to BM25 only
+    }
+    const server = new Server({
+        name: 'mnemonic',
+        version: '0.1.0',
+    }, {
+        capabilities: {
+            tools: {},
+        },
+    });
+    server.setRequestHandler(ListToolsRequestSchema, async () => ({
+        tools: [
+            {
+                name: 'query',
+                description: `Hybrid search with typed sub-queries (lex/vec/hyde) combined via RRF + reranking.
+Each search is a typed sub-query. Use 'lex' for exact keyword/BM25 search, 'vec' for semantic vector search,
+and 'hyde' for hypothetical document embedding search. The first search gets 2x weight.
+
+Parameters:
+- searches: Array of { type: 'lex'|'vec'|'hyde', query: string } — the sub-queries to run. At least 1 required.
+- intent: Optional disambiguation context that helps steer ranking away from nearby-but-wrong concepts.
+- collections: Optional array of collection names to filter (OR). Defaults to all included collections.
+- limit: Max results (default 10).
+- minScore: Minimum relevance 0-1 (default 0).
+- candidateLimit: Max candidates to rerank (default 40).
+- rerank: Whether to run LLM reranking (default true).
+- decay: Apply time decay (default false).
+- boostLinks: Apply link boost (default false).`,
+                inputSchema: {
+                    type: 'object',
+                    properties: {
+                        searches: {
+                            type: 'array',
+                            items: {
+                                type: 'object',
+                                properties: {
+                                    type: { type: 'string', enum: ['lex', 'vec', 'hyde'] },
+                                    query: { type: 'string' },
+                                },
+                                required: ['type', 'query'],
+                            },
+                            minItems: 1,
+                            maxItems: 10,
+                        },
+                        intent: { type: 'string' },
+                        collections: {
+                            type: 'array',
+                            items: { type: 'string' },
+                        },
+                        limit: { type: 'number', default: 10 },
+                        minScore: { type: 'number', default: 0 },
+                        candidateLimit: { type: 'number', default: 40 },
+                        rerank: { type: 'boolean', default: true },
+                        decay: { type: 'boolean', default: false },
+                        boostLinks: { type: 'boolean', default: false },
+                    },
+                    required: ['searches'],
+                },
+            },
+            {
+                name: 'get',
+                description: `Retrieve a document by path or docid (with fuzzy matching suggestions).
+Supports :from:count suffix for line ranges (e.g. #abc123:50:40).
+
+Parameters:
+- file: Path, docid (#abc123), or path:from:count (e.g. #abc123:120:40).
+- fromLine: Start line (1-indexed); overrides the :from suffix.
+- maxLines: Limit returned lines.
+- lineNumbers: Prefix lines with numbers (default true).`,
+                inputSchema: {
+                    type: 'object',
+                    properties: {
+                        file: { type: 'string', description: 'Path, docid (#abc123), or path:from:count' },
+                        fromLine: { type: 'number', description: 'Start line (1-indexed)' },
+                        maxLines: { type: 'number', description: 'Limit returned lines' },
+                        lineNumbers: { type: 'boolean', default: true },
+                    },
+                    required: ['file'],
+                },
+            },
+            {
+                name: 'multi_get',
+                description: `Batch retrieve documents by glob pattern or comma-separated list.
+Supports docids (#abc123) in comma-separated lists.
+
+Parameters:
+- pattern: Glob pattern or comma-separated list of paths/docids.
+- maxBytes: Skip files larger than N (default 10240).
+- maxLines: Limit lines per file.
+- lineNumbers: Prefix lines with numbers (default true).`,
+                inputSchema: {
+                    type: 'object',
+                    properties: {
+                        pattern: { type: 'string', description: 'Glob pattern or comma-separated list' },
+                        maxBytes: { type: 'number', default: 10240 },
+                        maxLines: { type: 'number', default: 100 },
+                        lineNumbers: { type: 'boolean', default: true },
+                    },
+                    required: ['pattern'],
+                },
+            },
+            {
+                name: 'status',
+                description: 'Index health and collection info. Returns collection stats, document counts, vector status, and link graph info.',
+                inputSchema: {
+                    type: 'object',
+                    properties: {},
+                },
+            },
+        ],
+    }));
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        const { name, arguments: args } = request.params;
+        try {
+            switch (name) {
+                case 'query':
+                    return handleQuery(args);
+                case 'get':
+                    return handleGet(args);
+                case 'multi_get':
+                    return handleMultiGet(args);
+                case 'status':
+                    return handleStatus();
+                default:
+                    throw new Error(`Unknown tool: ${name}`);
+            }
+        }
+        catch (err) {
+            return {
+                content: [{ type: 'text', text: `Error: ${err.message}` }],
+                isError: true,
+            };
+        }
+    });
+    if (httpPort) {
+        // HTTP transport using Node's built-in http module
+        const http = await import('node:http');
+        const url = await import('node:url');
+        const httpServer = http.createServer(async (req, res) => {
+            res.setHeader('Access-Control-Allow-Origin', '*');
+            res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
+            res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
+            if (req.method === 'OPTIONS') {
+                res.writeHead(200);
+                res.end();
+                return;
+            }
+            const parsedUrl = new URL(req.url ?? '/', `http://${req.headers.host}`);
+            if (parsedUrl.pathname === '/health') {
+                res.writeHead(200, { 'Content-Type': 'application/json' });
+                res.end(JSON.stringify({ status: 'ok', uptime: process.uptime() }));
+                return;
+            }
+            if (parsedUrl.pathname === '/mcp' && req.method === 'POST') {
+                let body = '';
+                req.on('data', (chunk) => (body += chunk));
+                req.on('end', async () => {
+                    try {
+                        const jsonBody = JSON.parse(body);
+                        const result = await server.request(jsonBody, {});
+                        res.writeHead(200, { 'Content-Type': 'application/json' });
+                        res.end(JSON.stringify(result));
+                    }
+                    catch (err) {
+                        res.writeHead(400, { 'Content-Type': 'application/json' });
+                        res.end(JSON.stringify({ error: err.message }));
+                    }
+                });
+                return;
+            }
+            res.writeHead(404);
+            res.end('Not found');
+        });
+        httpServer.listen(httpPort, () => {
+            console.log(`MCP server running on http://localhost:${httpPort}`);
+        });
+        if (daemon) {
+            // Write PID file
+            const pidDir = join(homedir(), '.cache', 'mnemonic');
+            mkdirSync(pidDir, { recursive: true });
+            writeFileSync(join(pidDir, 'mcp.pid'), String(process.pid));
+        }
+    }
+    else {
+        // Stdio transport
+        const transport = new StdioServerTransport();
+        await server.connect(transport);
+    }
+}
+// Command handlers
+async function handleQuery(args) {
+    if (!mcpCtx.pipeline)
+        throw new Error('Search pipeline not initialized');
+    const { searches, intent, collections, limit, minScore, candidateLimit, rerank, decay, boostLinks } = args || {};
+    if (!searches || !Array.isArray(searches) || searches.length === 0) {
+        throw new Error('At least one search query is required');
+    }
+    const options = {
+        queries: searches.map((s) => ({
+            type: s.type,
+            query: s.query,
+        })),
+        intent,
+        collection: collections,
+        limit: limit ?? 10,
+        minScore: minScore ?? 0,
+        candidateLimit: candidateLimit ?? 40,
+        rerank: rerank !== false && !!mcpCtx.llm,
+        expand: false, // Already have explicit queries
+        hyde: false,
+        decay: decay ?? false,
+        boostLinks: boostLinks ?? false,
+    };
+    const results = await mcpCtx.pipeline.search(options);
+    return {
+        content: [
+            {
+                type: 'text',
+                text: JSON.stringify(results, null, 2),
+            },
+        ],
+    };
+}
+async function handleGet(args) {
+    if (!mcpCtx.docs)
+        throw new Error('Document store not initialized');
+    const { file, fromLine, maxLines, lineNumbers } = args || {};
+    if (!file)
+        throw new Error('file parameter is required');
+    const result = mcpCtx.docs.getBody(file, { fromLine, maxLines });
+    if ('error' in result) {
+        const similar = result.similarFiles;
+        return {
+            content: [
+                { type: 'text', text: `Document not found: ${file}` },
+                ...(similar?.length > 0 ? [{ type: 'text', text: `Similar files: ${similar.join(', ')}` }] : []),
+            ],
+            isError: true,
+        };
+    }
+    const meta = mcpCtx.docs.get(file);
+    let output = '';
+    if (!('error' in meta)) {
+        output += `mne://${meta.collection}/${meta.path}  #${meta.docid}\n---\n`;
+    }
+    if (lineNumbers !== false) {
+        const start = fromLine ?? 1;
+        result.content.split('\n').forEach((line, i) => {
+            output += `${start + i}: ${line}\n`;
+        });
+    }
+    else {
+        output += result.content;
+    }
+    return {
+        content: [{ type: 'text', text: output }],
+    };
+}
+async function handleMultiGet(args) {
+    if (!mcpCtx.docs)
+        throw new Error('Document store not initialized');
+    const { pattern, maxBytes, maxLines, lineNumbers } = args || {};
+    if (!pattern)
+        throw new Error('pattern parameter is required');
+    const ids = pattern.includes(',') ? pattern.split(',').map((s) => s.trim()) : undefined;
+    const output = [];
+    if (ids) {
+        for (const id of ids) {
+            if (output.length > 10)
+                break; // Safety limit
+            const doc = mcpCtx.docs.getBody(id, { maxLines: maxLines ?? 50 });
+            if ('error' in doc) {
+                output.push(`# ${id}: Not found`);
+                continue;
+            }
+            const meta = mcpCtx.docs.get(id);
+            if (!('error' in meta)) {
+                output.push(`# ${meta.path}  #${meta.docid}`);
+                output.push('---');
+                if (lineNumbers !== false) {
+                    doc.content.split('\n').forEach((line, i) => {
+                        output.push(`${i + 1}: ${line}`);
+                    });
+                }
+                else {
+                    output.push(doc.content);
+                }
+                output.push('');
+            }
+        }
+    }
+    return {
+        content: [{ type: 'text', text: output.join('\n') || 'No results' }],
+    };
+}
+async function handleStatus() {
+    if (!mcpCtx.db)
+        throw new Error('Database not initialized');
+    if (!mcpCtx.collections)
+        throw new Error('Collection store not initialized');
+    const cols = mcpCtx.collections.list();
+    return {
+        content: [
+            {
+                type: 'text',
+                text: JSON.stringify({
+                    collections: cols,
+                    hasVectors: mcpCtx.db.hasVectorIndex(),
+                    hasLLM: !!mcpCtx.llm,
+                    llmType: mcpCtx.llm?.constructor.name ?? 'none',
+                }, null, 2),
+            },
+        ],
+    };
+}
+// Helpers
+function getArg(args, name) {
+    const idx = args.indexOf(name);
+    if (idx >= 0 && idx < args.length - 1)
+        return args[idx + 1];
+    return undefined;
+}
+function hasArg(args, ...names) {
+    return names.some((n) => args.includes(n));
+}
+//# sourceMappingURL=server.js.map

package/dist/mcp/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/mcp/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EACL,qBAAqB,EACrB,sBAAsB,GACvB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAYnD,IAAI,MAAM,GAAe,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;AAExC,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,MAAc,EAAE,IAAc;IACjE,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;IAEvB,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAChG,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAExC,sBAAsB;IACtB,MAAM,EAAE,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;IAClC,EAAE,CAAC,IAAI,EAAE,CAAC;IACV,MAAM,EAAE,CAAC,WAAW,EAAE,CAAC;IAEvB,MAAM,IAAI,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;IACnC,MAAM,WAAW,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;IAC5C,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,EAAE,CAAC,CAAC;IAExC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC;IACf,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,MAAM,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAE3B,oEAAoE;IACpE,IAAI,CAAC;QACH,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;QAC7C,MAAM,CAAC,GAAG,GAAG,OAAO,CAAC;QACrB,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAAC,MAAM,CAAC;QACP,uDAAuD;IACzD,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB;QACE,IAAI,EAAE,UAAU;QAChB,OAAO,EAAE,OAAO;KACjB,EACD;QACE,YAAY,EAAE;YACZ,KAAK,EAAE,EAAE;SACV;KACF,CACF,CAAC;IAEF,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAC5D,KAAK,EAAE;YACL;gBACE,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE;;;;;;;;;;;;;gDAa2B;gBACxC,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,QAAQ,EAAE;4BACR,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE;gCACL,IAAI,EAAE,QAAQ;gCACd,UAAU,EAAE;oCACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,EAAE;oCACtD,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;iCAC1B;gCACD,QAAQ,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;6BAC5B;4BACD,QAAQ,EAAE,CAAC;4BACX,QAAQ,EAAE,EAAE;yBACb;wBACD,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;wBAC1B,WAAW,EAAE;4BACX,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;yBAC1B;wBACD,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;wBACtC,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE;wBACxC,cAAc,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;wBAC/C,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE;wBAC1C,KAAK,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE;wBAC1C,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE;qBAChD;oBACD,QAAQ,EAAE,CAAC,UAAU,CAAC;iBACvB;aACF;YACD;gBACE,IAAI,EAAE,KAAK;gBACX,WAAW,EAAE;;;;;;;yDAOoC;gBACjD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,2CAA2C,EAAE;wBAClF,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,wBAAwB,EAAE;wBACnE,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,sBAAsB,EAAE;wBACjE,WAAW,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE;qBAChD;oBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;iBACnB;aACF;YACD;gBACE,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE;;;;;;;yDAOoC;gBACjD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,sCAAsC,EAAE;wBAChF,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE;wBAC5C,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,EAAE;wBAC1C,WAAW,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE;qBAChD;oBACD,QAAQ,EAAE,CAAC,SAAS,CAAC;iBACtB;aACF;YACD;gBACE,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,kHAAkH;gBAC/H,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,EAAE;iBACf;aACF;SACF;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QAEjD,IAAI,CAAC;YACH,QAAQ,IAAI,EAAE,CAAC;gBACb,KAAK,OAAO;oBACV,OAAO,WAAW,CAAC,IAAI,CAAC,CAAC;gBAC3B,KAAK,KAAK;oBACR,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC;gBACzB,KAAK,WAAW;oBACd,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC;gBAC9B,KAAK,QAAQ;oBACX,OAAO,YAAY,EAAE,CAAC;gBACxB;oBACE,MAAM,IAAI,KAAK,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAW,GAAa,CAAC,OAAO,EAAE,EAAE,CAAC;gBACrE,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,QAAQ,EAAE,CAAC;QACb,mDAAmD;QACnD,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QACvC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;QAErC,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;YACtD,GAAG,CAAC,SAAS,CAAC,6BAA6B,EAAE,GAAG,CAAC,CAAC;YAClD,GAAG,CAAC,SAAS,CAAC,8BAA8B,EAAE,oBAAoB,CAAC,CAAC;YACpE,GAAG,CAAC,SAAS,CAAC,8BAA8B,EAAE,cAAc,CAAC,CAAC;YAE9D,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC7B,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACnB,GAAG,CAAC,GAAG,EAAE,CAAC;gBACV,OAAO;YACT,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,GAAG,EAAE,UAAU,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YAExE,IAAI,SAAS,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;gBACrC,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;gBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;gBACpE,OAAO;YACT,CAAC;YAED,IAAI,SAAS,CAAC,QAAQ,KAAK,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;gBAC3D,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC;gBACnD,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE;oBACvB,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAClC,MAAM,MAAM,GAAG,MAAO,MAAc,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAS,CAAC,CAAC;wBAClE,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;wBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;oBAClC,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;wBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;oBAC7D,CAAC;gBACH,CAAC,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACnB,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,UAAU,CAAC,MAAM,CAAC,QAAQ,EAAE,GAAG,EAAE;YAC/B,OAAO,CAAC,GAAG,CAAC,0CAA0C,QAAQ,EAAE,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,IAAI,MAAM,EAAE,CAAC;YACX,iBAAiB;YACjB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;YACrD,SAAS,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACvC,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;SAAM,CAAC;QACN,kBAAkB;QAClB,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;AACH,CAAC;AAED,mBAAmB;AACnB,KAAK,UAAU,WAAW,CAAC,IAAS;IAClC,IAAI,CAAC,MAAM,CAAC,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;IAEzE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC;IAEjH,IAAI,CAAC,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;IAC3D,CAAC;IAED,MAAM,OAAO,GAAkB;QAC7B,OAAO,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;YACjC,IAAI,EAAE,CAAC,CAAC,IAA8B;YACtC,KAAK,EAAE,CAAC,CAAC,KAAK;SACf,CAAC,CAAC;QACH,MAAM;QACN,UAAU,EAAE,WAAW;QACvB,KAAK,EAAE,KAAK,IAAI,EAAE;QAClB,QAAQ,EAAE,QAAQ,IAAI,CAAC;QACvB,cAAc,EAAE,cAAc,IAAI,EAAE;QACpC,MAAM,EAAE,MAAM,KAAK,KAAK,IAAI,CAAC,CAAC,MAAM,CAAC,GAAG;QACxC,MAAM,EAAE,KAAK,EAAE,gCAAgC;QAC/C,IAAI,EAAE,KAAK;QACX,KAAK,EAAE,KAAK,IAAI,KAAK;QACrB,UAAU,EAAE,UAAU,IAAI,KAAK;KAChC,CAAC;IAEF,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAEtD,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;aACvC;SACF;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,SAAS,CAAC,IAAS;IAChC,IAAI,CAAC,MAAM,CAAC,IAAI;QAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IAEpE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC;IAC7D,IAAI,CAAC,IAAI;QAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;IAEzD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;IAEjE,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;QACtB,MAAM,OAAO,GAAI,MAAc,CAAC,YAAY,CAAC;QAC7C,OAAO;YACL,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,uBAAuB,IAAI,EAAE,EAAE;gBACrD,GAAG,CAAC,OAAO,EAAE,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;aACjG;YACD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAQ,CAAC;IAC1C,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,IAAI,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,SAAS,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,IAAI,MAAM,IAAI,CAAC,KAAK,SAAS,CAAC;IAC3E,CAAC;IAED,IAAI,WAAW,KAAK,KAAK,EAAE,CAAC;QAC1B,MAAM,KAAK,GAAG,QAAQ,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,IAAY,EAAE,CAAS,EAAE,EAAE;YAC7D,MAAM,IAAI,GAAG,KAAK,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,MAAM,CAAC,OAAO,CAAC;IAC3B,CAAC;IAED,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;KAC1C,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,IAAS;IACrC,IAAI,CAAC,MAAM,CAAC,IAAI;QAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IAEpE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC;IAChE,IAAI,CAAC,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IAE/D,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAChG,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,IAAI,GAAG,EAAE,CAAC;QACR,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;YACrB,IAAI,MAAM,CAAC,MAAM,GAAG,EAAE;gBAAE,MAAM,CAAC,eAAe;YAC9C,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,QAAQ,IAAI,EAAE,EAAE,CAAC,CAAC;YAClE,IAAI,OAAO,IAAI,GAAG,EAAE,CAAC;gBACnB,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;gBAClC,SAAS;YACX,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAQ,CAAC;YACxC,IAAI,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,EAAE,CAAC;gBACvB,MAAM,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;gBAC9C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACnB,IAAI,WAAW,KAAK,KAAK,EAAE,CAAC;oBAC1B,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,IAAY,EAAE,CAAS,EAAE,EAAE;wBAC1D,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;oBACnC,CAAC,CAAC,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC3B,CAAC;gBACD,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,YAAY,EAAE,CAAC;KACrE,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,YAAY;IACzB,IAAI,CAAC,MAAM,CAAC,EAAE;QAAE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC5D,IAAI,CAAC,MAAM,CAAC,WAAW;QAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;IAE7E,MAAM,IAAI,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC;IAEvC,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAClB;oBACE,WAAW,EAAE,IAAI;oBACjB,UAAU,EAAE,MAAM,CAAC,EAAE,CAAC,cAAc,EAAE;oBACtC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG;oBACpB,OAAO,EAAE,MAAM,CAAC,GAAG,EAAE,WAAW,CAAC,IAAI,IAAI,MAAM;iBAChD,EACD,IAAI,EACJ,CAAC,CACF;aACF;SACF;KACF,CAAC;AACJ,CAAC;AAED,UAAU;AACV,SAAS,MAAM,CAAC,IAAc,EAAE,IAAY;IAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/B,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;IAC5D,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,MAAM,CAAC,IAAc,EAAE,GAAG,KAAe;IAChD,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;AAC7C,CAAC"}

package/dist/pi-extension/index.d.ts

@@ -0,0 +1,3 @@
+import type { ExtensionAPI } from '@earendil-works/pi-coding-agent';
+export default function (pi: ExtensionAPI): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/pi-extension/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/pi-extension/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAKpE,MAAM,CAAC,OAAO,WAAW,EAAE,EAAE,YAAY,QAwNxC"}

package/dist/pi-extension/index.js

@@ -0,0 +1,193 @@
+import { Type } from 'typebox';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+export default function (pi) {
+    const dbPath = join(homedir(), '.cache', 'mnemonic', 'index.sqlite');
+    // ─── Tool: mnemonic_search ─────────────────────────────────────
+    pi.registerTool({
+        name: 'mnemonic_search',
+        label: 'Mnemonic Search',
+        description: 'Search indexed markdown knowledge bases with BM25 full-text search. Fast, no LLM needed.',
+        promptSnippet: 'Search local markdown knowledge bases',
+        promptGuidelines: [
+            'Use mnemonic_search when you need to find information in the user\'s indexed markdown notes, docs, or wikis before resorting to web search.',
+            'Use mnemonic_query for semantic/conceptual searches; prefer mnemonic_search when you know exact terms.',
+        ],
+        parameters: Type.Object({
+            query: Type.String({ description: 'Search query (keywords or phrases)' }),
+            collection: Type.Optional(Type.String({ description: 'Filter by collection name' })),
+            limit: Type.Optional(Type.Number({ default: 10, description: 'Number of results' })),
+        }),
+        async execute(toolCallId, params, signal, onUpdate, ctx) {
+            const { query, collection, limit } = params;
+            const { MnemonicDB } = await import('../store/database.js');
+            const { SearchPipeline } = await import('../search/pipeline.js');
+            const db = new MnemonicDB(dbPath);
+            db.init();
+            const pipeline = new SearchPipeline(db);
+            const results = pipeline.searchLex(query, {
+                collection: collection ? [collection] : undefined,
+                limit: limit ?? 10,
+            });
+            db.close();
+            const text = results.length === 0
+                ? 'No results found.'
+                : results.map((r, i) => `${i + 1}. ${r.path}  #${r.docid}\n   Title: ${r.title}\n   Score: ${Math.round(r.score * 100)}%\n   ${r.snippet || ''}`).join('\n\n');
+            return {
+                content: [{ type: 'text', text }],
+                details: { count: results.length, results },
+            };
+        },
+    });
+    // ─── Tool: mnemonic_query ─────────────────────────────────────
+    pi.registerTool({
+        name: 'mnemonic_query',
+        label: 'Mnemonic Query',
+        description: 'Hybrid semantic search over indexed markdown. Uses BM25 + vector search + optional LLM reranking for best quality. Use when the user asks conceptual questions or doesn\'t use exact keywords.',
+        promptSnippet: 'Hybrid semantic search over markdown knowledge bases',
+        promptGuidelines: [
+            'Use mnemonic_query for conceptual or indirect questions where exact terms aren\'t known.',
+            'Prefer mnemonic_query over mnemonic_search for most questions — it returns better results.',
+            'Supply the intent parameter when the user\'s wording is ambiguous to disambiguate between nearby concepts.',
+        ],
+        parameters: Type.Object({
+            query: Type.String({ description: 'Natural language search query' }),
+            intent: Type.Optional(Type.String({ description: 'Disambiguation context. States what you want to find AND what to avoid.' })),
+            collection: Type.Optional(Type.String({ description: 'Filter by collection name' })),
+            limit: Type.Optional(Type.Number({ default: 10, description: 'Number of results' })),
+            rerank: Type.Optional(Type.Boolean({ default: true, description: 'Run LLM reranking for better quality' })),
+        }),
+        async execute(toolCallId, params, signal, onUpdate, ctx) {
+            const { query, intent, collection, limit, rerank } = params;
+            const { MnemonicDB } = await import('../store/database.js');
+            const { SearchPipeline } = await import('../search/pipeline.js');
+            const { detectLLMBackend } = await import('../llm/factory.js');
+            const db = new MnemonicDB(dbPath);
+            db.init();
+            await db.loadVectors();
+            const pipeline = new SearchPipeline(db);
+            // Try to get LLM for reranking
+            try {
+                const { backend } = await detectLLMBackend();
+                pipeline.setLLM(backend);
+            }
+            catch {
+                // No LLM — search will fall back to BM25 only
+            }
+            const results = await pipeline.search({
+                query,
+                intent,
+                collection: collection ? [collection] : undefined,
+                limit: limit ?? 10,
+                rerank: rerank !== false,
+                expand: true,
+                hyde: true,
+            });
+            // Cleanup LLM if loaded
+            if (pipeline.llm) {
+                await pipeline.llm.close();
+            }
+            db.close();
+            const text = results.length === 0
+                ? 'No results found.'
+                : results.map((r, i) => `${i + 1}. ${r.path}  #${r.docid}\n   Title: ${r.title}\n   Score: ${Math.round(r.score * 100)}%${r.context.length ? `\n   Context: ${r.context.join(' > ')}` : ''}${r.tags.length ? `\n   Tags: ${r.tags.join(', ')}` : ''}\n   ${r.snippet || ''}`).join('\n\n');
+            return {
+                content: [{ type: 'text', text }],
+                details: { count: results.length, results },
+            };
+        },
+    });
+    // ─── Tool: mnemonic_get ───────────────────────────────────────
+    pi.registerTool({
+        name: 'mnemonic_get',
+        label: 'Mnemonic Get',
+        description: 'Retrieve a full document from the mnemonic index by path or docid (#abc123). Supports line ranges with :from:count suffix.',
+        promptSnippet: 'Retrieve a full indexed document',
+        promptGuidelines: [
+            'Use mnemonic_get after mnemonic_search or mnemonic_query to retrieve the full content of a matched document before answering.',
+            'Use the :from:count suffix to read specific line ranges instead of fetching full documents.',
+        ],
+        parameters: Type.Object({
+            file: Type.String({ description: 'Path, docid (#abc123), or path:from:count (e.g. #abc123:50:40)' }),
+            maxLines: Type.Optional(Type.Number({ default: 100, description: 'Maximum lines to return' })),
+            fromLine: Type.Optional(Type.Number({ description: 'Start line (1-indexed)' })),
+            lineNumbers: Type.Optional(Type.Boolean({ default: true, description: 'Prefix lines with numbers' })),
+        }),
+        async execute(toolCallId, params, signal, onUpdate, ctx) {
+            const { file, maxLines, fromLine, lineNumbers } = params;
+            const { MnemonicDB } = await import('../store/database.js');
+            const { DocumentStore } = await import('../store/documents.js');
+            const db = new MnemonicDB(dbPath);
+            db.init();
+            const docs = new DocumentStore(db);
+            const result = docs.getBody(file, { fromLine, maxLines });
+            if ('error' in result) {
+                db.close();
+                return {
+                    content: [{ type: 'text', text: `Document not found: ${file}${result.similarFiles?.length ? `\nSimilar: ${result.similarFiles.join(', ')}` : ''}` }],
+                    isError: true,
+                };
+            }
+            const meta = docs.get(file);
+            db.close();
+            let output = '';
+            if (!('error' in meta)) {
+                output += `qmd://${meta.collection}/${meta.path}  #${meta.docid}\n---\n`;
+            }
+            if (lineNumbers !== false) {
+                const start = fromLine ?? 1;
+                result.content.split('\n').forEach((line, i) => {
+                    output += `${start + i}: ${line}\n`;
+                });
+            }
+            else {
+                output += result.content;
+            }
+            return {
+                content: [{ type: 'text', text: output }],
+                details: { totalLines: result.totalLines, docid: ('error' in meta) ? undefined : meta.docid },
+            };
+        },
+    });
+    // ─── Tool: mnemonic_status ─────────────────────────────────────
+    pi.registerTool({
+        name: 'mnemonic_status',
+        label: 'Mnemonic Status',
+        description: 'Show mnemonic index status: collections, document counts, vector status, and cache health.',
+        promptSnippet: 'Show mnemonic knowledge base status',
+        promptGuidelines: [
+            'Use mnemonic_status at the start of a session to check what knowledge bases are available.',
+        ],
+        parameters: Type.Object({}),
+        async execute(toolCallId, params, signal, onUpdate, ctx) {
+            const { MnemonicDB } = await import('../store/database.js');
+            const { CollectionStore } = await import('../store/collections.js');
+            const { existsSync, readFileSync } = await import('node:fs');
+            const db = new MnemonicDB(dbPath);
+            db.init();
+            const collections = new CollectionStore(db);
+            const cols = collections.list();
+            const docCount = db.db.prepare('SELECT COUNT(*) as c FROM documents').get().c;
+            const chunkCount = db.db.prepare('SELECT COUNT(*) as c FROM chunks').get().c;
+            const linkCount = db.db.prepare('SELECT COUNT(*) as c FROM links').get().c;
+            const hasVectors = db.hasVectorIndex();
+            const dbSize = existsSync(dbPath) ? Math.round(readFileSync(dbPath).length / 1024) : 0;
+            db.close();
+            const text = [
+                `Index: ${dbPath}`,
+                `Size: ${dbSize} KB`,
+                `Collections: ${cols.length}`,
+                ...cols.map((c) => `  ${c.name}: ${c.docCount} docs, ${c.activeCount} embedded`),
+                `Total documents: ${docCount}`,
+                `Total chunks: ${chunkCount}`,
+                `Links: ${linkCount}`,
+                `Vectors: ${hasVectors ? 'yes' : 'no'}`,
+            ].join('\n');
+            return {
+                content: [{ type: 'text', text }],
+                details: { collections: cols, docCount, hasVectors },
+            };
+        },
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/pi-extension/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/pi-extension/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,MAAM,CAAC,OAAO,WAAW,EAAgB;IACvC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;IAErE,kEAAkE;IAClE,EAAE,CAAC,YAAY,CAAC;QACd,IAAI,EAAE,iBAAiB;QACvB,KAAK,EAAE,iBAAiB;QACxB,WAAW,EAAE,0FAA0F;QACvG,aAAa,EAAE,uCAAuC;QACtD,gBAAgB,EAAE;YAChB,6IAA6I;YAC7I,wGAAwG;SACzG;QACD,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;YACtB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,oCAAoC,EAAE,CAAC;YACzE,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,2BAA2B,EAAE,CAAC,CAAC;YACpF,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,mBAAmB,EAAE,CAAC,CAAC;SACrF,CAAC;QACF,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG;YACrD,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;YAE5C,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAC5D,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC;YACjE,MAAM,EAAE,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;YAClC,EAAE,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,EAAE,CAAC,CAAC;YACxC,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,KAAK,EAAE;gBACxC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS;gBACjD,KAAK,EAAE,KAAK,IAAI,EAAE;aACnB,CAAC,CAAC;YACH,EAAE,CAAC,KAAK,EAAE,CAAC;YAEX,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,KAAK,CAAC;gBAC/B,CAAC,CAAC,mBAAmB;gBACrB,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACnB,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,KAAK,eAAe,CAAC,CAAC,KAAK,eAAe,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,GAAG,CAAC,SAAS,CAAC,CAAC,OAAO,IAAI,EAAE,EAAE,CACzH,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEnB,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;gBACjC,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE;aAC5C,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,iEAAiE;IACjE,EAAE,CAAC,YAAY,CAAC;QACd,IAAI,EAAE,gBAAgB;QACtB,KAAK,EAAE,gBAAgB;QACvB,WAAW,EAAE,gMAAgM;QAC7M,aAAa,EAAE,sDAAsD;QACrE,gBAAgB,EAAE;YAChB,0FAA0F;YAC1F,4FAA4F;YAC5F,4GAA4G;SAC7G;QACD,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;YACtB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,+BAA+B,EAAE,CAAC;YACpE,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,yEAAyE,EAAE,CAAC,CAAC;YAC9H,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,2BAA2B,EAAE,CAAC,CAAC;YACpF,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,mBAAmB,EAAE,CAAC,CAAC;YACpF,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,sCAAsC,EAAE,CAAC,CAAC;SAC5G,CAAC;QACF,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG;YACrD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;YAE5D,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAC5D,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC;YACjE,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;YAE/D,MAAM,EAAE,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;YAClC,EAAE,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,EAAE,CAAC,WAAW,EAAE,CAAC;YACvB,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC,EAAE,CAAC,CAAC;YAExC,+BAA+B;YAC/B,IAAI,CAAC;gBACH,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,gBAAgB,EAAE,CAAC;gBAC7C,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;YAAC,MAAM,CAAC;gBACP,8CAA8C;YAChD,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC;gBACpC,KAAK;gBACL,MAAM;gBACN,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS;gBACjD,KAAK,EAAE,KAAK,IAAI,EAAE;gBAClB,MAAM,EAAE,MAAM,KAAK,KAAK;gBACxB,MAAM,EAAE,IAAI;gBACZ,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YAEH,wBAAwB;YACxB,IAAK,QAAgB,CAAC,GAAG,EAAE,CAAC;gBAC1B,MAAO,QAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;YACtC,CAAC;YACD,EAAE,CAAC,KAAK,EAAE,CAAC;YAEX,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,KAAK,CAAC;gBAC/B,CAAC,CAAC,mBAAmB;gBACrB,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACnB,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,KAAK,eAAe,CAAC,CAAC,KAAK,eAAe,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,OAAO,IAAI,EAAE,EAAE,CACrP,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEnB,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;gBACjC,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE;aAC5C,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,iEAAiE;IACjE,EAAE,CAAC,YAAY,CAAC;QACd,IAAI,EAAE,cAAc;QACpB,KAAK,EAAE,cAAc;QACrB,WAAW,EAAE,4HAA4H;QACzI,aAAa,EAAE,kCAAkC;QACjD,gBAAgB,EAAE;YAChB,+HAA+H;YAC/H,6FAA6F;SAC9F;QACD,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;YACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,gEAAgE,EAAE,CAAC;YACpG,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,yBAAyB,EAAE,CAAC,CAAC;YAC9F,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,wBAAwB,EAAE,CAAC,CAAC;YAC/E,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,2BAA2B,EAAE,CAAC,CAAC;SACtG,CAAC;QACF,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG;YACrD,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;YAEzD,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAC5D,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC;YAChE,MAAM,EAAE,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;YAClC,EAAE,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,GAAG,IAAI,aAAa,CAAC,EAAE,CAAC,CAAC;YAEnC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;YAE1D,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;gBACtB,EAAE,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,uBAAuB,IAAI,GAAI,MAAc,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC,cAAe,MAAc,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC;oBACtK,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAQ,CAAC;YACnC,EAAE,CAAC,KAAK,EAAE,CAAC;YAEX,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,IAAI,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,EAAE,CAAC;gBACvB,MAAM,IAAI,SAAS,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,IAAI,MAAM,IAAI,CAAC,KAAK,SAAS,CAAC;YAC3E,CAAC;YAED,IAAI,WAAW,KAAK,KAAK,EAAE,CAAC;gBAC1B,MAAM,KAAK,GAAG,QAAQ,IAAI,CAAC,CAAC;gBAC5B,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,IAAY,EAAE,CAAS,EAAE,EAAE;oBAC7D,MAAM,IAAI,GAAG,KAAK,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC;gBACtC,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,MAAM,CAAC,OAAO,CAAC;YAC3B,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;gBACzC,OAAO,EAAE,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,EAAE;aAC9F,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,kEAAkE;IAClE,EAAE,CAAC,YAAY,CAAC;QACd,IAAI,EAAE,iBAAiB;QACvB,KAAK,EAAE,iBAAiB;QACxB,WAAW,EAAE,4FAA4F;QACzG,aAAa,EAAE,qCAAqC;QACpD,gBAAgB,EAAE;YAChB,4FAA4F;SAC7F;QACD,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3B,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG;YACrD,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;YAC5D,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;YACpE,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;YAE7D,MAAM,EAAE,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;YAClC,EAAE,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,WAAW,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;YAC5C,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC;YAEhC,MAAM,QAAQ,GAAI,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC,GAAG,EAAU,CAAC,CAAC,CAAC;YACvF,MAAM,UAAU,GAAI,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC,GAAG,EAAU,CAAC,CAAC,CAAC;YACtF,MAAM,SAAS,GAAI,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,iCAAiC,CAAC,CAAC,GAAG,EAAU,CAAC,CAAC,CAAC;YACpF,MAAM,UAAU,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC;YACvC,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAEvF,EAAE,CAAC,KAAK,EAAE,CAAC;YAEX,MAAM,IAAI,GAAG;gBACX,UAAU,MAAM,EAAE;gBAClB,SAAS,MAAM,KAAK;gBACpB,gBAAgB,IAAI,CAAC,MAAM,EAAE;gBAC7B,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,QAAQ,UAAU,CAAC,CAAC,WAAW,WAAW,CAAC;gBAChF,oBAAoB,QAAQ,EAAE;gBAC9B,iBAAiB,UAAU,EAAE;gBAC7B,UAAU,SAAS,EAAE;gBACrB,YAAY,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE;aACxC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEb,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;gBACjC,OAAO,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE;aACrD,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/search/fts.d.ts

@@ -0,0 +1,17 @@
+import type { MnemonicDB } from '../store/database.js';
+import type { SearchResult } from '../types.js';
+export declare class FTSSearch {
+    private db;
+    constructor(db: MnemonicDB);
+    /** BM25 search via FTS5 */
+    search(query: string, options?: {
+        collection?: string[];
+        limit?: number;
+        minScore?: number;
+    }): SearchResult[];
+    /** Fallback to LIKE-based search when FTS5 syntax fails */
+    private fallbackSearch;
+    /** Clean a query for FTS5 syntax */
+    private cleanQuery;
+}
+//# sourceMappingURL=fts.d.ts.map

package/dist/search/fts.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fts.d.ts","sourceRoot":"","sources":["../../src/search/fts.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAEhD,qBAAa,SAAS;IACR,OAAO,CAAC,EAAE;gBAAF,EAAE,EAAE,UAAU;IAElC,2BAA2B;IAC3B,MAAM,CACJ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE;QACP,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;QACtB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,QAAQ,CAAC,EAAE,MAAM,CAAC;KACd,GACL,YAAY,EAAE;IA8DjB,2DAA2D;IAC3D,OAAO,CAAC,cAAc;IA2DtB,oCAAoC;IACpC,OAAO,CAAC,UAAU;CAcnB"}