@ixxrllg/localmind 0.1.0

package/dist/mcp/server.js

@@ -0,0 +1,291 @@
+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 { z } from "zod";
+import { ask } from "../ask/answer.js";
+import { loadConfig } from "../core/config.js";
+import { closeDb, openDb } from "../core/db.js";
+import { resolveRoot } from "../core/paths.js";
+import { findRelatedFiles } from "../graph/related.js";
+import { runIndex } from "../indexer/indexer.js";
+import { search } from "../search/search.js";
+import { getStatus, formatBytes } from "../status.js";
+const SearchInput = z.object({
+    query: z.string().min(1),
+    limit: z.number().int().min(1).max(50).optional(),
+});
+const AskInput = z.object({
+    question: z.string().min(1),
+    limit: z.number().int().min(1).max(20).optional(),
+});
+const RelatedInput = z.object({
+    path: z.string().min(1),
+    limit: z.number().int().min(1).max(50).optional(),
+});
+const ExplainInput = z.object({
+    path: z.string().min(1),
+    startLine: z.number().int().min(1).optional(),
+    endLine: z.number().int().min(1).optional(),
+});
+export async function startMcpServer(options = {}) {
+    const root = resolveRoot(options.root);
+    const config = loadConfig(root);
+    const server = new Server({
+        name: "localmind",
+        version: "0.1.0",
+    }, {
+        capabilities: {
+            tools: {},
+        },
+    });
+    server.setRequestHandler(ListToolsRequestSchema, async () => ({
+        tools: [
+            {
+                name: "search_code",
+                description: "Search the LocalMind index (BM25/hybrid) for code and docs relevant to a query. Returns ranked chunks with paths and line ranges.",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        query: { type: "string", description: "Natural language or symbol query" },
+                        limit: { type: "number", description: "Max results (default 8)" },
+                    },
+                    required: ["query"],
+                },
+            },
+            {
+                name: "ask_codebase",
+                description: "Ask a question about the indexed codebase. Uses local retrieval; generative answers if a provider is configured.",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        question: { type: "string" },
+                        limit: { type: "number" },
+                    },
+                    required: ["question"],
+                },
+            },
+            {
+                name: "find_related_files",
+                description: "Find files related via import/link graph edges for a given path or module name.",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        path: { type: "string", description: "File path or import fragment" },
+                        limit: { type: "number" },
+                    },
+                    required: ["path"],
+                },
+            },
+            {
+                name: "explain_file",
+                description: "Return indexed content for a file (optionally a line range) from the LocalMind database.",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        path: { type: "string" },
+                        startLine: { type: "number" },
+                        endLine: { type: "number" },
+                    },
+                    required: ["path"],
+                },
+            },
+            {
+                name: "index_status",
+                description: "Show LocalMind index statistics for the current project.",
+                inputSchema: {
+                    type: "object",
+                    properties: {},
+                },
+            },
+            {
+                name: "reindex",
+                description: "Re-index the project (incremental by default). Use full=true to rebuild everything.",
+                inputSchema: {
+                    type: "object",
+                    properties: {
+                        full: { type: "boolean" },
+                    },
+                },
+            },
+        ],
+    }));
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        const name = request.params.name;
+        const args = (request.params.arguments ?? {});
+        try {
+            switch (name) {
+                case "search_code": {
+                    const input = SearchInput.parse(args);
+                    const db = openDb(root);
+                    try {
+                        const hits = await search(db, input.query, {
+                            limit: input.limit ?? 8,
+                            provider: config.provider,
+                            mode: config.provider === "none" ? "bm25" : "hybrid",
+                        });
+                        return {
+                            content: [
+                                {
+                                    type: "text",
+                                    text: JSON.stringify(hits, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    finally {
+                        closeDb(db);
+                    }
+                }
+                case "ask_codebase": {
+                    const input = AskInput.parse(args);
+                    const db = openDb(root);
+                    try {
+                        const result = await ask(db, config, input.question, {
+                            limit: input.limit,
+                        });
+                        return {
+                            content: [
+                                {
+                                    type: "text",
+                                    text: JSON.stringify({
+                                        answer: result.answer,
+                                        mode: result.mode,
+                                        provider: result.provider,
+                                        tookMs: result.tookMs,
+                                        citations: result.hits.map((h) => ({
+                                            path: h.path,
+                                            startLine: h.startLine,
+                                            endLine: h.endLine,
+                                            score: h.score,
+                                        })),
+                                    }, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    finally {
+                        closeDb(db);
+                    }
+                }
+                case "find_related_files": {
+                    const input = RelatedInput.parse(args);
+                    const db = openDb(root);
+                    try {
+                        const related = findRelatedFiles(db, input.path, input.limit ?? 15);
+                        return {
+                            content: [
+                                {
+                                    type: "text",
+                                    text: JSON.stringify(related, null, 2),
+                                },
+                            ],
+                        };
+                    }
+                    finally {
+                        closeDb(db);
+                    }
+                }
+                case "explain_file": {
+                    const input = ExplainInput.parse(args);
+                    const db = openDb(root);
+                    try {
+                        const pathQ = input.path.replace(/\\/g, "/");
+                        const file = db
+                            .prepare(`SELECT id, path, language, line_count FROM files
+                 WHERE path = ? OR path LIKE ? OR path LIKE ?
+                 ORDER BY CASE WHEN path = ? THEN 0 ELSE 1 END
+                 LIMIT 1`)
+                            .get(pathQ, `%/${pathQ}`, `%${pathQ}%`, pathQ);
+                        if (!file) {
+                            return {
+                                content: [
+                                    {
+                                        type: "text",
+                                        text: `File not found in index: ${input.path}`,
+                                    },
+                                ],
+                                isError: true,
+                            };
+                        }
+                        const chunks = db
+                            .prepare(`SELECT start_line, end_line, content FROM chunks
+                 WHERE file_id = ?
+                 ORDER BY chunk_index`)
+                            .all(file.id);
+                        const start = input.startLine ?? 1;
+                        const end = input.endLine ?? file.line_count;
+                        const filtered = chunks.filter((c) => c.end_line >= start && c.start_line <= end);
+                        const text = [
+                            `# ${file.path} (${file.language})`,
+                            `Lines ${start}–${end}`,
+                            "",
+                            ...filtered.map((c) => `## L${c.start_line}–${c.end_line}\n\`\`\`\n${c.content}\n\`\`\``),
+                        ].join("\n");
+                        return {
+                            content: [{ type: "text", text }],
+                        };
+                    }
+                    finally {
+                        closeDb(db);
+                    }
+                }
+                case "index_status": {
+                    const status = getStatus(root);
+                    const text = [
+                        `Root: ${status.root}`,
+                        `Files: ${status.fileCount}`,
+                        `Chunks: ${status.chunkCount}`,
+                        `Terms: ${status.termCount}`,
+                        `Edges: ${status.edgeCount}`,
+                        `Embeddings: ${status.embeddingCount}`,
+                        `DB: ${formatBytes(status.dbSizeBytes)}`,
+                        `Last indexed: ${status.lastIndexedAt ?? "never"}`,
+                        `Provider: ${status.config.provider}`,
+                    ].join("\n");
+                    return {
+                        content: [{ type: "text", text }],
+                    };
+                }
+                case "reindex": {
+                    const full = Boolean(args.full);
+                    const stats = await runIndex(config, {
+                        full,
+                        quiet: true,
+                    });
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: JSON.stringify(stats, null, 2),
+                            },
+                        ],
+                    };
+                }
+                default:
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: `Unknown tool: ${name}`,
+                            },
+                        ],
+                        isError: true,
+                    };
+            }
+        }
+        catch (err) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: err instanceof Error ? err.message : String(err),
+                    },
+                ],
+                isError: true,
+            };
+        }
+    });
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+}
+//# 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,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,GAAG,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAEtD,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACxB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE;CAClD,CAAC,CAAC;AAEH,MAAM,QAAQ,GAAG,CAAC,CAAC,MAAM,CAAC;IACxB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IAC3B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE;CAClD,CAAC,CAAC;AAEH,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACvB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE;CAClD,CAAC,CAAC;AAEH,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACvB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;IAC7C,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;CAC5C,CAAC,CAAC;AAMH,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,UAA4B,EAAE;IACjE,MAAM,IAAI,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACvC,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;IAEhC,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB;QACE,IAAI,EAAE,WAAW;QACjB,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,aAAa;gBACnB,WAAW,EACT,mIAAmI;gBACrI,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,kCAAkC,EAAE;wBAC1E,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,yBAAyB,EAAE;qBAClE;oBACD,QAAQ,EAAE,CAAC,OAAO,CAAC;iBACpB;aACF;YACD;gBACE,IAAI,EAAE,cAAc;gBACpB,WAAW,EACT,kHAAkH;gBACpH,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;wBAC5B,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;qBAC1B;oBACD,QAAQ,EAAE,CAAC,UAAU,CAAC;iBACvB;aACF;YACD;gBACE,IAAI,EAAE,oBAAoB;gBAC1B,WAAW,EACT,iFAAiF;gBACnF,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,8BAA8B,EAAE;wBACrE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;qBAC1B;oBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;iBACnB;aACF;YACD;gBACE,IAAI,EAAE,cAAc;gBACpB,WAAW,EACT,0FAA0F;gBAC5F,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;wBACxB,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;wBAC7B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;qBAC5B;oBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;iBACnB;aACF;YACD;gBACE,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,0DAA0D;gBACvE,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,EAAE;iBACf;aACF;YACD;gBACE,IAAI,EAAE,SAAS;gBACf,WAAW,EACT,qFAAqF;gBACvF,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,IAAI,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;qBAC1B;iBACF;aACF;SACF;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;QACjC,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,IAAI,EAAE,CAA4B,CAAC;QAEzE,IAAI,CAAC;YACH,QAAQ,IAAI,EAAE,CAAC;gBACb,KAAK,aAAa,CAAC,CAAC,CAAC;oBACnB,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACtC,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;oBACxB,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,KAAK,EAAE;4BACzC,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,CAAC;4BACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;4BACzB,IAAI,EAAE,MAAM,CAAC,QAAQ,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ;yBACrD,CAAC,CAAC;wBACH,OAAO;4BACL,OAAO,EAAE;gCACP;oCACE,IAAI,EAAE,MAAe;oCACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;iCACpC;6BACF;yBACF,CAAC;oBACJ,CAAC;4BAAS,CAAC;wBACT,OAAO,CAAC,EAAE,CAAC,CAAC;oBACd,CAAC;gBACH,CAAC;gBACD,KAAK,cAAc,CAAC,CAAC,CAAC;oBACpB,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACnC,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;oBACxB,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,QAAQ,EAAE;4BACnD,KAAK,EAAE,KAAK,CAAC,KAAK;yBACnB,CAAC,CAAC;wBACH,OAAO;4BACL,OAAO,EAAE;gCACP;oCACE,IAAI,EAAE,MAAe;oCACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAClB;wCACE,MAAM,EAAE,MAAM,CAAC,MAAM;wCACrB,IAAI,EAAE,MAAM,CAAC,IAAI;wCACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;wCACzB,MAAM,EAAE,MAAM,CAAC,MAAM;wCACrB,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;4CACjC,IAAI,EAAE,CAAC,CAAC,IAAI;4CACZ,SAAS,EAAE,CAAC,CAAC,SAAS;4CACtB,OAAO,EAAE,CAAC,CAAC,OAAO;4CAClB,KAAK,EAAE,CAAC,CAAC,KAAK;yCACf,CAAC,CAAC;qCACJ,EACD,IAAI,EACJ,CAAC,CACF;iCACF;6BACF;yBACF,CAAC;oBACJ,CAAC;4BAAS,CAAC;wBACT,OAAO,CAAC,EAAE,CAAC,CAAC;oBACd,CAAC;gBACH,CAAC;gBACD,KAAK,oBAAoB,CAAC,CAAC,CAAC;oBAC1B,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACvC,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;oBACxB,IAAI,CAAC;wBACH,MAAM,OAAO,GAAG,gBAAgB,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;wBACpE,OAAO;4BACL,OAAO,EAAE;gCACP;oCACE,IAAI,EAAE,MAAe;oCACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;iCACvC;6BACF;yBACF,CAAC;oBACJ,CAAC;4BAAS,CAAC;wBACT,OAAO,CAAC,EAAE,CAAC,CAAC;oBACd,CAAC;gBACH,CAAC;gBACD,KAAK,cAAc,CAAC,CAAC,CAAC;oBACpB,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACvC,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;oBACxB,IAAI,CAAC;wBACH,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;wBAC7C,MAAM,IAAI,GAAG,EAAE;6BACZ,OAAO,CACN;;;yBAGS,CACV;6BACA,GAAG,CAAC,KAAK,EAAE,KAAK,KAAK,EAAE,EAAE,IAAI,KAAK,GAAG,EAAE,KAAK,CAOlC,CAAC;wBAEd,IAAI,CAAC,IAAI,EAAE,CAAC;4BACV,OAAO;gCACL,OAAO,EAAE;oCACP;wCACE,IAAI,EAAE,MAAe;wCACrB,IAAI,EAAE,4BAA4B,KAAK,CAAC,IAAI,EAAE;qCAC/C;iCACF;gCACD,OAAO,EAAE,IAAI;6BACd,CAAC;wBACJ,CAAC;wBAED,MAAM,MAAM,GAAG,EAAE;6BACd,OAAO,CACN;;sCAEsB,CACvB;6BACA,GAAG,CAAC,IAAI,CAAC,EAAE,CAIZ,CAAC;wBAEH,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC;wBACnC,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,UAAU,CAAC;wBAC7C,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAC5B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,IAAI,KAAK,IAAI,CAAC,CAAC,UAAU,IAAI,GAAG,CAClD,CAAC;wBAEF,MAAM,IAAI,GAAG;4BACX,KAAK,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,QAAQ,GAAG;4BACnC,SAAS,KAAK,IAAI,GAAG,EAAE;4BACvB,EAAE;4BACF,GAAG,QAAQ,CAAC,GAAG,CACb,CAAC,CAAC,EAAE,EAAE,CACJ,OAAO,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC,QAAQ,aAAa,CAAC,CAAC,OAAO,UAAU,CACpE;yBACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAEb,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC;yBAC3C,CAAC;oBACJ,CAAC;4BAAS,CAAC;wBACT,OAAO,CAAC,EAAE,CAAC,CAAC;oBACd,CAAC;gBACH,CAAC;gBACD,KAAK,cAAc,CAAC,CAAC,CAAC;oBACpB,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;oBAC/B,MAAM,IAAI,GAAG;wBACX,SAAS,MAAM,CAAC,IAAI,EAAE;wBACtB,UAAU,MAAM,CAAC,SAAS,EAAE;wBAC5B,WAAW,MAAM,CAAC,UAAU,EAAE;wBAC9B,UAAU,MAAM,CAAC,SAAS,EAAE;wBAC5B,UAAU,MAAM,CAAC,SAAS,EAAE;wBAC5B,eAAe,MAAM,CAAC,cAAc,EAAE;wBACtC,OAAO,WAAW,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE;wBACxC,iBAAiB,MAAM,CAAC,aAAa,IAAI,OAAO,EAAE;wBAClD,aAAa,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE;qBACtC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACb,OAAO;wBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC;qBAC3C,CAAC;gBACJ,CAAC;gBACD,KAAK,SAAS,CAAC,CAAC,CAAC;oBACf,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAChC,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,MAAM,EAAE;wBACnC,IAAI;wBACJ,KAAK,EAAE,IAAI;qBACZ,CAAC,CAAC;oBACH,OAAO;wBACL,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,MAAe;gCACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;6BACrC;yBACF;qBACF,CAAC;gBACJ,CAAC;gBACD;oBACE,OAAO;wBACL,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,MAAe;gCACrB,IAAI,EAAE,iBAAiB,IAAI,EAAE;6BAC9B;yBACF;wBACD,OAAO,EAAE,IAAI;qBACd,CAAC;YACN,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;qBACvD;iBACF;gBACD,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AAClC,CAAC"}

package/dist/search/bm25.d.ts

@@ -0,0 +1,9 @@
+import type { Db } from "../core/db.js";
+import type { SearchHit } from "../types.js";
+/**
+ * BM25 search over the inverted index stored in SQLite.
+ */
+export declare function bm25Search(db: Db, query: string, limit?: number): SearchHit[];
+/** Fallback substring search when BM25 finds nothing (e.g. exact symbols). */
+export declare function literalSearch(db: Db, query: string, limit?: number): SearchHit[];
+//# sourceMappingURL=bm25.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"bm25.d.ts","sourceRoot":"","sources":["../../src/search/bm25.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAY7C;;GAEG;AACH,wBAAgB,UAAU,CACxB,EAAE,EAAE,EAAE,EACN,KAAK,EAAE,MAAM,EACb,KAAK,SAAK,GACT,SAAS,EAAE,CA6Db;AA0CD,8EAA8E;AAC9E,wBAAgB,aAAa,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,SAAK,GAAG,SAAS,EAAE,CAmC5E"}

package/dist/search/bm25.js

@@ -0,0 +1,109 @@
+import { tokenize } from "./tokenize.js";
+/** Classic BM25 parameters. */
+const K1 = 1.2;
+const B = 0.75;
+/**
+ * BM25 search over the inverted index stored in SQLite.
+ */
+export function bm25Search(db, query, limit = 10) {
+    const terms = [...new Set(tokenize(query))];
+    if (!terms.length)
+        return [];
+    const totalDocsRow = db.prepare(`SELECT COUNT(*) AS c FROM chunks`).get();
+    const N = Number(totalDocsRow.c);
+    if (N === 0)
+        return [];
+    const avgRow = db
+        .prepare(`SELECT AVG(token_count) AS avg_len FROM chunks`)
+        .get();
+    const avgdl = Number(avgRow.avg_len) || 1;
+    const scores = new Map();
+    const termStmt = db.prepare(`SELECT id FROM terms WHERE term = ?`);
+    const dfStmt = db.prepare(`SELECT COUNT(*) AS df FROM postings WHERE term_id = ?`);
+    const postingsStmt = db.prepare(`SELECT p.chunk_id AS chunk_id, p.tf AS tf, c.token_count AS dl
+     FROM postings p
+     JOIN chunks c ON c.id = p.chunk_id
+     WHERE p.term_id = ?`);
+    for (const term of terms) {
+        const termRow = termStmt.get(term);
+        if (!termRow)
+            continue;
+        const dfRow = dfStmt.get(termRow.id);
+        const df = Number(dfRow.df);
+        if (df === 0)
+            continue;
+        const idf = Math.log(1 + (N - df + 0.5) / (df + 0.5));
+        const postings = postingsStmt.all(termRow.id);
+        for (const p of postings) {
+            const tf = Number(p.tf);
+            const dl = Number(p.dl) || 1;
+            const denom = tf + K1 * (1 - B + B * (dl / avgdl));
+            const score = idf * ((tf * (K1 + 1)) / denom);
+            scores.set(p.chunk_id, (scores.get(p.chunk_id) ?? 0) + score);
+        }
+    }
+    const ranked = [...scores.entries()]
+        .map(([chunkId, score]) => ({ chunkId, score }))
+        .sort((a, b) => b.score - a.score)
+        .slice(0, limit);
+    if (!ranked.length)
+        return [];
+    return hydrateHits(db, ranked);
+}
+function hydrateHits(db, ranked) {
+    const stmt = db.prepare(`SELECT c.id AS chunk_id, c.file_id AS file_id, c.start_line AS start_line,
+            c.end_line AS end_line, c.content AS content,
+            f.path AS path, f.language AS language
+     FROM chunks c
+     JOIN files f ON f.id = c.file_id
+     WHERE c.id = ?`);
+    const hits = [];
+    let rank = 1;
+    for (const r of ranked) {
+        const row = stmt.get(r.chunkId);
+        if (!row)
+            continue;
+        hits.push({
+            chunkId: row.chunk_id,
+            fileId: row.file_id,
+            path: row.path,
+            language: row.language,
+            startLine: row.start_line,
+            endLine: row.end_line,
+            content: row.content,
+            score: r.score,
+            rank: rank++,
+        });
+    }
+    return hits;
+}
+/** Fallback substring search when BM25 finds nothing (e.g. exact symbols). */
+export function literalSearch(db, query, limit = 10) {
+    const q = query.trim();
+    if (!q)
+        return [];
+    const rows = db
+        .prepare(`SELECT c.id AS chunk_id, c.file_id AS file_id, c.start_line AS start_line,
+              c.end_line AS end_line, c.content AS content,
+              f.path AS path, f.language AS language
+       FROM chunks c
+       JOIN files f ON f.id = c.file_id
+       WHERE c.content LIKE ? ESCAPE '\\'
+       LIMIT ?`)
+        .all(`%${escapeLike(q)}%`, limit);
+    return rows.map((row, i) => ({
+        chunkId: row.chunk_id,
+        fileId: row.file_id,
+        path: row.path,
+        language: row.language,
+        startLine: row.start_line,
+        endLine: row.end_line,
+        content: row.content,
+        score: 1 / (i + 1),
+        rank: i + 1,
+    }));
+}
+function escapeLike(s) {
+    return s.replace(/[\\%_]/g, (ch) => `\\${ch}`);
+}
+//# sourceMappingURL=bm25.js.map

package/dist/search/bm25.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bm25.js","sourceRoot":"","sources":["../../src/search/bm25.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAEzC,+BAA+B;AAC/B,MAAM,EAAE,GAAG,GAAG,CAAC;AACf,MAAM,CAAC,GAAG,IAAI,CAAC;AAOf;;GAEG;AACH,MAAM,UAAU,UAAU,CACxB,EAAM,EACN,KAAa,EACb,KAAK,GAAG,EAAE;IAEV,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC5C,IAAI,CAAC,KAAK,CAAC,MAAM;QAAE,OAAO,EAAE,CAAC;IAE7B,MAAM,YAAY,GAAG,EAAE,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC,GAAG,EAEtE,CAAC;IACF,MAAM,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACjC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAEvB,MAAM,MAAM,GAAG,EAAE;SACd,OAAO,CAAC,gDAAgD,CAAC;SACzD,GAAG,EAAgC,CAAC;IACvC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAE1C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEzC,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC;IACnE,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,CACvB,uDAAuD,CACxD,CAAC;IACF,MAAM,YAAY,GAAG,EAAE,CAAC,OAAO,CAC7B;;;yBAGqB,CACtB,CAAC;IAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAA+B,CAAC;QACjE,IAAI,CAAC,OAAO;YAAE,SAAS;QAEvB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAmB,CAAC;QACvD,MAAM,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC5B,IAAI,EAAE,KAAK,CAAC;YAAE,SAAS;QAEvB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,GAAG,CAAC,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC;QAEtD,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAI1C,CAAC;QAEH,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACxB,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,KAAK,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC;YACnD,MAAM,KAAK,GAAG,GAAG,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;YAC9C,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAkB,CAAC,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;SAChD,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;SAC/C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;SACjC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAEnB,IAAI,CAAC,MAAM,CAAC,MAAM;QAAE,OAAO,EAAE,CAAC;IAE9B,OAAO,WAAW,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,WAAW,CAAC,EAAM,EAAE,MAAqB;IAChD,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CACrB;;;;;oBAKgB,CACjB,CAAC;IAEF,MAAM,IAAI,GAAgB,EAAE,CAAC;IAC7B,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAUjB,CAAC;QACd,IAAI,CAAC,GAAG;YAAE,SAAS;QACnB,IAAI,CAAC,IAAI,CAAC;YACR,OAAO,EAAE,GAAG,CAAC,QAAQ;YACrB,MAAM,EAAE,GAAG,CAAC,OAAO;YACnB,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,OAAO,EAAE,GAAG,CAAC,QAAQ;YACrB,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,IAAI,EAAE,IAAI,EAAE;SACb,CAAC,CAAC;IACL,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,8EAA8E;AAC9E,MAAM,UAAU,aAAa,CAAC,EAAM,EAAE,KAAa,EAAE,KAAK,GAAG,EAAE;IAC7D,MAAM,CAAC,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IACvB,IAAI,CAAC,CAAC;QAAE,OAAO,EAAE,CAAC;IAElB,MAAM,IAAI,GAAG,EAAE;SACZ,OAAO,CACN;;;;;;eAMS,CACV;SACA,GAAG,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAQhC,CAAC;IAEH,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAC3B,OAAO,EAAE,GAAG,CAAC,QAAQ;QACrB,MAAM,EAAE,GAAG,CAAC,OAAO;QACnB,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,SAAS,EAAE,GAAG,CAAC,UAAU;QACzB,OAAO,EAAE,GAAG,CAAC,QAAQ;QACrB,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,KAAK,EAAE,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;QAClB,IAAI,EAAE,CAAC,GAAG,CAAC;KACZ,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,UAAU,CAAC,CAAS;IAC3B,OAAO,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;AACjD,CAAC"}

package/dist/search/embeddings.d.ts

@@ -0,0 +1,15 @@
+import type { Db } from "../core/db.js";
+import type { ProviderKind, SearchHit } from "../types.js";
+export interface Embedder {
+    readonly model: string;
+    readonly provider: ProviderKind;
+    embed(texts: string[]): Promise<Float32Array[]>;
+}
+export declare function createEmbedder(provider: ProviderKind): Embedder | null;
+export declare function storeEmbeddingsForChunks(db: Db, embedder: Embedder, items: Array<{
+    chunkId: number;
+    content: string;
+}>, batchSize?: number): Promise<number>;
+export declare function vectorSearch(db: Db, embedder: Embedder, query: string, limit?: number): Promise<SearchHit[]>;
+export declare function mergeHybrid(bm25Hits: SearchHit[], vectorHits: SearchHit[], limit: number): SearchHit[];
+//# sourceMappingURL=embeddings.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.d.ts","sourceRoot":"","sources":["../../src/search/embeddings.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,eAAe,CAAC;AAExC,OAAO,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAE3D,MAAM,WAAW,QAAQ;IACvB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,QAAQ,EAAE,YAAY,CAAC;IAChC,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;CACjD;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,YAAY,GAAG,QAAQ,GAAG,IAAI,CAYtE;AA+ED,wBAAsB,wBAAwB,CAC5C,EAAE,EAAE,EAAE,EACN,QAAQ,EAAE,QAAQ,EAClB,KAAK,EAAE,KAAK,CAAC;IAAE,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC,EAClD,SAAS,SAAK,GACb,OAAO,CAAC,MAAM,CAAC,CAcjB;AAkBD,wBAAsB,YAAY,CAChC,EAAE,EAAE,EAAE,EACN,QAAQ,EAAE,QAAQ,EAClB,KAAK,EAAE,MAAM,EACb,KAAK,SAAK,GACT,OAAO,CAAC,SAAS,EAAE,CAAC,CAoDtB;AAED,wBAAgB,WAAW,CACzB,QAAQ,EAAE,SAAS,EAAE,EACrB,UAAU,EAAE,SAAS,EAAE,EACvB,KAAK,EAAE,MAAM,GACZ,SAAS,EAAE,CA+Cb"}

package/dist/search/embeddings.js

@@ -0,0 +1,193 @@
+import { insertEmbedding } from "../core/db.js";
+export function createEmbedder(provider) {
+    if (provider === "openai" && process.env.OPENAI_API_KEY) {
+        return new OpenAIEmbedder();
+    }
+    if (provider === "ollama") {
+        return new OllamaEmbedder();
+    }
+    // Auto-detect openai key even if config says none? Keep explicit.
+    if (provider === "none" && process.env.OPENAI_API_KEY) {
+        return new OpenAIEmbedder();
+    }
+    return null;
+}
+class OpenAIEmbedder {
+    provider = "openai";
+    model;
+    baseUrl;
+    apiKey;
+    constructor() {
+        this.model =
+            process.env.OPENAI_EMBEDDING_MODEL?.trim() || "text-embedding-3-small";
+        this.baseUrl = (process.env.OPENAI_BASE_URL?.trim() || "https://api.openai.com/v1").replace(/\/$/, "");
+        this.apiKey = process.env.OPENAI_API_KEY ?? "";
+    }
+    async embed(texts) {
+        if (!texts.length)
+            return [];
+        const res = await fetch(`${this.baseUrl}/embeddings`, {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${this.apiKey}`,
+            },
+            body: JSON.stringify({
+                model: this.model,
+                input: texts.map((t) => t.slice(0, 8000)),
+            }),
+        });
+        if (!res.ok) {
+            const body = await res.text();
+            throw new Error(`OpenAI embeddings failed (${res.status}): ${body}`);
+        }
+        const data = (await res.json());
+        const sorted = [...data.data].sort((a, b) => a.index - b.index);
+        return sorted.map((d) => Float32Array.from(d.embedding));
+    }
+}
+class OllamaEmbedder {
+    provider = "ollama";
+    model;
+    baseUrl;
+    constructor() {
+        this.model =
+            process.env.OLLAMA_EMBEDDING_MODEL?.trim() || "nomic-embed-text";
+        this.baseUrl = (process.env.OLLAMA_BASE_URL?.trim() ||
+            process.env.OLLAMA_HOST?.trim() ||
+            "http://127.0.0.1:11434").replace(/\/$/, "");
+    }
+    async embed(texts) {
+        const out = [];
+        for (const text of texts) {
+            const res = await fetch(`${this.baseUrl}/api/embeddings`, {
+                method: "POST",
+                headers: { "Content-Type": "application/json" },
+                body: JSON.stringify({
+                    model: this.model,
+                    prompt: text.slice(0, 8000),
+                }),
+            });
+            if (!res.ok) {
+                const body = await res.text();
+                throw new Error(`Ollama embeddings failed (${res.status}): ${body}`);
+            }
+            const data = (await res.json());
+            out.push(Float32Array.from(data.embedding));
+        }
+        return out;
+    }
+}
+export async function storeEmbeddingsForChunks(db, embedder, items, batchSize = 16) {
+    let stored = 0;
+    for (let i = 0; i < items.length; i += batchSize) {
+        const batch = items.slice(i, i + batchSize);
+        const vectors = await embedder.embed(batch.map((b) => b.content));
+        for (let j = 0; j < batch.length; j++) {
+            const item = batch[j];
+            const vec = vectors[j];
+            if (!vec)
+                continue;
+            insertEmbedding(db, item.chunkId, embedder.model, vec);
+            stored++;
+        }
+    }
+    return stored;
+}
+function cosine(a, b) {
+    const n = Math.min(a.length, b.length);
+    let dot = 0;
+    let na = 0;
+    let nb = 0;
+    for (let i = 0; i < n; i++) {
+        const x = a[i] ?? 0;
+        const y = b[i] ?? 0;
+        dot += x * y;
+        na += x * x;
+        nb += y * y;
+    }
+    if (na === 0 || nb === 0)
+        return 0;
+    return dot / (Math.sqrt(na) * Math.sqrt(nb));
+}
+export async function vectorSearch(db, embedder, query, limit = 10) {
+    const [qVec] = await embedder.embed([query]);
+    if (!qVec)
+        return [];
+    const rows = db
+        .prepare(`SELECT e.chunk_id AS chunk_id, e.vector AS vector, e.dims AS dims,
+              c.file_id AS file_id, c.start_line AS start_line, c.end_line AS end_line,
+              c.content AS content, f.path AS path, f.language AS language
+       FROM embeddings e
+       JOIN chunks c ON c.id = e.chunk_id
+       JOIN files f ON f.id = c.file_id`)
+        .all();
+    const scored = rows
+        .map((row) => {
+        const buf = Buffer.isBuffer(row.vector) ? row.vector : Buffer.from(row.vector);
+        const vec = new Float32Array(buf.buffer, buf.byteOffset, Math.floor(buf.byteLength / 4));
+        return {
+            row,
+            score: cosine(qVec, vec),
+        };
+    })
+        .sort((a, b) => b.score - a.score)
+        .slice(0, limit);
+    return scored.map((s, i) => ({
+        chunkId: s.row.chunk_id,
+        fileId: s.row.file_id,
+        path: s.row.path,
+        language: s.row.language,
+        startLine: s.row.start_line,
+        endLine: s.row.end_line,
+        content: s.row.content,
+        score: s.score,
+        rank: i + 1,
+    }));
+}
+export function mergeHybrid(bm25Hits, vectorHits, limit) {
+    const map = new Map();
+    const maxB = Math.max(...bm25Hits.map((h) => h.score), 1e-9);
+    const maxV = Math.max(...vectorHits.map((h) => h.score), 1e-9);
+    for (const h of bm25Hits) {
+        map.set(h.chunkId, {
+            ...h,
+            bm25: h.score / maxB,
+            vec: 0,
+            score: 0,
+        });
+    }
+    for (const h of vectorHits) {
+        const existing = map.get(h.chunkId);
+        if (existing) {
+            existing.vec = h.score / maxV;
+        }
+        else {
+            map.set(h.chunkId, {
+                ...h,
+                bm25: 0,
+                vec: h.score / maxV,
+                score: 0,
+            });
+        }
+    }
+    const merged = [...map.values()]
+        .map((h) => ({
+        ...h,
+        score: 0.55 * h.bm25 + 0.45 * h.vec,
+    }))
+        .sort((a, b) => b.score - a.score)
+        .slice(0, limit);
+    return merged.map((h, i) => ({
+        chunkId: h.chunkId,
+        fileId: h.fileId,
+        path: h.path,
+        language: h.language,
+        startLine: h.startLine,
+        endLine: h.endLine,
+        content: h.content,
+        score: h.score,
+        rank: i + 1,
+    }));
+}
+//# sourceMappingURL=embeddings.js.map

package/dist/search/embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.js","sourceRoot":"","sources":["../../src/search/embeddings.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAShD,MAAM,UAAU,cAAc,CAAC,QAAsB;IACnD,IAAI,QAAQ,KAAK,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;QACxD,OAAO,IAAI,cAAc,EAAE,CAAC;IAC9B,CAAC;IACD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,OAAO,IAAI,cAAc,EAAE,CAAC;IAC9B,CAAC;IACD,kEAAkE;IAClE,IAAI,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;QACtD,OAAO,IAAI,cAAc,EAAE,CAAC;IAC9B,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,cAAc;IACT,QAAQ,GAAiB,QAAQ,CAAC;IAClC,KAAK,CAAS;IACf,OAAO,CAAS;IAChB,MAAM,CAAS;IAEvB;QACE,IAAI,CAAC,KAAK;YACR,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,IAAI,EAAE,IAAI,wBAAwB,CAAC;QACzE,IAAI,CAAC,OAAO,GAAG,CACb,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,EAAE,IAAI,2BAA2B,CACnE,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAe;QACzB,IAAI,CAAC,KAAK,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,aAAa,EAAE;YACpD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;aACvC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;aAC1C,CAAC;SACH,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,6BAA6B,GAAG,CAAC,MAAM,MAAM,IAAI,EAAE,CAAC,CAAC;QACvE,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAE7B,CAAC;QACF,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAChE,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3D,CAAC;CACF;AAED,MAAM,cAAc;IACT,QAAQ,GAAiB,QAAQ,CAAC;IAClC,KAAK,CAAS;IACf,OAAO,CAAS;IAExB;QACE,IAAI,CAAC,KAAK;YACR,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,IAAI,EAAE,IAAI,kBAAkB,CAAC;QACnE,IAAI,CAAC,OAAO,GAAG,CACb,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,EAAE;YACnC,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,EAAE;YAC/B,wBAAwB,CACzB,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAe;QACzB,MAAM,GAAG,GAAmB,EAAE,CAAC;QAC/B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,iBAAiB,EAAE;gBACxD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;iBAC5B,CAAC;aACH,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACZ,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC,6BAA6B,GAAG,CAAC,MAAM,MAAM,IAAI,EAAE,CAAC,CAAC;YACvE,CAAC;YACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA4B,CAAC;YAC3D,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QAC9C,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,EAAM,EACN,QAAkB,EAClB,KAAkD,EAClD,SAAS,GAAG,EAAE;IAEd,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;QACjD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;QAClE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC;YACvB,MAAM,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACvB,IAAI,CAAC,GAAG;gBAAE,SAAS;YACnB,eAAe,CAAC,EAAE,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YACvD,MAAM,EAAE,CAAC;QACX,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,MAAM,CAAC,CAAe,EAAE,CAAe;IAC9C,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;IACvC,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,IAAI,EAAE,GAAG,CAAC,CAAC;IACX,IAAI,EAAE,GAAG,CAAC,CAAC;IACX,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACpB,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACpB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;QACb,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QACZ,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;IACd,CAAC;IACD,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IACnC,OAAO,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;AAC/C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,EAAM,EACN,QAAkB,EAClB,KAAa,EACb,KAAK,GAAG,EAAE;IAEV,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IAC7C,IAAI,CAAC,IAAI;QAAE,OAAO,EAAE,CAAC;IAErB,MAAM,IAAI,GAAG,EAAE;SACZ,OAAO,CACN;;;;;wCAKkC,CACnC;SACA,GAAG,EAUJ,CAAC;IAEH,MAAM,MAAM,GAAG,IAAI;SAChB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QACX,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,MAAqB,CAAC,CAAC;QAC9F,MAAM,GAAG,GAAG,IAAI,YAAY,CAC1B,GAAG,CAAC,MAAM,EACV,GAAG,CAAC,UAAU,EACd,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,GAAG,CAAC,CAAC,CAC/B,CAAC;QACF,OAAO;YACL,GAAG;YACH,KAAK,EAAE,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC;SACzB,CAAC;IACJ,CAAC,CAAC;SACD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;SACjC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAEnB,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAC3B,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,QAAQ;QACvB,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO;QACrB,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI;QAChB,QAAQ,EAAE,CAAC,CAAC,GAAG,CAAC,QAAQ;QACxB,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,UAAU;QAC3B,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,QAAQ;QACvB,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO;QACtB,KAAK,EAAE,CAAC,CAAC,KAAK;QACd,IAAI,EAAE,CAAC,GAAG,CAAC;KACZ,CAAC,CAAC,CAAC;AACN,CAAC;AAED,MAAM,UAAU,WAAW,CACzB,QAAqB,EACrB,UAAuB,EACvB,KAAa;IAEb,MAAM,GAAG,GAAG,IAAI,GAAG,EAAqD,CAAC;IAEzE,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;IAC7D,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;IAE/D,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE;YACjB,GAAG,CAAC;YACJ,IAAI,EAAE,CAAC,CAAC,KAAK,GAAG,IAAI;YACpB,GAAG,EAAE,CAAC;YACN,KAAK,EAAE,CAAC;SACT,CAAC,CAAC;IACL,CAAC;IACD,KAAK,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACpC,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,GAAG,GAAG,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC;QAChC,CAAC;aAAM,CAAC;YACN,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE;gBACjB,GAAG,CAAC;gBACJ,IAAI,EAAE,CAAC;gBACP,GAAG,EAAE,CAAC,CAAC,KAAK,GAAG,IAAI;gBACnB,KAAK,EAAE,CAAC;aACT,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC;SAC7B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACX,GAAG,CAAC;QACJ,KAAK,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG;KACpC,CAAC,CAAC;SACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;SACjC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAEnB,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAC3B,OAAO,EAAE,CAAC,CAAC,OAAO;QAClB,MAAM,EAAE,CAAC,CAAC,MAAM;QAChB,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,QAAQ,EAAE,CAAC,CAAC,QAAQ;QACpB,SAAS,EAAE,CAAC,CAAC,SAAS;QACtB,OAAO,EAAE,CAAC,CAAC,OAAO;QAClB,OAAO,EAAE,CAAC,CAAC,OAAO;QAClB,KAAK,EAAE,CAAC,CAAC,KAAK;QACd,IAAI,EAAE,CAAC,GAAG,CAAC;KACZ,CAAC,CAAC,CAAC;AACN,CAAC"}