lmgrep 0.1.10 → 0.1.11

package/README.md

@@ -133,6 +133,68 @@ lmgrep includes an MCP server for use with AI coding assistants. When launched w
 
 You can also start it explicitly with `lmgrep mcp`.
 
+### Claude Code
+
+```sh
+# If lmgrep is installed globally
+claude mcp add lmgrep -s user -- lmgrep mcp
+
+# Or without a global install
+claude mcp add lmgrep -s user -- npx -y lmgrep mcp
+```
+
+### Codex CLI
+
+```sh
+# If lmgrep is installed globally
+codex mcp add lmgrep -- lmgrep mcp
+
+# Or without a global install
+codex mcp add lmgrep -- npx -y lmgrep mcp
+```
+
+### Gemini CLI
+
+```sh
+# If lmgrep is installed globally
+gemini mcp add lmgrep -- lmgrep mcp
+
+# Or without a global install
+gemini mcp add lmgrep -- npx -y lmgrep mcp
+```
+
+### Pi coding agent
+
+Pi doesn't speak MCP — it uses TypeScript extensions instead. lmgrep ships one at [`pi-extension/`](./pi-extension) that registers two tools: `lmgrep_search` and `lmgrep_list_other_indexed_projects`. It imports lmgrep directly, runs an in-process file watcher to keep the index fresh, and gates tool visibility on embedder health — if lmgrep isn't configured, or the embedding provider is unreachable, the tools stay hidden so you get a clean tool surface instead of a broken one. Configure lmgrep first (`lmgrep init`) before relying on it inside Pi.
+
+Install via Pi's package manager:
+
+```sh
+pi install git:github.com/Aetherall/lmgrep
+```
+
+Update with `pi update`, remove with `pi remove git:github.com/Aetherall/lmgrep`, and list installed extensions with `pi list`.
+
+### OpenCode
+
+OpenCode has no one-shot install flag — add an entry to `~/.config/opencode/opencode.json` (or project-level `opencode.json`):
+
+```jsonc
+{
+  "$schema": "https://opencode.ai/config.json",
+  "mcp": {
+    "lmgrep": {
+      "type": "local",
+      // If lmgrep is installed globally
+      "command": ["lmgrep", "mcp"],
+      // Or without a global install
+      // "command": ["npx", "-y", "lmgrep", "mcp"],
+      "enabled": true
+    }
+  }
+}
+```
+
 The MCP server exposes a `search` tool and a `list_other_indexed_projects` tool. It automatically watches for file changes and keeps the index up to date.
 
 ## Configuration

package/dist/lib/search-tool.d.ts

@@ -0,0 +1,47 @@
+export type HealthReason = "ok" | "not_indexed" | "embedding_failed" | "search_empty";
+export interface HealthState {
+    healthy: boolean;
+    reason: HealthReason;
+}
+export interface SearchArgs {
+    query: string;
+    limit?: number;
+    filePrefix?: string;
+    type?: string[];
+    language?: string[];
+    project?: string;
+}
+export interface ToolResult {
+    text: string;
+    isError?: boolean;
+}
+export interface ParamSpec {
+    description: string;
+}
+export interface SearchParamSpecs {
+    query: ParamSpec;
+    limit: ParamSpec & {
+        default: number;
+    };
+    filePrefix: ParamSpec;
+    type: ParamSpec;
+    language: ParamSpec;
+    project: ParamSpec;
+}
+export declare const searchParamSpecs: SearchParamSpecs;
+export declare const listProjectsDescription: string;
+export interface LmgrepCore {
+    readonly cwd: string;
+    readonly searchParams: SearchParamSpecs;
+    readonly listProjectsDescription: string;
+    buildSearchDescription(): string;
+    currentHealth(): HealthState;
+    onHealthChange(cb: (state: HealthState) => void): () => void;
+    startHealthLoop(): void;
+    executeSearch(args: SearchArgs): Promise<ToolResult>;
+    executeListProjects(): Promise<ToolResult>;
+    dispose(): Promise<void>;
+}
+export declare function createLmgrepCore(opts: {
+    cwd: string;
+}): Promise<LmgrepCore>;

package/dist/lib/search-tool.js

@@ -0,0 +1,220 @@
+import { existsSync } from "node:fs";
+import { createIndex } from "../index.js";
+import { TreeSitterChunker } from "./chunker/index.js";
+import { loadConfig } from "./config.js";
+import { AISDKEmbedder } from "./embedder.js";
+import { startWatcher } from "./serve.js";
+import { discoverIndexedProjects, findIndexedAncestor, getDbPath, Store, } from "./store.js";
+import { silentLogger } from "./types.js";
+export const searchParamSpecs = {
+    query: {
+        description: 'Natural-language description of what you\'re looking for — phrase it as a question or intent, not keywords. Good: "how are webhooks authenticated", "where is user deletion handled", "what happens when a record is created". Bad: "webhook auth", "deleteUser", "createRecord".',
+    },
+    limit: {
+        description: "Maximum number of results",
+        default: 10,
+    },
+    filePrefix: {
+        description: "Restrict to files under this path (e.g. 'src/lib')",
+    },
+    type: {
+        description: "AST node types to filter by (e.g. ['function_declaration', 'class_declaration'])",
+    },
+    language: {
+        description: "File extensions to filter by (e.g. ['.ts', '.py'])",
+    },
+    project: {
+        description: "Search a different indexed project by its root path instead of the current one",
+    },
+};
+export const listProjectsDescription = "List all indexed projects other than the current one. " +
+    "Use this to discover what projects are available for cross-project search " +
+    "via the `project` parameter on the search tool.";
+export async function createLmgrepCore(opts) {
+    const cwd = opts.cwd;
+    const index = await createIndex({ cwd });
+    function isCurrentProjectIndexed() {
+        if (findIndexedAncestor(cwd))
+            return true;
+        return existsSync(getDbPath(cwd));
+    }
+    function getOtherProjects() {
+        const currentDb = getDbPath(cwd);
+        return discoverIndexedProjects()
+            .filter((p) => getDbPath(p.metadata.root) !== currentDb)
+            .map((p) => ({ root: p.metadata.root, remote: p.metadata.remote }));
+    }
+    async function checkHealth() {
+        if (!isCurrentProjectIndexed()) {
+            return { healthy: false, reason: "not_indexed" };
+        }
+        try {
+            const info = await index.status();
+            if (info.fileCount === 0) {
+                return { healthy: false, reason: "not_indexed" };
+            }
+            if (!info.embeddingOk) {
+                return { healthy: false, reason: "embedding_failed" };
+            }
+            if (!info.searchOk) {
+                return { healthy: false, reason: "search_empty" };
+            }
+            return { healthy: true, reason: "ok" };
+        }
+        catch {
+            return { healthy: false, reason: "embedding_failed" };
+        }
+    }
+    let state = isCurrentProjectIndexed()
+        ? { healthy: true, reason: "ok" }
+        : { healthy: false, reason: "not_indexed" };
+    const listeners = new Set();
+    let stopWatcher;
+    let pollTimer;
+    let disposed = false;
+    function tryStartWatcher() {
+        if (stopWatcher)
+            return;
+        if (!isCurrentProjectIndexed())
+            return;
+        const config = loadConfig(cwd);
+        const store = Store.forProject(cwd);
+        const embedder = new AISDKEmbedder(config);
+        const chunker = new TreeSitterChunker();
+        stopWatcher = startWatcher(cwd, store, config, embedder, chunker, silentLogger);
+    }
+    function buildSearchDescription() {
+        if (state.healthy) {
+            return [
+                "**lmgrep — primary search tool for this codebase.** Semantic code search powered by a local embedding model; lmgrep understands intent, not string patterns. Prefer lmgrep over Grep/Glob/find/ripgrep for almost all exploration and lookup tasks.",
+                "",
+                '**Use lmgrep for:** finding where something is handled, how something works, locating relevant code, discovering related files, understanding unfamiliar code, tracing side effects, finding usage patterns, answering "where is X?" or "how does Y work?". One good lmgrep query is usually enough to understand how to proceed.',
+                "",
+                "**Query lmgrep as natural questions or intent descriptions**, not keyword dumps:",
+                '- "how are webhooks authenticated" → finds middleware, token validation, auth checks',
+                '- "where is user deletion handled" → finds the handler and related cleanup logic',
+                '- "what happens when a record is created" → finds controllers, event emitters, side effects',
+                '- "config loading and validation"',
+                '- "how to run the playwright tests" → finds config, scripts, prerequisites',
+                "",
+                "**lmgrep results include** file paths, line numbers, AST node types, and surrounding context (scope, leading comments, role) — often enough to act on directly without re-reading the file. Trust lmgrep results; don't follow up with Glob/Read on files already surfaced by lmgrep unless you genuinely need content that wasn't returned.",
+                "",
+                "**Fall back to Grep only** when you need exact string or regex matches (specific identifiers, literal constants, error messages, TODO markers). Don't use Grep/Glob/find for conceptual or intent-based search — lmgrep will do better.",
+            ].join("\n");
+        }
+        const others = getOtherProjects();
+        const suffix = others.length > 0
+            ? " You can still search other indexed projects via the `project` parameter — call `list_other_indexed_projects` to see what's available."
+            : "";
+        switch (state.reason) {
+            case "not_indexed":
+                return ("This project is not indexed. Semantic search is not available for the current directory." +
+                    suffix);
+            case "embedding_failed":
+                return ("The embedding provider is unreachable. Semantic search is temporarily unavailable for the current directory." +
+                    suffix);
+            case "search_empty":
+                return ("The index for the current branch appears empty or stale — a smoke query returned no results. Re-run `lmgrep index` to rebuild." +
+                    suffix);
+            default:
+                return "Semantic search is unavailable." + suffix;
+        }
+    }
+    async function refreshHealth() {
+        if (disposed)
+            return;
+        const next = await checkHealth();
+        if (next.healthy !== state.healthy || next.reason !== state.reason) {
+            state = next;
+            for (const cb of listeners)
+                cb(next);
+        }
+        if (next.healthy)
+            tryStartWatcher();
+    }
+    function startHealthLoop() {
+        if (pollTimer)
+            return;
+        tryStartWatcher();
+        // Non-local providers may bill per request — poll less often.
+        const intervalMs = index.config.local ? 10_000 : 60_000;
+        refreshHealth();
+        pollTimer = setInterval(refreshHealth, intervalMs);
+    }
+    async function executeSearch(args) {
+        if (state.reason === "embedding_failed") {
+            return {
+                text: "lmgrep is unavailable: the embedding provider is unreachable. Ask the user to check their lmgrep configuration (`lmgrep status`) before retrying.",
+                isError: true,
+            };
+        }
+        try {
+            const results = await index.search(args.query, {
+                limit: args.limit ?? searchParamSpecs.limit.default,
+                filePrefix: args.filePrefix,
+                type: args.type,
+                language: args.language,
+                project: args.project,
+            });
+            if (results.length === 0) {
+                return { text: "No results found." };
+            }
+            const text = results
+                .map((r) => {
+                const loc = `${r.filePath}:${r.startLine}-${r.endLine}`;
+                const header = `${loc} [${r.type}] ${r.name} (score: ${r.score.toFixed(3)})`;
+                const parts = [header];
+                if (r.context)
+                    parts.push(r.context);
+                parts.push(r.content);
+                return parts.join("\n");
+            })
+                .join("\n\n---\n\n");
+            return { text };
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            return { text: `Error: ${msg}`, isError: true };
+        }
+    }
+    async function executeListProjects() {
+        const others = getOtherProjects();
+        if (others.length === 0) {
+            return { text: "No other indexed projects found." };
+        }
+        const lines = others.map((p) => {
+            const parts = [p.root];
+            if (p.remote)
+                parts.push(`(${p.remote})`);
+            return `- ${parts.join(" ")}`;
+        });
+        return { text: `Indexed projects:\n${lines.join("\n")}` };
+    }
+    async function dispose() {
+        disposed = true;
+        if (pollTimer) {
+            clearInterval(pollTimer);
+            pollTimer = undefined;
+        }
+        stopWatcher?.();
+        stopWatcher = undefined;
+        listeners.clear();
+        await index.close();
+    }
+    return {
+        cwd,
+        searchParams: searchParamSpecs,
+        listProjectsDescription,
+        buildSearchDescription,
+        currentHealth: () => state,
+        onHealthChange(cb) {
+            listeners.add(cb);
+            return () => listeners.delete(cb);
+        },
+        startHealthLoop,
+        executeSearch,
+        executeListProjects,
+        dispose,
+    };
+}
+//# sourceMappingURL=search-tool.js.map

package/dist/lib/search-tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"search-tool.js","sourceRoot":"","sources":["../../src/lib/search-tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,WAAW,EAAoB,MAAM,aAAa,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC1C,OAAO,EACN,uBAAuB,EACvB,mBAAmB,EACnB,SAAS,EACT,KAAK,GACL,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAwC1C,MAAM,CAAC,MAAM,gBAAgB,GAAqB;IACjD,KAAK,EAAE;QACN,WAAW,EACV,mRAAmR;KACpR;IACD,KAAK,EAAE;QACN,WAAW,EAAE,2BAA2B;QACxC,OAAO,EAAE,EAAE;KACX;IACD,UAAU,EAAE;QACX,WAAW,EAAE,oDAAoD;KACjE;IACD,IAAI,EAAE;QACL,WAAW,EACV,kFAAkF;KACnF;IACD,QAAQ,EAAE;QACT,WAAW,EAAE,oDAAoD;KACjE;IACD,OAAO,EAAE;QACR,WAAW,EACV,gFAAgF;KACjF;CACD,CAAC;AAEF,MAAM,CAAC,MAAM,uBAAuB,GACnC,wDAAwD;IACxD,4EAA4E;IAC5E,iDAAiD,CAAC;AAenD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,IAEtC;IACA,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;IACrB,MAAM,KAAK,GAAgB,MAAM,WAAW,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IAEtD,SAAS,uBAAuB;QAC/B,IAAI,mBAAmB,CAAC,GAAG,CAAC;YAAE,OAAO,IAAI,CAAC;QAC1C,OAAO,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;IACnC,CAAC;IAED,SAAS,gBAAgB;QACxB,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;QACjC,OAAO,uBAAuB,EAAE;aAC9B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,SAAS,CAAC;aACvD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,KAAK,UAAU,WAAW;QACzB,IAAI,CAAC,uBAAuB,EAAE,EAAE,CAAC;YAChC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;QAClD,CAAC;QACD,IAAI,CAAC;YACJ,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,EAAE,CAAC;YAClC,IAAI,IAAI,CAAC,SAAS,KAAK,CAAC,EAAE,CAAC;gBAC1B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;YAClD,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACvB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC;YACvD,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACpB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;YACnD,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QACxC,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC;QACvD,CAAC;IACF,CAAC;IAED,IAAI,KAAK,GAAgB,uBAAuB,EAAE;QACjD,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;QACjC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;IAE7C,MAAM,SAAS,GAAG,IAAI,GAAG,EAA4B,CAAC;IACtD,IAAI,WAAqC,CAAC;IAC1C,IAAI,SAAqC,CAAC;IAC1C,IAAI,QAAQ,GAAG,KAAK,CAAC;IAErB,SAAS,eAAe;QACvB,IAAI,WAAW;YAAE,OAAO;QACxB,IAAI,CAAC,uBAAuB,EAAE;YAAE,OAAO;QACvC,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;QAC/B,MAAM,KAAK,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI,iBAAiB,EAAE,CAAC;QACxC,WAAW,GAAG,YAAY,CACzB,GAAG,EACH,KAAK,EACL,MAAM,EACN,QAAQ,EACR,OAAO,EACP,YAAY,CACZ,CAAC;IACH,CAAC;IAED,SAAS,sBAAsB;QAC9B,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACnB,OAAO;gBACN,qPAAqP;gBACrP,EAAE;gBACF,mUAAmU;gBACnU,EAAE;gBACF,kFAAkF;gBAClF,sFAAsF;gBACtF,kFAAkF;gBAClF,6FAA6F;gBAC7F,mCAAmC;gBACnC,4EAA4E;gBAC5E,EAAE;gBACF,8UAA8U;gBAC9U,EAAE;gBACF,yOAAyO;aACzO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,CAAC;QAED,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;QAClC,MAAM,MAAM,GACX,MAAM,CAAC,MAAM,GAAG,CAAC;YAChB,CAAC,CAAC,wIAAwI;YAC1I,CAAC,CAAC,EAAE,CAAC;QAEP,QAAQ,KAAK,CAAC,MAAM,EAAE,CAAC;YACtB,KAAK,aAAa;gBACjB,OAAO,CACN,0FAA0F;oBAC1F,MAAM,CACN,CAAC;YACH,KAAK,kBAAkB;gBACtB,OAAO,CACN,8GAA8G;oBAC9G,MAAM,CACN,CAAC;YACH,KAAK,cAAc;gBAClB,OAAO,CACN,gIAAgI;oBAChI,MAAM,CACN,CAAC;YACH;gBACC,OAAO,iCAAiC,GAAG,MAAM,CAAC;QACpD,CAAC;IACF,CAAC;IAED,KAAK,UAAU,aAAa;QAC3B,IAAI,QAAQ;YAAE,OAAO;QACrB,MAAM,IAAI,GAAG,MAAM,WAAW,EAAE,CAAC;QACjC,IAAI,IAAI,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,EAAE,CAAC;YACpE,KAAK,GAAG,IAAI,CAAC;YACb,KAAK,MAAM,EAAE,IAAI,SAAS;gBAAE,EAAE,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,IAAI,CAAC,OAAO;YAAE,eAAe,EAAE,CAAC;IACrC,CAAC;IAED,SAAS,eAAe;QACvB,IAAI,SAAS;YAAE,OAAO;QACtB,eAAe,EAAE,CAAC;QAClB,8DAA8D;QAC9D,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;QACxD,aAAa,EAAE,CAAC;QAChB,SAAS,GAAG,WAAW,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,UAAU,aAAa,CAAC,IAAgB;QAC5C,IAAI,KAAK,CAAC,MAAM,KAAK,kBAAkB,EAAE,CAAC;YACzC,OAAO;gBACN,IAAI,EAAE,mJAAmJ;gBACzJ,OAAO,EAAE,IAAI;aACb,CAAC;QACH,CAAC;QACD,IAAI,CAAC;YACJ,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE;gBAC9C,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,gBAAgB,CAAC,KAAK,CAAC,OAAO;gBACnD,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,OAAO,EAAE,IAAI,CAAC,OAAO;aACrB,CAAC,CAAC;YAEH,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1B,OAAO,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC;YACtC,CAAC;YAED,MAAM,IAAI,GAAG,OAAO;iBAClB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACV,MAAM,GAAG,GAAG,GAAG,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;gBACxD,MAAM,MAAM,GAAG,GAAG,GAAG,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,YAAY,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;gBAC7E,MAAM,KAAK,GAAG,CAAC,MAAM,CAAC,CAAC;gBACvB,IAAI,CAAC,CAAC,OAAO;oBAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBACrC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBACtB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC,CAAC;iBACD,IAAI,CAAC,aAAa,CAAC,CAAC;YAEtB,OAAO,EAAE,IAAI,EAAE,CAAC;QACjB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACd,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,OAAO,EAAE,IAAI,EAAE,UAAU,GAAG,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QACjD,CAAC;IACF,CAAC;IAED,KAAK,UAAU,mBAAmB;QACjC,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;QAClC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,IAAI,EAAE,kCAAkC,EAAE,CAAC;QACrD,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YAC9B,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACvB,IAAI,CAAC,CAAC,MAAM;gBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;YAC1C,OAAO,KAAK,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QAC/B,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,IAAI,EAAE,sBAAsB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;IAC3D,CAAC;IAED,KAAK,UAAU,OAAO;QACrB,QAAQ,GAAG,IAAI,CAAC;QAChB,IAAI,SAAS,EAAE,CAAC;YACf,aAAa,CAAC,SAAS,CAAC,CAAC;YACzB,SAAS,GAAG,SAAS,CAAC;QACvB,CAAC;QACD,WAAW,EAAE,EAAE,CAAC;QAChB,WAAW,GAAG,SAAS,CAAC;QACxB,SAAS,CAAC,KAAK,EAAE,CAAC;QAClB,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAED,OAAO;QACN,GAAG;QACH,YAAY,EAAE,gBAAgB;QAC9B,uBAAuB;QACvB,sBAAsB;QACtB,aAAa,EAAE,GAAG,EAAE,CAAC,KAAK;QAC1B,cAAc,CAAC,EAAE;YAChB,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAClB,OAAO,GAAG,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACnC,CAAC;QACD,eAAe;QACf,aAAa;QACb,mBAAmB;QACnB,OAAO;KACP,CAAC;AACH,CAAC"}

package/dist/mcp.js

@@ -3,242 +3,59 @@ process.title = "lmgrep-mcp";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { z } from "zod";
-import { createIndex } from "./index.js";
-import { findIndexedAncestor, discoverIndexedProjects, getDbPath, } from "./lib/store.js";
-import { startWatcher } from "./lib/serve.js";
-import { loadConfig } from "./lib/config.js";
-import { AISDKEmbedder } from "./lib/embedder.js";
-import { TreeSitterChunker } from "./lib/chunker/index.js";
-import { Store } from "./lib/store.js";
-import { existsSync } from "node:fs";
-import { silentLogger } from "./lib/types.js";
-const cwd = process.cwd();
-function isCurrentProjectIndexed() {
-    const ancestor = findIndexedAncestor(cwd);
-    if (ancestor)
-        return true;
-    return existsSync(getDbPath(cwd));
-}
-async function checkHealth() {
-    if (!isCurrentProjectIndexed()) {
-        return { healthy: false, reason: "not_indexed" };
-    }
-    try {
-        const info = await index.status();
-        if (info.fileCount === 0) {
-            return { healthy: false, reason: "not_indexed" };
-        }
-        if (!info.embeddingOk) {
-            return { healthy: false, reason: "embedding_failed" };
-        }
-        if (!info.searchOk) {
-            return { healthy: false, reason: "search_empty" };
-        }
-        return { healthy: true, reason: "ok" };
-    }
-    catch {
-        return { healthy: false, reason: "embedding_failed" };
-    }
-}
-function getOtherProjects() {
-    const currentDb = getDbPath(cwd);
-    return discoverIndexedProjects()
-        .filter((p) => {
-        const pDb = getDbPath(p.metadata.root);
-        return pDb !== currentDb;
-    })
-        .map((p) => ({
-        root: p.metadata.root,
-        remote: p.metadata.remote,
-    }));
-}
-function buildSearchDescription(state) {
-    if (state.healthy) {
-        return [
-            "**lmgrep — primary search tool for this codebase.** Semantic code search powered by a local embedding model; lmgrep understands intent, not string patterns. Prefer lmgrep over Grep/Glob/find/ripgrep for almost all exploration and lookup tasks.",
-            "",
-            "**Use lmgrep for:** finding where something is handled, how something works, locating relevant code, discovering related files, understanding unfamiliar code, tracing side effects, finding usage patterns, answering \"where is X?\" or \"how does Y work?\". One good lmgrep query is usually enough to understand how to proceed.",
-            "",
-            "**Query lmgrep as natural questions or intent descriptions**, not keyword dumps:",
-            '- "how are webhooks authenticated" → finds middleware, token validation, auth checks',
-            '- "where is user deletion handled" → finds the handler and related cleanup logic',
-            '- "what happens when a record is created" → finds controllers, event emitters, side effects',
-            '- "config loading and validation"',
-            '- "how to run the playwright tests" → finds config, scripts, prerequisites',
-            "",
-            "**lmgrep results include** file paths, line numbers, AST node types, and surrounding context (scope, leading comments, role) — often enough to act on directly without re-reading the file. Trust lmgrep results; don't follow up with Glob/Read on files already surfaced by lmgrep unless you genuinely need content that wasn't returned.",
-            "",
-            "**Fall back to Grep only** when you need exact string or regex matches (specific identifiers, literal constants, error messages, TODO markers). Don't use Grep/Glob/find for conceptual or intent-based search — lmgrep will do better.",
-        ].join("\n");
-    }
-    const others = getOtherProjects();
-    const suffix = others.length > 0
-        ? " You can still search other indexed projects via the `project` parameter — call `list_other_indexed_projects` to see what's available."
-        : "";
-    switch (state.reason) {
-        case "not_indexed":
-            return ("This project is not indexed. Semantic search is not available for the current directory." +
-                suffix);
-        case "embedding_failed":
-            return ("The embedding provider is unreachable. Semantic search is temporarily unavailable for the current directory." +
-                suffix);
-        case "search_empty":
-            return ("The index for the current branch appears empty or stale — a smoke query returned no results. Re-run `lmgrep index` to rebuild." +
-                suffix);
-        default:
-            return "Semantic search is unavailable." + suffix;
-    }
-}
-// --- In-process watcher ---
-let stopWatcher;
-function tryStartWatcher() {
-    if (stopWatcher)
-        return; // already watching
-    if (!isCurrentProjectIndexed())
-        return;
-    const config = loadConfig(cwd);
-    const store = Store.forProject(cwd);
-    const embedder = new AISDKEmbedder(config);
-    const chunker = new TreeSitterChunker();
-    stopWatcher = startWatcher(cwd, store, config, embedder, chunker, silentLogger);
-    // undefined means lock was held by another process — that's fine
-}
-// --- MCP server ---
+import { createLmgrepCore } from "./lib/search-tool.js";
+const core = await createLmgrepCore({ cwd: process.cwd() });
 const server = new McpServer({
     name: "lmgrep",
     version: "0.1.0",
 });
-const index = await createIndex({ cwd });
-// Initial optimistic/pessimistic description based on the cheap file-system check.
-// The first async health check below will refine this and push an update if needed.
-const initialState = isCurrentProjectIndexed()
-    ? { healthy: true, reason: "ok" }
-    : { healthy: false, reason: "not_indexed" };
-const searchTool = server.tool("search", buildSearchDescription(initialState), {
-    query: z
-        .string()
-        .describe('Natural-language description of what you\'re looking for — phrase it as a question or intent, not keywords. Good: "how are webhooks authenticated", "where is user deletion handled", "what happens when a record is created". Bad: "webhook auth", "deleteUser", "createRecord".'),
+const searchTool = server.tool("search", core.buildSearchDescription(), {
+    query: z.string().describe(core.searchParams.query.description),
     limit: z
         .number()
         .optional()
-        .default(10)
-        .describe("Maximum number of results"),
+        .default(core.searchParams.limit.default)
+        .describe(core.searchParams.limit.description),
     filePrefix: z
         .string()
         .optional()
-        .describe("Restrict to files under this path (e.g. 'src/lib')"),
+        .describe(core.searchParams.filePrefix.description),
     type: z
         .array(z.string())
         .optional()
-        .describe("AST node types to filter by (e.g. ['function_declaration', 'class_declaration'])"),
+        .describe(core.searchParams.type.description),
     language: z
         .array(z.string())
         .optional()
-        .describe("File extensions to filter by (e.g. ['.ts', '.py'])"),
+        .describe(core.searchParams.language.description),
     project: z
         .string()
         .optional()
-        .describe("Search a different indexed project by its root path instead of the current one"),
-}, async ({ query, limit, filePrefix, type, language, project }) => {
-    try {
-        const results = await index.search(query, {
-            limit,
-            filePrefix,
-            type,
-            language,
-            project,
-        });
-        if (results.length === 0) {
-            return {
-                content: [{ type: "text", text: "No results found." }],
-            };
-        }
-        const text = results
-            .map((r) => {
-            const loc = `${r.filePath}:${r.startLine}-${r.endLine}`;
-            const header = `${loc} [${r.type}] ${r.name} (score: ${r.score.toFixed(3)})`;
-            const parts = [header];
-            if (r.context)
-                parts.push(r.context);
-            parts.push(r.content);
-            return parts.join("\n");
-        })
-            .join("\n\n---\n\n");
-        return {
-            content: [{ type: "text", text }],
-        };
-    }
-    catch (err) {
-        const msg = err instanceof Error ? err.message : String(err);
-        return {
-            content: [{ type: "text", text: `Error: ${msg}` }],
-            isError: true,
-        };
-    }
-});
-server.tool("list_other_indexed_projects", "List all indexed projects other than the current one. " +
-    "Use this to discover what projects are available for cross-project search " +
-    "via the `project` parameter on the search tool.", {}, async () => {
-    const others = getOtherProjects();
-    if (others.length === 0) {
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: "No other indexed projects found.",
-                },
-            ],
-        };
-    }
-    const lines = others.map((p) => {
-        const parts = [p.root];
-        if (p.remote)
-            parts.push(`(${p.remote})`);
-        return `- ${parts.join(" ")}`;
-    });
+        .describe(core.searchParams.project.description),
+}, async (args) => {
+    const result = await core.executeSearch(args);
     return {
-        content: [
-            {
-                type: "text",
-                text: `Indexed projects:\n${lines.join("\n")}`,
-            },
-        ],
+        content: [{ type: "text", text: result.text }],
+        ...(result.isError ? { isError: true } : {}),
     };
 });
-// --- Watch for index state changes and update tool description ---
-let lastState = initialState;
-tryStartWatcher();
-async function refreshHealth() {
-    const next = await checkHealth();
-    if (next.healthy !== lastState.healthy ||
-        next.reason !== lastState.reason) {
-        lastState = next;
-        searchTool.update({ description: buildSearchDescription(next) });
-    }
-    if (next.healthy) {
-        tryStartWatcher();
-    }
-}
-// Non-local providers may bill per request — poll less often to avoid a drip.
-// Local providers (ollama, lmstudio, etc.) can be checked aggressively.
-const healthPollIntervalMs = index.config.local ? 10_000 : 60_000;
-// Kick off an immediate refresh so the initial description reflects the full
-// health check (embedding + smoke search), not just the file-system probe.
-refreshHealth();
-setInterval(refreshHealth, healthPollIntervalMs);
-// --- Cleanup on exit ---
+server.tool("list_other_indexed_projects", core.listProjectsDescription, {}, async () => {
+    const result = await core.executeListProjects();
+    return { content: [{ type: "text", text: result.text }] };
+});
+core.onHealthChange(() => {
+    searchTool.update({ description: core.buildSearchDescription() });
+});
+core.startHealthLoop();
 process.on("exit", () => {
-    stopWatcher?.();
+    core.dispose().catch(() => { });
 });
 process.on("SIGINT", () => {
-    stopWatcher?.();
-    process.exit(0);
+    core.dispose().finally(() => process.exit(0));
 });
 process.on("SIGTERM", () => {
-    stopWatcher?.();
-    process.exit(0);
+    core.dispose().finally(() => process.exit(0));
 });
-// --- Start ---
 async function main() {
     const transport = new StdioServerTransport();
     await server.connect(transport);

package/dist/mcp.js.map

@@ -1 +1 @@
-{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":";AACA,OAAO,CAAC,KAAK,GAAG,YAAY,CAAC;AAE7B,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACzC,OAAO,EACN,mBAAmB,EACnB,uBAAuB,EACvB,SAAS,GACT,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AACvC,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE9C,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;AAe1B,SAAS,uBAAuB;IAC/B,MAAM,QAAQ,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;IAC1C,IAAI,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC1B,OAAO,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;AACnC,CAAC;AAED,KAAK,UAAU,WAAW;IACzB,IAAI,CAAC,uBAAuB,EAAE,EAAE,CAAC;QAChC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;IAClD,CAAC;IACD,IAAI,CAAC;QACJ,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,EAAE,CAAC;QAClC,IAAI,IAAI,CAAC,SAAS,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;QAClD,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACvB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC;QACvD,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;QACnD,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACxC,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC;IACvD,CAAC;AACF,CAAC;AAED,SAAS,gBAAgB;IACxB,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;IACjC,OAAO,uBAAuB,EAAE;SAC9B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;QACb,MAAM,GAAG,GAAG,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACvC,OAAO,GAAG,KAAK,SAAS,CAAC;IAC1B,CAAC,CAAC;SACD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACZ,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI;QACrB,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,MAAM;KACzB,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,sBAAsB,CAAC,KAAkB;IACjD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO;YACN,qPAAqP;YACrP,EAAE;YACF,uUAAuU;YACvU,EAAE;YACF,kFAAkF;YAClF,sFAAsF;YACtF,kFAAkF;YAClF,6FAA6F;YAC7F,mCAAmC;YACnC,4EAA4E;YAC5E,EAAE;YACF,8UAA8U;YAC9U,EAAE;YACF,yOAAyO;SACzO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;IAClC,MAAM,MAAM,GACX,MAAM,CAAC,MAAM,GAAG,CAAC;QAChB,CAAC,CAAC,wIAAwI;QAC1I,CAAC,CAAC,EAAE,CAAC;IAEP,QAAQ,KAAK,CAAC,MAAM,EAAE,CAAC;QACtB,KAAK,aAAa;YACjB,OAAO,CACN,0FAA0F;gBAC1F,MAAM,CACN,CAAC;QACH,KAAK,kBAAkB;YACtB,OAAO,CACN,8GAA8G;gBAC9G,MAAM,CACN,CAAC;QACH,KAAK,cAAc;YAClB,OAAO,CACN,gIAAgI;gBAChI,MAAM,CACN,CAAC;QACH;YACC,OAAO,iCAAiC,GAAG,MAAM,CAAC;IACpD,CAAC;AACF,CAAC;AAED,6BAA6B;AAE7B,IAAI,WAAqC,CAAC;AAE1C,SAAS,eAAe;IACvB,IAAI,WAAW;QAAE,OAAO,CAAC,mBAAmB;IAC5C,IAAI,CAAC,uBAAuB,EAAE;QAAE,OAAO;IAEvC,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IAC/B,MAAM,KAAK,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IACpC,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;IAC3C,MAAM,OAAO,GAAG,IAAI,iBAAiB,EAAE,CAAC;IAExC,WAAW,GAAG,YAAY,CAAC,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;IAChF,iEAAiE;AAClE,CAAC;AAED,qBAAqB;AAErB,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;IAC5B,IAAI,EAAE,QAAQ;IACd,OAAO,EAAE,OAAO;CAChB,CAAC,CAAC;AAEH,MAAM,KAAK,GAAG,MAAM,WAAW,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AAEzC,mFAAmF;AACnF,oFAAoF;AACpF,MAAM,YAAY,GAAgB,uBAAuB,EAAE;IAC1D,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IACjC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;AAE7C,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAC7B,QAAQ,EACR,sBAAsB,CAAC,YAAY,CAAC,EACpC;IACC,KAAK,EAAE,CAAC;SACN,MAAM,EAAE;SACR,QAAQ,CACR,mRAAmR,CACnR;IACF,KAAK,EAAE,CAAC;SACN,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,OAAO,CAAC,EAAE,CAAC;SACX,QAAQ,CAAC,2BAA2B,CAAC;IACvC,UAAU,EAAE,CAAC;SACX,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,oDAAoD,CAAC;IAChE,IAAI,EAAE,CAAC;SACL,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,QAAQ,EAAE;SACV,QAAQ,CACR,kFAAkF,CAClF;IACF,QAAQ,EAAE,CAAC;SACT,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,QAAQ,EAAE;SACV,QAAQ,CAAC,oDAAoD,CAAC;IAChE,OAAO,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACR,gFAAgF,CAChF;CACF,EACD,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;IAC/D,IAAI,CAAC;QACJ,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE;YACzC,KAAK;YACL,UAAU;YACV,IAAI;YACJ,QAAQ;YACR,OAAO;SACP,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC;aAC/D,CAAC;QACH,CAAC;QAED,MAAM,IAAI,GAAG,OAAO;aAClB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACV,MAAM,GAAG,GAAG,GAAG,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;YACxD,MAAM,MAAM,GAAG,GAAG,GAAG,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,YAAY,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;YAC7E,MAAM,KAAK,GAAG,CAAC,MAAM,CAAC,CAAC;YACvB,IAAI,CAAC,CAAC,OAAO;gBAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YACrC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YACtB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC,CAAC;aACD,IAAI,CAAC,aAAa,CAAC,CAAC;QAEtB,OAAO;YACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC;SAC1C,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACd,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC7D,OAAO;YACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,UAAU,GAAG,EAAE,EAAE,CAAC;YAC3D,OAAO,EAAE,IAAI;SACb,CAAC;IACH,CAAC;AACF,CAAC,CACD,CAAC;AAEF,MAAM,CAAC,IAAI,CACV,6BAA6B,EAC7B,wDAAwD;IACvD,4EAA4E;IAC5E,iDAAiD,EAClD,EAAE,EACF,KAAK,IAAI,EAAE;IACV,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;IAElC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO;YACN,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,kCAAkC;iBACxC;aACD;SACD,CAAC;IACH,CAAC;IAED,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC9B,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACvB,IAAI,CAAC,CAAC,MAAM;YAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QAC1C,OAAO,KAAK,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,OAAO;QACN,OAAO,EAAE;YACR;gBACC,IAAI,EAAE,MAAe;gBACrB,IAAI,EAAE,sBAAsB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;aAC9C;SACD;KACD,CAAC;AACH,CAAC,CACD,CAAC;AAEF,oEAAoE;AAEpE,IAAI,SAAS,GAAgB,YAAY,CAAC;AAC1C,eAAe,EAAE,CAAC;AAElB,KAAK,UAAU,aAAa;IAC3B,MAAM,IAAI,GAAG,MAAM,WAAW,EAAE,CAAC;IACjC,IACC,IAAI,CAAC,OAAO,KAAK,SAAS,CAAC,OAAO;QAClC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,MAAM,EAC/B,CAAC;QACF,SAAS,GAAG,IAAI,CAAC;QACjB,UAAU,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,sBAAsB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAClE,CAAC;IACD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAClB,eAAe,EAAE,CAAC;IACnB,CAAC;AACF,CAAC;AAED,8EAA8E;AAC9E,wEAAwE;AACxE,MAAM,oBAAoB,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;AAElE,6EAA6E;AAC7E,2EAA2E;AAC3E,aAAa,EAAE,CAAC;AAEhB,WAAW,CAAC,aAAa,EAAE,oBAAoB,CAAC,CAAC;AAEjD,0BAA0B;AAE1B,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;IACvB,WAAW,EAAE,EAAE,CAAC;AACjB,CAAC,CAAC,CAAC;AACH,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;IACzB,WAAW,EAAE,EAAE,CAAC;IAChB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC;AACH,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;IAC1B,WAAW,EAAE,EAAE,CAAC;IAChB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC;AAEH,gBAAgB;AAEhB,KAAK,UAAU,IAAI;IAClB,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AACjC,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACpB,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;IAC/C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC"}
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":";AACA,OAAO,CAAC,KAAK,GAAG,YAAY,CAAC;AAE7B,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAExD,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;AAE5D,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;IAC5B,IAAI,EAAE,QAAQ;IACd,OAAO,EAAE,OAAO;CAChB,CAAC,CAAC;AAEH,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAC7B,QAAQ,EACR,IAAI,CAAC,sBAAsB,EAAE,EAC7B;IACC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC;IAC/D,KAAK,EAAE,CAAC;SACN,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC;SACxC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC;IAC/C,UAAU,EAAE,CAAC;SACX,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,WAAW,CAAC;IACpD,IAAI,EAAE,CAAC;SACL,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,QAAQ,EAAE;SACV,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC;IAC9C,QAAQ,EAAE,CAAC;SACT,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACjB,QAAQ,EAAE;SACV,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,WAAW,CAAC;IAClD,OAAO,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,WAAW,CAAC;CACjD,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;IACd,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;IAC9C,OAAO;QACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;QACvD,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC5C,CAAC;AACH,CAAC,CACD,CAAC;AAEF,MAAM,CAAC,IAAI,CACV,6BAA6B,EAC7B,IAAI,CAAC,uBAAuB,EAC5B,EAAE,EACF,KAAK,IAAI,EAAE;IACV,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC;IAChD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;AACpE,CAAC,CACD,CAAC;AAEF,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE;IACxB,UAAU,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,sBAAsB,EAAE,EAAE,CAAC,CAAC;AACnE,CAAC,CAAC,CAAC;AAEH,IAAI,CAAC,eAAe,EAAE,CAAC;AAEvB,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;IACvB,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;AAChC,CAAC,CAAC,CAAC;AACH,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;IACzB,IAAI,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;AAC/C,CAAC,CAAC,CAAC;AACH,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;IAC1B,IAAI,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;AAC/C,CAAC,CAAC,CAAC;AAEH,KAAK,UAAU,IAAI;IAClB,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AACjC,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACpB,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;IAC/C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC,CAAC,CAAC"}

package/package.json

@@ -1,12 +1,17 @@
 {
   "name": "lmgrep",
-  "version": "0.1.10",
+  "version": "0.1.11",
   "description": "Semantic code search with any AI embedding provider",
   "type": "module",
   "bin": {
     "lmgrep": "./dist/cli.js",
     "lmgrep-mcp": "./dist/mcp.js"
   },
+  "pi": {
+    "extensions": [
+      "./pi-extension/index.ts"
+    ]
+  },
   "scripts": {
     "build": "tsc",
     "dev": "tsc --watch",