@millstone/synapse-context-mcp 0.3.0

package/dist/tools/file-tree.js

@@ -0,0 +1,45 @@
+import { FileWalker } from '../utils/file-walker.js';
+import * as path from 'path';
+/**
+ * Get workspace file tree
+ */
+export async function getFileTreeTool(args, workspaceDir) {
+    // Return workspace roots
+    if (args.type === 'roots') {
+        return {
+            type: 'roots',
+            roots: [workspaceDir],
+        };
+    }
+    // Get files
+    try {
+        const walker = new FileWalker(workspaceDir);
+        const startPath = args.path || workspaceDir;
+        const files = await walker.walk(startPath);
+        // Filter by depth if specified
+        let filteredFiles = files;
+        if (args.max_depth !== undefined) {
+            filteredFiles = files.filter(file => {
+                const relativePath = path.relative(workspaceDir, file);
+                const depth = relativePath.split(path.sep).length;
+                return depth <= args.max_depth;
+            });
+        }
+        return {
+            type: args.type,
+            mode: args.mode,
+            files: filteredFiles,
+            count: filteredFiles.length,
+        };
+    }
+    catch (error) {
+        console.error('File tree error:', error);
+        return {
+            type: args.type,
+            mode: args.mode,
+            files: [],
+            count: 0,
+        };
+    }
+}
+//# sourceMappingURL=file-tree.js.map

package/dist/tools/file-tree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-tree.js","sourceRoot":"","sources":["../../src/tools/file-tree.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACrD,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAiB7B;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,IAAkB,EAClB,YAAoB;IAEpB,yBAAyB;IACzB,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC1B,OAAO;YACL,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,CAAC,YAAY,CAAC;SACtB,CAAC;IACJ,CAAC;IAED,YAAY;IACZ,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,YAAY,CAAC,CAAC;QAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,IAAI,YAAY,CAAC;QAC5C,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAE3C,+BAA+B;QAC/B,IAAI,aAAa,GAAG,KAAK,CAAC;QAC1B,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACjC,aAAa,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;gBAClC,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;gBACvD,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC;gBAClD,OAAO,KAAK,IAAI,IAAI,CAAC,SAAU,CAAC;YAClC,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,aAAa;YACpB,KAAK,EAAE,aAAa,CAAC,MAAM;SAC5B,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;QACzC,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC;SACT,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/tools/manage-selection.d.ts

@@ -0,0 +1,28 @@
+import type { SelectionManager, SelectionMode, Slice } from '../selection/SelectionManager.js';
+export interface ManageSelectionArgs {
+    op: 'get' | 'add' | 'remove' | 'set' | 'clear' | 'preview' | 'promote' | 'demote';
+    paths?: string[];
+    mode?: SelectionMode;
+    slices?: Array<{
+        path: string;
+        ranges: Slice[];
+    }>;
+    view?: 'summary' | 'files' | 'content';
+}
+export interface ManageSelectionResult {
+    operation: string;
+    files?: Array<{
+        path: string;
+        mode: SelectionMode;
+        slices?: Slice[];
+    }>;
+    totalFiles?: number;
+    totalTokens?: number;
+    content?: string;
+    error?: string;
+}
+/**
+ * Manage file selection (add/remove/get/clear)
+ */
+export declare function manageSelectionTool(args: ManageSelectionArgs, manager: SelectionManager): Promise<ManageSelectionResult>;
+//# sourceMappingURL=manage-selection.d.ts.map

package/dist/tools/manage-selection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"manage-selection.d.ts","sourceRoot":"","sources":["../../src/tools/manage-selection.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,kCAAkC,CAAC;AAE/F,MAAM,WAAW,mBAAmB;IAClC,EAAE,EAAE,KAAK,GAAG,KAAK,GAAG,QAAQ,GAAG,KAAK,GAAG,OAAO,GAAG,SAAS,GAAG,SAAS,GAAG,QAAQ,CAAC;IAClF,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,IAAI,CAAC,EAAE,aAAa,CAAC;IACrB,MAAM,CAAC,EAAE,KAAK,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,EAAE,KAAK,EAAE,CAAC;KACjB,CAAC,CAAC;IACH,IAAI,CAAC,EAAE,SAAS,GAAG,OAAO,GAAG,SAAS,CAAC;CACxC;AAED,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,aAAa,CAAC;QACpB,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC;KAClB,CAAC,CAAC;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,wBAAsB,mBAAmB,CACvC,IAAI,EAAE,mBAAmB,EACzB,OAAO,EAAE,gBAAgB,GACxB,OAAO,CAAC,qBAAqB,CAAC,CA8GhC"}

package/dist/tools/manage-selection.js

@@ -0,0 +1,98 @@
+/**
+ * Manage file selection (add/remove/get/clear)
+ */
+export async function manageSelectionTool(args, manager) {
+    // Validate operation
+    const validOps = ['get', 'add', 'remove', 'set', 'clear', 'preview', 'promote', 'demote'];
+    if (!validOps.includes(args.op)) {
+        return {
+            operation: args.op,
+            error: `Invalid operation '${args.op}'. Valid operations: ${validOps.join(', ')}`,
+        };
+    }
+    // Validate that at least one of paths or slices is provided for operations that need them
+    if (['add', 'remove', 'set', 'promote', 'demote'].includes(args.op)) {
+        if (!args.paths && !args.slices) {
+            return {
+                operation: args.op,
+                error: `Operation '${args.op}' requires either 'paths' or 'slices' parameter`,
+            };
+        }
+    }
+    switch (args.op) {
+        case 'add':
+            // Handle adding full files
+            for (const path of args.paths || []) {
+                await manager.add(path, args.mode || 'full');
+            }
+            // Handle adding slices
+            if (args.slices) {
+                for (const slice of args.slices) {
+                    await manager.addSlices(slice.path, slice.ranges);
+                }
+            }
+            break;
+        case 'remove':
+            if (args.paths) {
+                manager.removeMultiple(args.paths);
+            }
+            break;
+        case 'set':
+            const filesToSet = args.paths?.map(path => ({
+                path,
+                mode: args.mode || 'full',
+            })) || [];
+            if (args.slices) {
+                for (const slice of args.slices) {
+                    filesToSet.push({
+                        path: slice.path,
+                        mode: 'slices',
+                        slices: slice.ranges,
+                    });
+                }
+            }
+            await manager.set(filesToSet);
+            break;
+        case 'clear':
+            manager.clear();
+            break;
+        case 'preview':
+            const preview = await manager.preview();
+            return {
+                operation: 'preview',
+                content: preview.content,
+                totalTokens: preview.tokens,
+            };
+        case 'promote':
+            for (const path of args.paths || []) {
+                await manager.promote(path);
+            }
+            break;
+        case 'demote':
+            for (const path of args.paths || []) {
+                await manager.demote(path);
+            }
+            break;
+        case 'get':
+        default:
+            const summary = await manager.getSummary();
+            const result = {
+                operation: 'get',
+                files: summary.files,
+                totalFiles: summary.totalFiles,
+                totalTokens: summary.totalTokens,
+            };
+            if (args.view === 'content') {
+                result.content = await manager.getContent();
+            }
+            return result;
+    }
+    const summary = await manager.getSummary();
+    return {
+        operation: args.op,
+        files: summary.files,
+        totalFiles: summary.totalFiles,
+        totalTokens: summary.totalTokens,
+    };
+}
+//# sourceMappingURL=manage-selection.js.map

package/dist/tools/manage-selection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"manage-selection.js","sourceRoot":"","sources":["../../src/tools/manage-selection.ts"],"names":[],"mappings":"AA0BA;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,IAAyB,EACzB,OAAyB;IAEzB,qBAAqB;IACrB,MAAM,QAAQ,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC1F,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;QAChC,OAAO;YACL,SAAS,EAAE,IAAI,CAAC,EAAE;YAClB,KAAK,EAAE,sBAAsB,IAAI,CAAC,EAAE,wBAAwB,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;SAClF,CAAC;IACJ,CAAC;IAED,0FAA0F;IAC1F,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;QACpE,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,OAAO;gBACL,SAAS,EAAE,IAAI,CAAC,EAAE;gBAClB,KAAK,EAAE,cAAc,IAAI,CAAC,EAAE,iDAAiD;aAC9E,CAAC;QACJ,CAAC;IACH,CAAC;IAED,QAAQ,IAAI,CAAC,EAAE,EAAE,CAAC;QAChB,KAAK,KAAK;YACR,2BAA2B;YAC3B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;gBACpC,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,MAAM,CAAC,CAAC;YAC/C,CAAC;YAED,uBAAuB;YACvB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;oBAChC,MAAM,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;YACD,MAAM;QAER,KAAK,QAAQ;YACX,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrC,CAAC;YACD,MAAM;QAER,KAAK,KAAK;YACR,MAAM,UAAU,GACd,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACvB,IAAI;gBACJ,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,MAAuB;aAC3C,CAAC,CAAC,IAAI,EAAE,CAAC;YAEZ,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;oBAChC,UAAU,CAAC,IAAI,CAAC;wBACd,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,IAAI,EAAE,QAAyB;wBAC/B,MAAM,EAAE,KAAK,CAAC,MAAM;qBACrB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,MAAM,OAAO,CAAC,GAAG,CAAC,UAAiB,CAAC,CAAC;YACrC,MAAM;QAER,KAAK,OAAO;YACV,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM;QAER,KAAK,SAAS;YACZ,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;YACxC,OAAO;gBACL,SAAS,EAAE,SAAS;gBACpB,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,WAAW,EAAE,OAAO,CAAC,MAAM;aAC5B,CAAC;QAEJ,KAAK,SAAS;YACZ,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;gBACpC,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC9B,CAAC;YACD,MAAM;QAER,KAAK,QAAQ;YACX,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;gBACpC,MAAM,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;YACD,MAAM;QAER,KAAK,KAAK,CAAC;QACX;YACE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;YAE3C,MAAM,MAAM,GAA0B;gBACpC,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,WAAW,EAAE,OAAO,CAAC,WAAW;aACjC,CAAC;YAEF,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC5B,MAAM,CAAC,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;YAC9C,CAAC;YAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;IAC3C,OAAO;QACL,SAAS,EAAE,IAAI,CAAC,EAAE;QAClB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,WAAW,EAAE,OAAO,CAAC,WAAW;KACjC,CAAC;AACJ,CAAC"}

package/dist/tools/read-file.d.ts

@@ -0,0 +1,19 @@
+export interface ReadFileArgs {
+    path: string;
+    start_line?: number;
+    end_line?: number;
+}
+export interface ReadFileResult {
+    path: string;
+    content: string;
+    lines?: {
+        start: number;
+        end: number;
+    };
+    size: number;
+}
+/**
+ * Read file contents
+ */
+export declare function readFileTool(args: ReadFileArgs): Promise<ReadFileResult>;
+//# sourceMappingURL=read-file.d.ts.map

package/dist/tools/read-file.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"read-file.d.ts","sourceRoot":"","sources":["../../src/tools/read-file.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE;QACN,KAAK,EAAE,MAAM,CAAC;QACd,GAAG,EAAE,MAAM,CAAC;KACb,CAAC;IACF,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,wBAAsB,YAAY,CAChC,IAAI,EAAE,YAAY,GACjB,OAAO,CAAC,cAAc,CAAC,CA6BzB"}

package/dist/tools/read-file.js

@@ -0,0 +1,35 @@
+import * as fs from 'fs/promises';
+/**
+ * Read file contents
+ */
+export async function readFileTool(args) {
+    try {
+        const fullContent = await fs.readFile(args.path, 'utf-8');
+        // Read specific line range if specified
+        if (args.start_line !== undefined && args.end_line !== undefined) {
+            const lines = fullContent.split('\n');
+            const selectedLines = lines.slice(args.start_line, args.end_line + 1);
+            const content = selectedLines.join('\n');
+            return {
+                path: args.path,
+                content,
+                lines: {
+                    start: args.start_line,
+                    end: args.end_line,
+                },
+                size: content.length,
+            };
+        }
+        else {
+            return {
+                path: args.path,
+                content: fullContent,
+                size: fullContent.length,
+            };
+        }
+    }
+    catch (error) {
+        throw new Error(`Failed to read file ${args.path}: ${error}`);
+    }
+}
+//# sourceMappingURL=read-file.js.map

package/dist/tools/read-file.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"read-file.js","sourceRoot":"","sources":["../../src/tools/read-file.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAkBlC;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,IAAkB;IAElB,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAE1D,wCAAwC;QACxC,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YACjE,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;YACtE,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAEzC,OAAO;gBACL,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,OAAO;gBACP,KAAK,EAAE;oBACL,KAAK,EAAE,IAAI,CAAC,UAAU;oBACtB,GAAG,EAAE,IAAI,CAAC,QAAQ;iBACnB;gBACD,IAAI,EAAE,OAAO,CAAC,MAAM;aACrB,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,OAAO;gBACL,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,OAAO,EAAE,WAAW;gBACpB,IAAI,EAAE,WAAW,CAAC,MAAM;aACzB,CAAC;QACJ,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC,CAAC;IAChE,CAAC;AACH,CAAC"}

package/dist/tools/semantic-search.d.ts

@@ -0,0 +1,50 @@
+/**
+ * Semantic Search Tool for Context-MCP
+ *
+ * Provides hybrid search combining:
+ * - FTS5 full-text search with BM25 ranking (lexical matching)
+ * - LanceDB vector search with embeddings (semantic similarity)
+ *
+ * Results are merged using Reciprocal Rank Fusion (RRF) to combine
+ * both ranking signals effectively.
+ *
+ * IMPORTANT: This tool requires LanceDB. If LanceDB is unavailable
+ * (unsupported platform), the tool will fail rather than fall back.
+ */
+import type { ContinueIDE } from '../continue/ContinueIDE.js';
+export interface SemanticSearchArgs {
+    query: string;
+    mode?: 'hybrid' | 'semantic' | 'lexical';
+    filter?: {
+        extensions?: string[];
+        exclude?: string[];
+    };
+    max_results?: number;
+    include_content?: boolean;
+    semantic_weight?: number;
+}
+export interface SemanticSearchResult {
+    query: string;
+    mode: string;
+    results: Array<{
+        path: string;
+        line_range: [number, number];
+        content?: string;
+        score: number;
+        source: 'semantic' | 'lexical' | 'hybrid';
+    }>;
+    count: number;
+    performance?: {
+        searchTime: number;
+        semanticTime?: number;
+        lexicalTime?: number;
+    };
+}
+/**
+ * Main semantic search tool.
+ *
+ * @throws Error if vector search is unavailable (LanceDB not supported)
+ */
+export declare function semanticSearchTool(args: SemanticSearchArgs, workspaceDirs: string | string[], ide: ContinueIDE): Promise<SemanticSearchResult>;
+export default semanticSearchTool;
+//# sourceMappingURL=semantic-search.d.ts.map

package/dist/tools/semantic-search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"semantic-search.d.ts","sourceRoot":"","sources":["../../src/tools/semantic-search.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AASH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAE9D,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,QAAQ,GAAG,UAAU,GAAG,SAAS,CAAC;IACzC,MAAM,CAAC,EAAE;QACP,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;QACtB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;KACpB,CAAC;IACF,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,KAAK,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,UAAU,GAAG,SAAS,GAAG,QAAQ,CAAC;KAC3C,CAAC,CAAC;IACH,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE;QACZ,UAAU,EAAE,MAAM,CAAC;QACnB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAwKD;;;;GAIG;AACH,wBAAsB,kBAAkB,CACtC,IAAI,EAAE,kBAAkB,EACxB,aAAa,EAAE,MAAM,GAAG,MAAM,EAAE,EAChC,GAAG,EAAE,WAAW,GACf,OAAO,CAAC,oBAAoB,CAAC,CA6G/B;AAED,eAAe,kBAAkB,CAAC"}

package/dist/tools/semantic-search.js

@@ -0,0 +1,243 @@
+/**
+ * Semantic Search Tool for Context-MCP
+ *
+ * Provides hybrid search combining:
+ * - FTS5 full-text search with BM25 ranking (lexical matching)
+ * - LanceDB vector search with embeddings (semantic similarity)
+ *
+ * Results are merged using Reciprocal Rank Fusion (RRF) to combine
+ * both ranking signals effectively.
+ *
+ * IMPORTANT: This tool requires LanceDB. If LanceDB is unavailable
+ * (unsupported platform), the tool will fail rather than fall back.
+ */
+import { FullTextSearchCodebaseIndex } from '@continuedev/core/dist/indexing/FullTextSearchCodebaseIndex.js';
+import { getComputeDeleteAddRemove } from '@continuedev/core/dist/indexing/refreshIndex.js';
+import { ChunkCodebaseIndex } from '@continuedev/core/dist/indexing/chunk/ChunkCodebaseIndex.js';
+import { promises as fs } from 'node:fs';
+import { FileWalker } from '../utils/file-walker.js';
+import { vectorSearch, isVectorSearchAvailable } from '../embeddings/vector-index.js';
+// Track which directories have FTS index built
+const ftsIndexedDirs = new Set();
+/**
+ * Sanitize query for FTS5 phrase search.
+ */
+function sanitizeFTS5Query(text) {
+    const escaped = text.replace(/"/g, '""');
+    return `"${escaped}"`;
+}
+/**
+ * Build FTS5 index for a directory (same logic as file-search.ts).
+ */
+async function ensureFTSIndex(workspaceDir, ide) {
+    if (ftsIndexedDirs.has(workspaceDir)) {
+        return {
+            branch: 'main',
+            directory: workspaceDir,
+            artifactId: 'chunks',
+        };
+    }
+    console.error(`[semantic-search] Building FTS index for: ${workspaceDir}`);
+    const tag = {
+        branch: 'main',
+        directory: workspaceDir,
+        artifactId: 'chunks',
+    };
+    const walker = new FileWalker(workspaceDir);
+    const allFiles = await walker.walk();
+    const fileStats = {};
+    for (const file of allFiles) {
+        try {
+            const stats = await fs.stat(file);
+            fileStats[file] = {
+                size: stats.size,
+                lastModified: stats.mtimeMs,
+            };
+        }
+        catch {
+            // Skip files we can't stat
+        }
+    }
+    const [results, , markComplete] = await getComputeDeleteAddRemove(tag, fileStats, async (path) => await fs.readFile(path, 'utf-8'), undefined);
+    // Build chunks index
+    const mockServerClient = {
+        connected: false,
+        url: undefined,
+        getUserToken: () => undefined,
+        getConfig: async () => ({ configJson: '{}' }),
+        getFromIndexCache: async () => ({ files: {} }),
+    };
+    const chunkIndex = new ChunkCodebaseIndex(ide.readFile.bind(ide), mockServerClient, 512);
+    for await (const progress of chunkIndex.update(tag, results, markComplete, undefined)) {
+        // Progress logging
+    }
+    // Build FTS index
+    const ftsIndex = new FullTextSearchCodebaseIndex();
+    for await (const progress of ftsIndex.update(tag, results, markComplete, undefined)) {
+        // Progress logging
+    }
+    ftsIndexedDirs.add(workspaceDir);
+    console.error(`[semantic-search] ✅ FTS index built for ${workspaceDir}`);
+    return tag;
+}
+/**
+ * Perform FTS5 lexical search.
+ */
+async function lexicalSearch(query, workspaceDirs, ide, maxResults) {
+    const tags = [];
+    for (const dir of workspaceDirs) {
+        const tag = await ensureFTSIndex(dir, ide);
+        tags.push(tag);
+    }
+    const ftsIndex = new FullTextSearchCodebaseIndex();
+    return await ftsIndex.retrieve({
+        n: maxResults,
+        text: sanitizeFTS5Query(query),
+        tags,
+        bm25Threshold: -2.5,
+    });
+}
+/**
+ * Reciprocal Rank Fusion (RRF) for combining ranked lists.
+ * RRF(d) = Σ 1/(k + rank(d)) for each ranking
+ * k=60 is standard constant to prevent high ranks from dominating.
+ */
+function reciprocalRankFusion(semanticResults, lexicalResults, semanticWeight = 0.5, k = 60) {
+    const scores = new Map();
+    const lexicalWeight = 1 - semanticWeight;
+    // Score semantic results
+    semanticResults.forEach((chunk, rank) => {
+        const key = `${chunk.filepath}:${chunk.startLine}-${chunk.endLine}`;
+        const rrfScore = semanticWeight / (k + rank + 1);
+        if (scores.has(key)) {
+            const existing = scores.get(key);
+            existing.score += rrfScore;
+            existing.sources.add('semantic');
+        }
+        else {
+            scores.set(key, {
+                chunk,
+                score: rrfScore,
+                sources: new Set(['semantic']),
+            });
+        }
+    });
+    // Score lexical results
+    lexicalResults.forEach((chunk, rank) => {
+        const key = `${chunk.filepath}:${chunk.startLine}-${chunk.endLine}`;
+        const rrfScore = lexicalWeight / (k + rank + 1);
+        if (scores.has(key)) {
+            const existing = scores.get(key);
+            existing.score += rrfScore;
+            existing.sources.add('lexical');
+        }
+        else {
+            scores.set(key, {
+                chunk,
+                score: rrfScore,
+                sources: new Set(['lexical']),
+            });
+        }
+    });
+    return scores;
+}
+/**
+ * Main semantic search tool.
+ *
+ * @throws Error if vector search is unavailable (LanceDB not supported)
+ */
+export async function semanticSearchTool(args, workspaceDirs, ide) {
+    const dirs = Array.isArray(workspaceDirs) ? workspaceDirs : [workspaceDirs];
+    const mode = args.mode || 'hybrid';
+    const maxResults = args.max_results || 30;
+    const includeContent = args.include_content ?? false;
+    const semanticWeight = args.semantic_weight ?? 0.5;
+    const startTime = Date.now();
+    let semanticTime;
+    let lexicalTime;
+    // Check vector search availability - FAIL if unavailable
+    if (mode === 'hybrid' || mode === 'semantic') {
+        const available = await isVectorSearchAvailable();
+        if (!available) {
+            throw new Error('Semantic search requires LanceDB which is not available on this platform. ' +
+                'Use file_search tool for lexical search instead.');
+        }
+    }
+    let semanticResults = [];
+    let lexicalResults = [];
+    // Perform semantic search
+    if (mode === 'hybrid' || mode === 'semantic') {
+        const semStart = Date.now();
+        semanticResults = await vectorSearch(args.query, dirs, ide, {
+            maxResults: maxResults * 2, // Get more to allow for deduplication
+        });
+        semanticTime = Date.now() - semStart;
+    }
+    // Perform lexical search
+    if (mode === 'hybrid' || mode === 'lexical') {
+        const lexStart = Date.now();
+        lexicalResults = await lexicalSearch(args.query, dirs, ide, maxResults * 2);
+        lexicalTime = Date.now() - lexStart;
+    }
+    // Merge results
+    let finalResults = [];
+    if (mode === 'hybrid') {
+        // Use RRF to combine rankings
+        const fusedScores = reciprocalRankFusion(semanticResults, lexicalResults, semanticWeight);
+        // Sort by fused score and take top results
+        const sorted = Array.from(fusedScores.values())
+            .sort((a, b) => b.score - a.score)
+            .slice(0, maxResults);
+        finalResults = sorted.map(({ chunk, score, sources }) => ({
+            path: chunk.filepath,
+            line_range: [chunk.startLine, chunk.endLine],
+            content: includeContent ? chunk.content : undefined,
+            score,
+            source: sources.size > 1 ? 'hybrid' : (sources.has('semantic') ? 'semantic' : 'lexical'),
+        }));
+    }
+    else if (mode === 'semantic') {
+        finalResults = semanticResults.slice(0, maxResults).map((chunk, idx) => ({
+            path: chunk.filepath,
+            line_range: [chunk.startLine, chunk.endLine],
+            content: includeContent ? chunk.content : undefined,
+            score: 1 / (idx + 1), // Normalized rank score
+            source: 'semantic',
+        }));
+    }
+    else {
+        finalResults = lexicalResults.slice(0, maxResults).map((chunk, idx) => ({
+            path: chunk.filepath,
+            line_range: [chunk.startLine, chunk.endLine],
+            content: includeContent ? chunk.content : undefined,
+            score: 1 / (idx + 1),
+            source: 'lexical',
+        }));
+    }
+    // Apply filters
+    if (args.filter?.extensions?.length) {
+        finalResults = finalResults.filter((r) => {
+            const ext = r.path.split('.').pop();
+            return ext && args.filter.extensions.includes(ext);
+        });
+    }
+    if (args.filter?.exclude?.length) {
+        finalResults = finalResults.filter((r) => {
+            return !args.filter.exclude.some((ex) => r.path.includes(ex));
+        });
+    }
+    const searchTime = Date.now() - startTime;
+    return {
+        query: args.query,
+        mode,
+        results: finalResults,
+        count: finalResults.length,
+        performance: {
+            searchTime,
+            semanticTime,
+            lexicalTime,
+        },
+    };
+}
+export default semanticSearchTool;
+//# sourceMappingURL=semantic-search.js.map

package/dist/tools/semantic-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"semantic-search.js","sourceRoot":"","sources":["../../src/tools/semantic-search.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAGH,OAAO,EAAE,2BAA2B,EAAE,MAAM,gEAAgE,CAAC;AAC7G,OAAO,EAAE,yBAAyB,EAAE,MAAM,iDAAiD,CAAC;AAC5F,OAAO,EAAE,kBAAkB,EAAE,MAAM,6DAA6D,CAAC;AACjG,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAoB,uBAAuB,EAAE,MAAM,+BAA+B,CAAC;AAiCxG,+CAA+C;AAC/C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;AAEzC;;GAEG;AACH,SAAS,iBAAiB,CAAC,IAAY;IACrC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IACzC,OAAO,IAAI,OAAO,GAAG,CAAC;AACxB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,cAAc,CAC3B,YAAoB,EACpB,GAAgB;IAEhB,IAAI,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;QACrC,OAAO;YACL,MAAM,EAAE,MAAM;YACd,SAAS,EAAE,YAAY;YACvB,UAAU,EAAE,QAAQ;SACrB,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,KAAK,CAAC,6CAA6C,YAAY,EAAE,CAAC,CAAC;IAE3E,MAAM,GAAG,GAAa;QACpB,MAAM,EAAE,MAAM;QACd,SAAS,EAAE,YAAY;QACvB,UAAU,EAAE,QAAQ;KACrB,CAAC;IAEF,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,YAAY,CAAC,CAAC;IAC5C,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;IAErC,MAAM,SAAS,GAA2D,EAAE,CAAC;IAC7E,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;QAC5B,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,SAAS,CAAC,IAAI,CAAC,GAAG;gBAChB,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,YAAY,EAAE,KAAK,CAAC,OAAO;aAC5B,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,2BAA2B;QAC7B,CAAC;IACH,CAAC;IAED,MAAM,CAAC,OAAO,EAAE,AAAD,EAAG,YAAY,CAAC,GAAG,MAAM,yBAAyB,CAC/D,GAAG,EACH,SAAS,EACT,KAAK,EAAE,IAAY,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,EACxD,SAAS,CACV,CAAC;IAEF,qBAAqB;IACrB,MAAM,gBAAgB,GAAG;QACvB,SAAS,EAAE,KAAK;QAChB,GAAG,EAAE,SAAS;QACd,YAAY,EAAE,GAAG,EAAE,CAAC,SAAS;QAC7B,SAAS,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;QAC7C,iBAAiB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;KAC/C,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,kBAAkB,CACvC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,EACtB,gBAAgB,EAChB,GAAG,CACJ,CAAC;IAEF,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,UAAU,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO,EAAE,YAAY,EAAE,SAAS,CAAC,EAAE,CAAC;QACtF,mBAAmB;IACrB,CAAC;IAED,kBAAkB;IAClB,MAAM,QAAQ,GAAG,IAAI,2BAA2B,EAAE,CAAC;IACnD,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO,EAAE,YAAY,EAAE,SAAS,CAAC,EAAE,CAAC;QACpF,mBAAmB;IACrB,CAAC;IAED,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IACjC,OAAO,CAAC,KAAK,CAAC,2CAA2C,YAAY,EAAE,CAAC,CAAC;IACzE,OAAO,GAAG,CAAC;AACb,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,aAAa,CAC1B,KAAa,EACb,aAAuB,EACvB,GAAgB,EAChB,UAAkB;IAElB,MAAM,IAAI,GAAe,EAAE,CAAC;IAE5B,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;QAChC,MAAM,GAAG,GAAG,MAAM,cAAc,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACjB,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,2BAA2B,EAAE,CAAC;IACnD,OAAO,MAAM,QAAQ,CAAC,QAAQ,CAAC;QAC7B,CAAC,EAAE,UAAU;QACb,IAAI,EAAE,iBAAiB,CAAC,KAAK,CAAC;QAC9B,IAAI;QACJ,aAAa,EAAE,CAAC,GAAG;KACpB,CAAC,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,SAAS,oBAAoB,CAC3B,eAAwB,EACxB,cAAuB,EACvB,iBAAyB,GAAG,EAC5B,IAAY,EAAE;IAEd,MAAM,MAAM,GAAG,IAAI,GAAG,EAAiE,CAAC;IAExF,MAAM,aAAa,GAAG,CAAC,GAAG,cAAc,CAAC;IAEzC,yBAAyB;IACzB,eAAe,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACtC,MAAM,GAAG,GAAG,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QACpE,MAAM,QAAQ,GAAG,cAAc,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC;QAEjD,IAAI,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACpB,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;YAClC,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC;YAC3B,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE;gBACd,KAAK;gBACL,KAAK,EAAE,QAAQ;gBACf,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;aAC/B,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,wBAAwB;IACxB,cAAc,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACrC,MAAM,GAAG,GAAG,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QACpE,MAAM,QAAQ,GAAG,aAAa,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC;QAEhD,IAAI,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACpB,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;YAClC,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC;YAC3B,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE;gBACd,KAAK;gBACL,KAAK,EAAE,QAAQ;gBACf,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;aAC9B,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,IAAwB,EACxB,aAAgC,EAChC,GAAgB;IAEhB,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC;IAC5E,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,QAAQ,CAAC;IACnC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC;IAC1C,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,IAAI,KAAK,CAAC;IACrD,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,IAAI,GAAG,CAAC;IAEnD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,IAAI,YAAgC,CAAC;IACrC,IAAI,WAA+B,CAAC;IAEpC,yDAAyD;IACzD,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;QAC7C,MAAM,SAAS,GAAG,MAAM,uBAAuB,EAAE,CAAC;QAClD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,4EAA4E;gBAC5E,kDAAkD,CACnD,CAAC;QACJ,CAAC;IACH,CAAC;IAED,IAAI,eAAe,GAAY,EAAE,CAAC;IAClC,IAAI,cAAc,GAAY,EAAE,CAAC;IAEjC,0BAA0B;IAC1B,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;QAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,eAAe,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE;YAC1D,UAAU,EAAE,UAAU,GAAG,CAAC,EAAE,sCAAsC;SACnE,CAAC,CAAC;QACH,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC;IACvC,CAAC;IAED,yBAAyB;IACzB,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,cAAc,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC;QAC5E,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC;IACtC,CAAC;IAED,gBAAgB;IAChB,IAAI,YAAY,GAAoC,EAAE,CAAC;IAEvD,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;QACtB,8BAA8B;QAC9B,MAAM,WAAW,GAAG,oBAAoB,CACtC,eAAe,EACf,cAAc,EACd,cAAc,CACf,CAAC;QAEF,2CAA2C;QAC3C,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;aAC5C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;aACjC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;QAExB,YAAY,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;YACxD,IAAI,EAAE,KAAK,CAAC,QAAQ;YACpB,UAAU,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAqB;YAChE,OAAO,EAAE,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;YACnD,KAAK;YACL,MAAM,EAAE,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;SACzF,CAAC,CAAC,CAAC;IACN,CAAC;SAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;QAC/B,YAAY,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;YACvE,IAAI,EAAE,KAAK,CAAC,QAAQ;YACpB,UAAU,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAqB;YAChE,OAAO,EAAE,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;YACnD,KAAK,EAAE,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,EAAE,wBAAwB;YAC9C,MAAM,EAAE,UAAmB;SAC5B,CAAC,CAAC,CAAC;IACN,CAAC;SAAM,CAAC;QACN,YAAY,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;YACtE,IAAI,EAAE,KAAK,CAAC,QAAQ;YACpB,UAAU,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAqB;YAChE,OAAO,EAAE,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;YACnD,KAAK,EAAE,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC;YACpB,MAAM,EAAE,SAAkB;SAC3B,CAAC,CAAC,CAAC;IACN,CAAC;IAED,gBAAgB;IAChB,IAAI,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;QACpC,YAAY,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;YACvC,MAAM,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;YACpC,OAAO,GAAG,IAAI,IAAI,CAAC,MAAO,CAAC,UAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;QACjC,YAAY,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;YACvC,OAAO,CAAC,IAAI,CAAC,MAAO,CAAC,OAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;IAE1C,OAAO;QACL,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,IAAI;QACJ,OAAO,EAAE,YAAY;QACrB,KAAK,EAAE,YAAY,CAAC,MAAM;QAC1B,WAAW,EAAE;YACX,UAAU;YACV,YAAY;YACZ,WAAW;SACZ;KACF,CAAC;AACJ,CAAC;AAED,eAAe,kBAAkB,CAAC"}

package/dist/tools/workspace-context.d.ts

@@ -0,0 +1,20 @@
+import type { SelectionManager } from '../selection/SelectionManager.js';
+export interface WorkspaceContextArgs {
+    include?: Array<'selection' | 'code' | 'files' | 'tree' | 'tokens'>;
+}
+export interface WorkspaceContextResult {
+    selection?: {
+        files: any[];
+        totalFiles: number;
+        totalTokens: number;
+    };
+    code_structures?: any;
+    file_contents?: string;
+    tree?: any;
+    tokens?: number;
+}
+/**
+ * Get comprehensive workspace context snapshot
+ */
+export declare function workspaceContextTool(args: WorkspaceContextArgs, manager: SelectionManager, workspaceDir: string): Promise<WorkspaceContextResult>;
+//# sourceMappingURL=workspace-context.d.ts.map

package/dist/tools/workspace-context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace-context.d.ts","sourceRoot":"","sources":["../../src/tools/workspace-context.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AAGzE,MAAM,WAAW,oBAAoB;IACnC,OAAO,CAAC,EAAE,KAAK,CAAC,WAAW,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,QAAQ,CAAC,CAAC;CACrE;AAED,MAAM,WAAW,sBAAsB;IACrC,SAAS,CAAC,EAAE;QACV,KAAK,EAAE,GAAG,EAAE,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,eAAe,CAAC,EAAE,GAAG,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,IAAI,CAAC,EAAE,GAAG,CAAC;IACX,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CACxC,IAAI,EAAE,oBAAoB,EAC1B,OAAO,EAAE,gBAAgB,EACzB,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC,sBAAsB,CAAC,CAsCjC"}

package/dist/tools/workspace-context.js

@@ -0,0 +1,36 @@
+import { getFileTreeTool } from './file-tree.js';
+/**
+ * Get comprehensive workspace context snapshot
+ */
+export async function workspaceContextTool(args, manager, workspaceDir) {
+    const include = args.include || ['selection', 'tokens'];
+    const result = {};
+    try {
+        if (include.includes('selection')) {
+            result.selection = await manager.getSummary();
+        }
+        if (include.includes('code')) {
+            result.code_structures = {
+                note: 'Code structure extraction will be added in V2',
+            };
+        }
+        if (include.includes('files')) {
+            result.file_contents = await manager.getContent();
+        }
+        if (include.includes('tree')) {
+            result.tree = await getFileTreeTool({
+                type: 'files',
+                mode: 'full',
+            }, workspaceDir);
+        }
+        if (include.includes('tokens')) {
+            const preview = await manager.preview();
+            result.tokens = preview.tokens;
+        }
+    }
+    catch (error) {
+        console.error('Workspace context error:', error);
+    }
+    return result;
+}
+//# sourceMappingURL=workspace-context.js.map

package/dist/tools/workspace-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace-context.js","sourceRoot":"","sources":["../../src/tools/workspace-context.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAkBjD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,IAA0B,EAC1B,OAAyB,EACzB,YAAoB;IAEpB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;IACxD,MAAM,MAAM,GAA2B,EAAE,CAAC;IAE1C,IAAI,CAAC;QACH,IAAI,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YAClC,MAAM,CAAC,SAAS,GAAG,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;QAChD,CAAC;QAED,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7B,MAAM,CAAC,eAAe,GAAG;gBACvB,IAAI,EAAE,+CAA+C;aACtD,CAAC;QACJ,CAAC;QAED,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9B,MAAM,CAAC,aAAa,GAAG,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;QACpD,CAAC;QAED,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7B,MAAM,CAAC,IAAI,GAAG,MAAM,eAAe,CACjC;gBACE,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,MAAM;aACb,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC/B,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;YACxC,MAAM,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QACjC,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;IACnD,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/tree-sitter/code-snippet-queries/bash.scm

@@ -0,0 +1,7 @@
+(
+  (comment)? @comment
+  (function_definition
+    name: (word) @name
+    body: (_) @body
+  ) @definition
+)