@phmotad/fire-code 0.2.8 → 0.2.9

package/dist/mcp/tools/smart_search.js

@@ -5,7 +5,8 @@ exports.smartSearchTool = smartSearchTool;
 const zod_1 = require("zod");
 const path_1 = require("path");
 const fs_1 = require("fs");
-const path_2 = require("path");
+const DatabaseManager_js_1 = require("../../db/DatabaseManager.js");
+const paths_js_1 = require("../../utils/paths.js");
 exports.SmartSearchInputSchema = zod_1.z.object({
     query: zod_1.z.string().describe('Search term — matches symbol names, file names, and content'),
     path: zod_1.z.string().optional().describe('Root directory to search (default: cwd)'),
@@ -14,6 +15,15 @@ exports.SmartSearchInputSchema = zod_1.z.object({
 });
 const SKIP_DIRS = new Set(['node_modules', '.git', 'dist', 'build', '.firecode', 'coverage', '.next']);
 const CODE_EXTS = new Set(['.ts', '.tsx', '.js', '.jsx', '.mts', '.cts', '.py', '.go', '.rs', '.java', '.cs', '.rb', '.php', '.swift', '.kt', '.md']);
+function getProjectName(rootDir) {
+    try {
+        const pkg = JSON.parse((0, fs_1.readFileSync)((0, path_1.join)(rootDir, 'package.json'), 'utf8'));
+        return pkg.name ?? (0, path_1.basename)(rootDir);
+    }
+    catch {
+        return (0, path_1.basename)(rootDir);
+    }
+}
 function walkFiles(dir, filePattern) {
     const files = [];
     function walk(current) {
@@ -27,7 +37,7 @@ function walkFiles(dir, filePattern) {
         for (const entry of entries) {
             if (SKIP_DIRS.has(entry))
                 continue;
-            const full = (0, path_2.join)(current, entry);
+            const full = (0, path_1.join)(current, entry);
             let stat;
             try {
                 stat = (0, fs_1.statSync)(full);
@@ -38,7 +48,7 @@ function walkFiles(dir, filePattern) {
             if (stat.isDirectory()) {
                 walk(full);
             }
-            else if (CODE_EXTS.has((0, path_2.extname)(entry))) {
+            else if (CODE_EXTS.has((0, path_1.extname)(entry))) {
                 if (!filePattern || full.includes(filePattern)) {
                     files.push(full);
                 }
@@ -48,35 +58,89 @@ function walkFiles(dir, filePattern) {
     walk(dir);
     return files;
 }
-function scoreMatch(line, query) {
-    const lq = query.toLowerCase();
+function scoreMatch(line, queryToken) {
+    const lq = queryToken.toLowerCase();
     const ll = line.toLowerCase();
-    // Exact symbol match (function name, class name)
+    // Exact symbol declaration
     const symMatch = line.match(/(?:function|class|interface|type|const|let|var)\s+(\w+)/);
-    if (symMatch && symMatch[1].toLowerCase().includes(lq))
+    if (symMatch && symMatch[1].toLowerCase() === lq)
         return 100;
+    if (symMatch && symMatch[1].toLowerCase().includes(lq))
+        return 80;
+    // Method declaration (handles class methods: `  methodName(`)
+    const methodMatch = line.match(/^\s+(?:async\s+|static\s+|private\s+|public\s+|protected\s+)*(\w+)\s*\(/);
+    if (methodMatch && methodMatch[1].toLowerCase() === lq)
+        return 90;
     // Export match
     if (line.includes('export') && ll.includes(lq))
-        return 80;
+        return 70;
     // Import match
     if (line.includes('import') && ll.includes(lq))
-        return 60;
+        return 50;
     // General content
     if (ll.includes(lq))
-        return 40;
+        return 30;
     return 0;
 }
+/** Query the indexed graph for exact symbol matches (fast, precise). */
+function graphSymbolSearch(tokens, rootDir) {
+    const firedotDir = (0, paths_js_1.getFireCodeDir)(rootDir);
+    if (!(0, fs_1.existsSync)(firedotDir))
+        return [];
+    try {
+        const project = getProjectName(rootDir);
+        const db = DatabaseManager_js_1.DatabaseManager.getInstance(firedotDir);
+        const graphStore = db.getGraphStore(project);
+        const results = [];
+        for (const token of tokens) {
+            if (token.length < 2)
+                continue;
+            // Exact function/method match
+            const fnNodes = graphStore.query({ type: 'function', label: token, exact: true });
+            for (const node of fnNodes) {
+                const filePath = node.filePath.replace(/\\/g, '/');
+                const label = node.parentClass ? `${node.parentClass}.${node.label}` : node.label;
+                results.push({
+                    file: filePath,
+                    line: node.line,
+                    kind: 'graph',
+                    match: `${label}(${node.parameters.join(', ')})${node.returnType ? ': ' + node.returnType : ''}`,
+                    context: `defined at L${node.line}${node.parentClass ? ` [class: ${node.parentClass}]` : ''}`,
+                });
+            }
+            // Exact file name match
+            const fileNodes = graphStore.query({ type: 'file', label: token, exact: true });
+            for (const node of fileNodes) {
+                results.push({
+                    file: (node.path ?? node.label).replace(/\\/g, '/'),
+                    line: 0,
+                    kind: 'graph',
+                    match: node.label,
+                    context: `file (exports: ${node.exports.slice(0, 4).join(', ')})`,
+                });
+            }
+        }
+        return results;
+    }
+    catch {
+        return [];
+    }
+}
 async function smartSearchTool(input, cwd) {
     const rootDir = (0, path_1.resolve)(input.path ?? cwd);
     if (!(0, fs_1.existsSync)(rootDir)) {
         return JSON.stringify({ error: `Directory not found: ${rootDir}` });
     }
-    const files = walkFiles(rootDir, input.file_pattern);
-    const results = [];
-    const query = input.query.toLowerCase();
+    // Split query into individual tokens for multi-symbol queries
+    const tokens = input.query.split(/[\s,]+/).filter(t => t.length > 1);
     const maxResults = input.max_results ?? 20;
+    // ── 1. Graph-based symbol lookup (highest precision) ─────────────────────
+    const graphResults = graphSymbolSearch(tokens, rootDir);
+    // ── 2. File scan (catches anything not yet indexed) ───────────────────────
+    const files = walkFiles(rootDir, input.file_pattern);
+    const fileResults = [];
     for (const filePath of files) {
-        if (results.length >= maxResults * 3)
+        if (fileResults.length >= maxResults * 4)
             break;
         let content;
         try {
@@ -85,47 +149,76 @@ async function smartSearchTool(input, cwd) {
         catch {
             continue;
         }
-        const relPath = (0, path_2.relative)(rootDir, filePath);
-        // File name match
-        if (relPath.toLowerCase().includes(query)) {
-            results.push({ file: relPath, line: 0, kind: 'symbol', match: relPath, context: `file: ${relPath}` });
+        const relPath = (0, path_1.relative)(rootDir, filePath).replace(/\\/g, '/');
+        // File name match against any token
+        const fileNameLower = relPath.toLowerCase();
+        if (tokens.some(t => fileNameLower.includes(t.toLowerCase()))) {
+            fileResults.push({ file: relPath, line: 0, kind: 'symbol', match: relPath, context: `file: ${relPath}` });
         }
         const lines = content.split('\n');
         for (let i = 0; i < lines.length; i++) {
-            const score = scoreMatch(lines[i], query);
-            if (score > 0) {
+            // Score against each token, take highest
+            let bestScore = 0;
+            for (const token of tokens) {
+                const s = scoreMatch(lines[i], token);
+                if (s > bestScore)
+                    bestScore = s;
+            }
+            if (bestScore > 0) {
                 const ctx = lines.slice(Math.max(0, i - 1), i + 2).join('\n');
-                results.push({
+                fileResults.push({
                     file: relPath,
                     line: i + 1,
-                    kind: score >= 80 ? 'symbol' : 'content',
+                    kind: bestScore >= 70 ? 'symbol' : 'content',
                     match: lines[i].trim().slice(0, 120),
                     context: ctx,
                 });
             }
         }
     }
-    // Sort: symbols first, then by file name
-    results.sort((a, b) => {
+    // Sort file results: symbols first, then by file name
+    fileResults.sort((a, b) => {
         if (a.kind !== b.kind)
             return a.kind === 'symbol' ? -1 : 1;
         return a.file.localeCompare(b.file);
     });
-    const top = results.slice(0, maxResults);
+    // ── 3. Merge: graph results first (deduplicated by file+line) ────────────
+    const seen = new Set();
+    const merged = [];
+    for (const r of graphResults) {
+        const key = `${r.file}:${r.line}`;
+        if (!seen.has(key)) {
+            seen.add(key);
+            merged.push(r);
+        }
+    }
+    for (const r of fileResults) {
+        const key = `${r.file}:${r.line}`;
+        if (!seen.has(key)) {
+            seen.add(key);
+            merged.push(r);
+        }
+    }
+    const top = merged.slice(0, maxResults);
     if (top.length === 0) {
         return `No results for "${input.query}" in ${rootDir}`;
     }
-    const lines = [`# Search: "${input.query}" — ${top.length} results (${files.length} files scanned)\n`];
+    const outputLines = [
+        `# Search: "${input.query}" — ${top.length} results (${graphResults.length} from index, ${files.length} files scanned)\n`,
+    ];
     let lastFile = '';
     for (const r of top) {
         if (r.file !== lastFile) {
-            lines.push(`\n## ${r.file}`);
+            outputLines.push(`\n## ${r.file}`);
             lastFile = r.file;
         }
         if (r.line > 0) {
-            lines.push(`  L${r.line}  [${r.kind}]  ${r.match}`);
+            outputLines.push(`  L${r.line}  [${r.kind}]  ${r.match}`);
+        }
+        else if (r.kind === 'graph') {
+            outputLines.push(`  [${r.kind}]  ${r.match}  — ${r.context}`);
         }
     }
-    return lines.join('\n');
+    return outputLines.join('\n');
 }
 //# sourceMappingURL=smart_search.js.map

package/dist/mcp/tools/smart_search.js.map

@@ -1 +1 @@
-{"version":3,"file":"smart_search.js","sourceRoot":"","sources":["../../../src/mcp/tools/smart_search.ts"],"names":[],"mappings":";;;AAwEA,0CAmEC;AA3ID,6BAAwB;AACxB,+BAA+B;AAC/B,2BAAqE;AACrE,+BAA+C;AAElC,QAAA,sBAAsB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7C,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6DAA6D,CAAC;IACzF,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;IAC/E,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IACpF,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4DAA4D,CAAC;CAC3G,CAAC,CAAC;AAIH,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;AACvG,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;AAUtJ,SAAS,SAAS,CAAC,GAAW,EAAE,WAAoB;IAClD,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,SAAS,IAAI,CAAC,OAAe;QAC3B,IAAI,OAAiB,CAAC;QACtB,IAAI,CAAC;YAAC,OAAO,GAAG,IAAA,gBAAW,EAAC,OAAO,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC;YAAC,OAAO;QAAC,CAAC;QAEzD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,IAAI,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC;gBAAE,SAAS;YACnC,MAAM,IAAI,GAAG,IAAA,WAAI,EAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAClC,IAAI,IAAiC,CAAC;YACtC,IAAI,CAAC;gBAAC,IAAI,GAAG,IAAA,aAAQ,EAAC,IAAI,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC;gBAAC,SAAS;YAAC,CAAC;YAElD,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBACvB,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,CAAC;iBAAM,IAAI,SAAS,CAAC,GAAG,CAAC,IAAA,cAAO,EAAC,KAAK,CAAC,CAAC,EAAE,CAAC;gBACzC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC/C,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,CAAC;IACV,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,UAAU,CAAC,IAAY,EAAE,KAAa;IAC7C,MAAM,EAAE,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAC/B,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IAE9B,iDAAiD;IACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,yDAAyD,CAAC,CAAC;IACvF,IAAI,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAAE,OAAO,GAAG,CAAC;IAEnE,eAAe;IACf,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAAE,OAAO,EAAE,CAAC;IAE1D,eAAe;IACf,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAAE,OAAO,EAAE,CAAC;IAE1D,kBAAkB;IAClB,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAAE,OAAO,EAAE,CAAC;IAE/B,OAAO,CAAC,CAAC;AACX,CAAC;AAEM,KAAK,UAAU,eAAe,CAAC,KAAuB,EAAE,GAAW;IACxE,MAAM,OAAO,GAAG,IAAA,cAAO,EAAC,KAAK,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;IAE3C,IAAI,CAAC,IAAA,eAAU,EAAC,OAAO,CAAC,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,wBAAwB,OAAO,EAAE,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,KAAK,GAAG,SAAS,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;IACrD,MAAM,OAAO,GAAmB,EAAE,CAAC;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;IACxC,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,IAAI,EAAE,CAAC;IAE3C,KAAK,MAAM,QAAQ,IAAI,KAAK,EAAE,CAAC;QAC7B,IAAI,OAAO,CAAC,MAAM,IAAI,UAAU,GAAG,CAAC;YAAE,MAAM;QAE5C,IAAI,OAAe,CAAC;QACpB,IAAI,CAAC;YAAC,OAAO,GAAG,IAAA,iBAAY,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC;YAAC,SAAS;QAAC,CAAC;QAErE,MAAM,OAAO,GAAG,IAAA,eAAQ,EAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAE5C,kBAAkB;QAClB,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,OAAO,EAAE,EAAE,CAAC,CAAC;QACxG,CAAC;QAED,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;YAC1C,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;gBACd,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC9D,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,CAAC,GAAG,CAAC;oBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;oBACxC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;oBACpC,OAAO,EAAE,GAAG;iBACb,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,yCAAyC;IACzC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACpB,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI;YAAE,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAEzC,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrB,OAAO,mBAAmB,KAAK,CAAC,KAAK,QAAQ,OAAO,EAAE,CAAC;IACzD,CAAC;IAED,MAAM,KAAK,GAAG,CAAC,cAAc,KAAK,CAAC,KAAK,OAAO,GAAG,CAAC,MAAM,aAAa,KAAK,CAAC,MAAM,mBAAmB,CAAC,CAAC;IACvG,IAAI,QAAQ,GAAG,EAAE,CAAC;IAElB,KAAK,MAAM,CAAC,IAAI,GAAG,EAAE,CAAC;QACpB,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACxB,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YAC7B,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACf,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}
+{"version":3,"file":"smart_search.js","sourceRoot":"","sources":["../../../src/mcp/tools/smart_search.ts"],"names":[],"mappings":";;;AAqIA,0CAgGC;AArOD,6BAAwB;AACxB,+BAAkE;AAClE,2BAAqE;AACrE,oEAA8D;AAC9D,mDAAsD;AAGzC,QAAA,sBAAsB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7C,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6DAA6D,CAAC;IACzF,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;IAC/E,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IACpF,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4DAA4D,CAAC;CAC3G,CAAC,CAAC;AAIH,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;AACvG,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;AAUtJ,SAAS,cAAc,CAAC,OAAe;IACrC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAA,iBAAY,EAAC,IAAA,WAAI,EAAC,OAAO,EAAE,cAAc,CAAC,EAAE,MAAM,CAAC,CAAsB,CAAC;QACjG,OAAO,GAAG,CAAC,IAAI,IAAI,IAAA,eAAQ,EAAC,OAAO,CAAC,CAAC;IACvC,CAAC;IAAC,MAAM,CAAC;QAAC,OAAO,IAAA,eAAQ,EAAC,OAAO,CAAC,CAAC;IAAC,CAAC;AACvC,CAAC;AAED,SAAS,SAAS,CAAC,GAAW,EAAE,WAAoB;IAClD,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,SAAS,IAAI,CAAC,OAAe;QAC3B,IAAI,OAAiB,CAAC;QACtB,IAAI,CAAC;YAAC,OAAO,GAAG,IAAA,gBAAW,EAAC,OAAO,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC;YAAC,OAAO;QAAC,CAAC;QAEzD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,IAAI,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC;gBAAE,SAAS;YACnC,MAAM,IAAI,GAAG,IAAA,WAAI,EAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAClC,IAAI,IAAiC,CAAC;YACtC,IAAI,CAAC;gBAAC,IAAI,GAAG,IAAA,aAAQ,EAAC,IAAI,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC;gBAAC,SAAS;YAAC,CAAC;YAElD,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBACvB,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,CAAC;iBAAM,IAAI,SAAS,CAAC,GAAG,CAAC,IAAA,cAAO,EAAC,KAAK,CAAC,CAAC,EAAE,CAAC;gBACzC,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC/C,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,CAAC;IACV,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,UAAU,CAAC,IAAY,EAAE,UAAkB;IAClD,MAAM,EAAE,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC;IACpC,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IAE9B,2BAA2B;IAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,yDAAyD,CAAC,CAAC;IACvF,IAAI,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,EAAE;QAAE,OAAO,GAAG,CAAC;IAC7D,IAAI,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAAE,OAAO,EAAE,CAAC;IAElE,8DAA8D;IAC9D,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,yEAAyE,CAAC,CAAC;IAC1G,IAAI,WAAW,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,EAAE;QAAE,OAAO,EAAE,CAAC;IAElE,eAAe;IACf,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAAE,OAAO,EAAE,CAAC;IAE1D,eAAe;IACf,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAAE,OAAO,EAAE,CAAC;IAE1D,kBAAkB;IAClB,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAAE,OAAO,EAAE,CAAC;IAE/B,OAAO,CAAC,CAAC;AACX,CAAC;AAED,wEAAwE;AACxE,SAAS,iBAAiB,CAAC,MAAgB,EAAE,OAAe;IAC1D,MAAM,UAAU,GAAG,IAAA,yBAAc,EAAC,OAAO,CAAC,CAAC;IAC3C,IAAI,CAAC,IAAA,eAAU,EAAC,UAAU,CAAC;QAAE,OAAO,EAAE,CAAC;IAEvC,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;QACxC,MAAM,EAAE,GAAG,oCAAe,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QACnD,MAAM,UAAU,GAAG,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,OAAO,GAAmB,EAAE,CAAC;QAEnC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC;gBAAE,SAAS;YAE/B,8BAA8B;YAC9B,MAAM,OAAO,GAAG,UAAU,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,CAAmB,CAAC;YACpG,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;gBAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;gBACnD,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;gBAClF,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,GAAG,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE;oBAChG,OAAO,EAAE,eAAe,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;iBAC9F,CAAC,CAAC;YACL,CAAC;YAED,wBAAwB;YACxB,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,CAAe,CAAC;YAC9F,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;gBAC7B,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;oBACnD,IAAI,EAAE,CAAC;oBACP,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,OAAO,EAAE,kBAAkB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG;iBAClE,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,eAAe,CAAC,KAAuB,EAAE,GAAW;IACxE,MAAM,OAAO,GAAG,IAAA,cAAO,EAAC,KAAK,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;IAE3C,IAAI,CAAC,IAAA,eAAU,EAAC,OAAO,CAAC,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,wBAAwB,OAAO,EAAE,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,8DAA8D;IAC9D,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACrE,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,IAAI,EAAE,CAAC;IAE3C,4EAA4E;IAC5E,MAAM,YAAY,GAAG,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAExD,6EAA6E;IAC7E,MAAM,KAAK,GAAG,SAAS,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;IACrD,MAAM,WAAW,GAAmB,EAAE,CAAC;IAEvC,KAAK,MAAM,QAAQ,IAAI,KAAK,EAAE,CAAC;QAC7B,IAAI,WAAW,CAAC,MAAM,IAAI,UAAU,GAAG,CAAC;YAAE,MAAM;QAEhD,IAAI,OAAe,CAAC;QACpB,IAAI,CAAC;YAAC,OAAO,GAAG,IAAA,iBAAY,EAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC;YAAC,SAAS;QAAC,CAAC;QAErE,MAAM,OAAO,GAAG,IAAA,eAAQ,EAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAEhE,oCAAoC;QACpC,MAAM,aAAa,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAC5C,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,EAAE,CAAC;YAC9D,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,OAAO,EAAE,EAAE,CAAC,CAAC;QAC5G,CAAC;QAED,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,yCAAyC;YACzC,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAC3B,MAAM,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;gBACtC,IAAI,CAAC,GAAG,SAAS;oBAAE,SAAS,GAAG,CAAC,CAAC;YACnC,CAAC;YACD,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;gBAClB,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC9D,WAAW,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,CAAC,GAAG,CAAC;oBACX,IAAI,EAAE,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;oBAC5C,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;oBACpC,OAAO,EAAE,GAAG;iBACb,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,sDAAsD;IACtD,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACxB,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI;YAAE,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAC5E,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,MAAM,GAAmB,EAAE,CAAC;IAElC,KAAK,MAAM,CAAC,IAAI,YAAY,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAAC,CAAC;IACxD,CAAC;IACD,KAAK,MAAM,CAAC,IAAI,WAAW,EAAE,CAAC;QAC5B,MAAM,GAAG,GAAG,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAAC,CAAC;IACxD,CAAC;IAED,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAExC,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrB,OAAO,mBAAmB,KAAK,CAAC,KAAK,QAAQ,OAAO,EAAE,CAAC;IACzD,CAAC;IAED,MAAM,WAAW,GAAG;QAClB,cAAc,KAAK,CAAC,KAAK,OAAO,GAAG,CAAC,MAAM,aAAa,YAAY,CAAC,MAAM,gBAAgB,KAAK,CAAC,MAAM,mBAAmB;KAC1H,CAAC;IACF,IAAI,QAAQ,GAAG,EAAE,CAAC;IAElB,KAAK,MAAM,CAAC,IAAI,GAAG,EAAE,CAAC;QACpB,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACxB,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YACnC,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC;QACpB,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACf,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QAC5D,CAAC;aAAM,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC9B,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,KAAK,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChC,CAAC"}

package/dist/memory/HybridMemory.d.ts

@@ -9,12 +9,16 @@ export interface RetrieveOptions {
     k?: number;
     includeGraph?: boolean;
     graphDepth?: number;
+    workingDiff?: string;
 }
 export declare class HybridMemory {
     private readonly vectorStore;
     private readonly graphStore;
     constructor(vectorStore: VectorStore, graphStore: GraphStore);
     retrieve(query: string, opts?: RetrieveOptions): Promise<RetrievedContext>;
+    private retrieveGraphContext;
+    /** Find commits (from graph) that changed any of the given file paths. */
+    private getCommitsForFiles;
     private buildContextString;
 }
 //# sourceMappingURL=HybridMemory.d.ts.map

package/dist/memory/HybridMemory.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"HybridMemory.d.ts","sourceRoot":"","sources":["../../src/memory/HybridMemory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC5E,OAAO,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAGpE,MAAM,WAAW,gBAAgB;IAC/B,aAAa,EAAE,cAAc,EAAE,CAAC;IAChC,YAAY,EAAE,SAAS,EAAE,CAAC;IAC1B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,eAAe;IAC9B,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,YAAY;IAErB,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,UAAU;gBADV,WAAW,EAAE,WAAW,EACxB,UAAU,EAAE,UAAU;IAGnC,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,eAAoB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA4CpF,OAAO,CAAC,kBAAkB;CA6B3B"}
+{"version":3,"file":"HybridMemory.d.ts","sourceRoot":"","sources":["../../src/memory/HybridMemory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC5E,OAAO,KAAK,EAAE,UAAU,EAAE,SAAS,EAAsC,MAAM,wBAAwB,CAAC;AAGxG,MAAM,WAAW,gBAAgB;IAC/B,aAAa,EAAE,cAAc,EAAE,CAAC;IAChC,YAAY,EAAE,SAAS,EAAE,CAAC;IAC1B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,eAAe;IAC9B,CAAC,CAAC,EAAE,MAAM,CAAC;IACX,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,qBAAa,YAAY;IAErB,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,UAAU;gBADV,WAAW,EAAE,WAAW,EACxB,UAAU,EAAE,UAAU;IAGnC,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,GAAE,eAAoB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAiBpF,OAAO,CAAC,oBAAoB;IAyD5B,0EAA0E;IAC1E,OAAO,CAAC,kBAAkB;IAqB1B,OAAO,CAAC,kBAAkB;CA+E3B"}

package/dist/memory/HybridMemory.js

@@ -16,57 +16,148 @@ class HybridMemory {
         const vectorResults = await this.vectorStore.search(query, k);
         let graphResults = [];
         if (includeGraph) {
-            // Use vector result paths to seed graph traversal
-            const paths = new Set();
-            for (const r of vectorResults) {
-                const relPath = r.document.metadata.relativePath;
-                if (relPath)
-                    paths.add(relPath);
+            graphResults = this.retrieveGraphContext(query, vectorResults);
+        }
+        const combined = this.buildContextString(query, vectorResults, graphResults, opts.workingDiff);
+        return { vectorResults, graphResults, combined };
+    }
+    retrieveGraphContext(query, vectorResults) {
+        const stopWords = new Set(['the', 'and', 'for', 'with', 'from', 'that', 'this', 'are', 'has', 'have', 'not', 'but', 'can', 'when']);
+        const tokens = query
+            .toLowerCase()
+            .split(/[\s\W]+/)
+            .filter(t => t.length > 2 && !stopWords.has(t));
+        const results = [];
+        const seen = new Set();
+        const add = (node) => {
+            if (!seen.has(node.id)) {
+                seen.add(node.id);
+                results.push(node);
             }
-            for (const path of paths) {
-                const fileNode = this.graphStore.getNode(`file:${path}`);
-                if (fileNode) {
-                    graphResults.push(fileNode);
-                    const neighbors = this.graphStore.getNeighbors(fileNode.id);
-                    graphResults.push(...neighbors.slice(0, 3));
+        };
+        // 1. Exact symbol matches — query each token
+        for (const token of tokens) {
+            const fnNodes = this.graphStore.query({ type: 'function', label: token, exact: true });
+            fnNodes.forEach(add);
+        }
+        // 2. Seed graph traversal from vector search results
+        const paths = new Set();
+        for (const r of vectorResults) {
+            const relPath = r.document.metadata.relativePath;
+            if (relPath)
+                paths.add(relPath);
+        }
+        for (const path of paths) {
+            const fileNode = this.graphStore.getNode(`file:${path}`);
+            if (fileNode) {
+                add(fileNode);
+                const neighbors = this.graphStore.getNeighbors(fileNode.id);
+                neighbors.slice(0, 5).forEach(add);
+            }
+        }
+        // 3. Substring keyword matches
+        for (const token of tokens.slice(0, 3)) {
+            const partial = this.graphStore.query({ label: token }).slice(0, 5);
+            partial.forEach(add);
+        }
+        // 4. Sort: exact matches first
+        const tokenSet = new Set(tokens);
+        results.sort((a, b) => {
+            const score = (n) => {
+                const label = n.label.toLowerCase();
+                if (tokenSet.has(label))
+                    return 100;
+                if (tokens.some(t => label.includes(t) || t.includes(label)))
+                    return 50;
+                return 0;
+            };
+            return score(b) - score(a);
+        });
+        return results;
+    }
+    /** Find commits (from graph) that changed any of the given file paths. */
+    getCommitsForFiles(filePaths) {
+        const commits = [];
+        const seen = new Set();
+        for (const path of filePaths) {
+            const normalizedPath = path.replace(/\\/g, '/');
+            // dependantsOf returns nodes with edges pointing TO fileNode — i.e., commits that changed it
+            const dependants = this.graphStore.dependantsOf(`file:${normalizedPath}`);
+            for (const node of dependants) {
+                if (node.type === 'commit' && !seen.has(node.id)) {
+                    seen.add(node.id);
+                    commits.push(node);
                 }
             }
-            // Also query graph by keyword
-            const graphKeywords = this.graphStore.query({ label: query.split(' ')[0] }).slice(0, 3);
-            graphResults.push(...graphKeywords);
-            // Deduplicate
-            const seen = new Set();
-            graphResults = graphResults.filter((n) => {
-                if (seen.has(n.id))
-                    return false;
-                seen.add(n.id);
-                return true;
-            });
         }
-        const combined = this.buildContextString(query, vectorResults, graphResults);
-        return { vectorResults, graphResults, combined };
+        // Sort by timestamp descending (most recent first)
+        commits.sort((a, b) => new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime());
+        return commits.slice(0, 8);
     }
-    buildContextString(query, vectorResults, graphResults) {
+    buildContextString(query, vectorResults, graphResults, workingDiff) {
         const parts = [`# Context for: "${query}"\n`];
+        // ── Code Snippets ──────────────────────────────────────────────────────
         if (vectorResults.length > 0) {
             parts.push('## Relevant Code Snippets\n');
             for (const r of vectorResults) {
                 const path = r.document.metadata.relativePath ?? r.document.metadata.path ?? 'unknown';
-                parts.push(`### ${path} (score: ${r.score.toFixed(3)})\n\`\`\`\n${r.document.text.slice(0, 400)}\n\`\`\`\n`);
+                const scoreLabel = r.score > 0 ? ` (score: ${r.score.toFixed(3)})` : '';
+                parts.push(`### ${path}${scoreLabel}\n\`\`\`\n${r.document.text.slice(0, 400)}\n\`\`\`\n`);
             }
         }
+        // ── Structural Context ─────────────────────────────────────────────────
         if (graphResults.length > 0) {
             parts.push('## Structural Context\n');
+            const byFile = new Map();
+            const fileNodes = [];
             for (const node of graphResults) {
                 if (node.type === 'file') {
-                    parts.push(`- File: \`${node.path}\` (exports: ${node.exports.slice(0, 5).join(', ')})`);
+                    fileNodes.push(node);
                 }
                 else if (node.type === 'function') {
-                    parts.push(`- Function: \`${node.label}\` in \`${node.filePath}\`:${node.line}`);
+                    const fn = node;
+                    const key = fn.filePath.replace(/\\/g, '/');
+                    if (!byFile.has(key))
+                        byFile.set(key, []);
+                    byFile.get(key).push(fn);
                 }
             }
+            for (const [filePath, fns] of byFile) {
+                parts.push(`### \`${filePath}\``);
+                for (const fn of fns.slice(0, 8)) {
+                    const qualifiedName = fn.parentClass ? `${fn.parentClass}.${fn.label}` : fn.label;
+                    const sig = `${qualifiedName}(${fn.parameters.join(', ')})${fn.returnType ? ': ' + fn.returnType : ''}`;
+                    parts.push(`  - \`${sig}\` — L${fn.line}`);
+                }
+                parts.push('');
+            }
+            for (const node of fileNodes) {
+                parts.push(`- File: \`${node.path}\` (exports: ${node.exports.slice(0, 5).join(', ')})`);
+            }
             parts.push('');
         }
+        // ── Git History ────────────────────────────────────────────────────────
+        const filePaths = vectorResults
+            .map(r => (r.document.metadata.relativePath ?? r.document.metadata.path ?? ''))
+            .filter(Boolean);
+        if (filePaths.length > 0) {
+            const commits = this.getCommitsForFiles(filePaths);
+            if (commits.length > 0) {
+                parts.push('## Recent Git History\n');
+                for (const c of commits) {
+                    const date = new Date(c.timestamp).toLocaleDateString('pt-BR', { day: '2-digit', month: '2-digit', year: '2-digit' });
+                    const files = c.filesChanged.slice(0, 3).join(', ');
+                    const more = c.filesChanged.length > 3 ? ` +${c.filesChanged.length - 3}` : '';
+                    parts.push(`- \`${c.sha.slice(0, 7)}\` ${c.message} *(${date})* — ${files}${more}`);
+                }
+                parts.push('');
+            }
+        }
+        // ── Working Diff ───────────────────────────────────────────────────────
+        if (workingDiff && workingDiff.trim().length > 0) {
+            parts.push('## Uncommitted Changes (git diff HEAD)\n');
+            parts.push(`\`\`\`diff\n${workingDiff.slice(0, 2000)}\n\`\`\`\n`);
+        }
         return parts.join('\n');
     }
 }

package/dist/memory/HybridMemory.js.map

@@ -1 +1 @@
-{"version":3,"file":"HybridMemory.js","sourceRoot":"","sources":["../../src/memory/HybridMemory.ts"],"names":[],"mappings":";;;AAEA,kDAA4C;AAc5C,MAAa,YAAY;IAEJ;IACA;IAFnB,YACmB,WAAwB,EACxB,UAAsB;QADtB,gBAAW,GAAX,WAAW,CAAa;QACxB,eAAU,GAAV,UAAU,CAAY;IACtC,CAAC;IAEJ,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,OAAwB,EAAE;QACtD,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;QACtB,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC;QAE/C,kBAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,YAAY,EAAE,EAAE,oBAAoB,CAAC,CAAC;QAEnF,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAE9D,IAAI,YAAY,GAAgB,EAAE,CAAC;QACnC,IAAI,YAAY,EAAE,CAAC;YACjB,kDAAkD;YAClD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;YAChC,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;gBAC9B,MAAM,OAAO,GAAG,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAkC,CAAC;gBACvE,IAAI,OAAO;oBAAE,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAClC,CAAC;YAED,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;gBACzD,IAAI,QAAQ,EAAE,CAAC;oBACb,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBAC5B,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;oBAC5D,YAAY,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC9C,CAAC;YACH,CAAC;YAED,8BAA8B;YAC9B,MAAM,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACxF,YAAY,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;YAEpC,cAAc;YACd,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;YAC/B,YAAY,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;gBACvC,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;oBAAE,OAAO,KAAK,CAAC;gBACjC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACf,OAAO,IAAI,CAAC;YACd,CAAC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,aAAa,EAAE,YAAY,CAAC,CAAC;QAE7E,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC;IACnD,CAAC;IAEO,kBAAkB,CACxB,KAAa,EACb,aAA+B,EAC/B,YAAyB;QAEzB,MAAM,KAAK,GAAa,CAAC,mBAAmB,KAAK,KAAK,CAAC,CAAC;QAExD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,KAAK,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC1C,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;gBAC9B,MAAM,IAAI,GAAG,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,SAAS,CAAC;gBACvF,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,YAAY,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC;YAC/G,CAAC;QACH,CAAC;QAED,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,KAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YACtC,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;gBAChC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACzB,KAAK,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,gBAAgB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC3F,CAAC;qBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACpC,KAAK,CAAC,IAAI,CAAC,iBAAiB,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,QAAQ,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBACnF,CAAC;YACH,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF;AA/ED,oCA+EC"}
+{"version":3,"file":"HybridMemory.js","sourceRoot":"","sources":["../../src/memory/HybridMemory.ts"],"names":[],"mappings":";;;AAEA,kDAA4C;AAe5C,MAAa,YAAY;IAEJ;IACA;IAFnB,YACmB,WAAwB,EACxB,UAAsB;QADtB,gBAAW,GAAX,WAAW,CAAa;QACxB,eAAU,GAAV,UAAU,CAAY;IACtC,CAAC;IAEJ,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,OAAwB,EAAE;QACtD,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;QACtB,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC;QAE/C,kBAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,YAAY,EAAE,EAAE,oBAAoB,CAAC,CAAC;QAEnF,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAE9D,IAAI,YAAY,GAAgB,EAAE,CAAC;QACnC,IAAI,YAAY,EAAE,CAAC;YACjB,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QACjE,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAC/F,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC;IACnD,CAAC;IAEO,oBAAoB,CAAC,KAAa,EAAE,aAA+B;QACzE,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;QACpI,MAAM,MAAM,GAAG,KAAK;aACjB,WAAW,EAAE;aACb,KAAK,CAAC,SAAS,CAAC;aAChB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAElD,MAAM,OAAO,GAAgB,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAE/B,MAAM,GAAG,GAAG,CAAC,IAAe,EAAE,EAAE;YAC9B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;gBAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAAC,CAAC;QACpE,CAAC,CAAC;QAEF,6CAA6C;QAC7C,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACvF,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACvB,CAAC;QAED,qDAAqD;QACrD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;QAChC,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;YAC9B,MAAM,OAAO,GAAG,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAkC,CAAC;YACvE,IAAI,OAAO;gBAAE,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAClC,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;YACzD,IAAI,QAAQ,EAAE,CAAC;gBACb,GAAG,CAAC,QAAQ,CAAC,CAAC;gBACd,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;gBAC5D,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QAED,+BAA+B;QAC/B,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACvC,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACpE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACvB,CAAC;QAED,+BAA+B;QAC/B,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACpB,MAAM,KAAK,GAAG,CAAC,CAAY,EAAU,EAAE;gBACrC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;gBACpC,IAAI,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC;oBAAE,OAAO,GAAG,CAAC;gBACpC,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;oBAAE,OAAO,EAAE,CAAC;gBACxE,OAAO,CAAC,CAAC;YACX,CAAC,CAAC;YACF,OAAO,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,0EAA0E;IAClE,kBAAkB,CAAC,SAAmB;QAC5C,MAAM,OAAO,GAAiB,EAAE,CAAC;QACjC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAE/B,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;YAC7B,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YAChD,6FAA6F;YAC7F,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,QAAQ,cAAc,EAAE,CAAC,CAAC;YAC1E,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC9B,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;oBACjD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAClB,OAAO,CAAC,IAAI,CAAC,IAAkB,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;QACH,CAAC;QAED,mDAAmD;QACnD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1F,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7B,CAAC;IAEO,kBAAkB,CACxB,KAAa,EACb,aAA+B,EAC/B,YAAyB,EACzB,WAAoB;QAEpB,MAAM,KAAK,GAAa,CAAC,mBAAmB,KAAK,KAAK,CAAC,CAAC;QAExD,0EAA0E;QAC1E,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,KAAK,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC1C,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;gBAC9B,MAAM,IAAI,GAAG,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,SAAS,CAAC;gBACvF,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxE,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,GAAG,UAAU,aAAa,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC;YAC7F,CAAC;QACH,CAAC;QAED,0EAA0E;QAC1E,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,KAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YAEtC,MAAM,MAAM,GAAG,IAAI,GAAG,EAA0B,CAAC;YACjD,MAAM,SAAS,GAAe,EAAE,CAAC;YAEjC,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;gBAChC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACzB,SAAS,CAAC,IAAI,CAAC,IAAgB,CAAC,CAAC;gBACnC,CAAC;qBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACpC,MAAM,EAAE,GAAG,IAAoB,CAAC;oBAChC,MAAM,GAAG,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;oBAC5C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC;wBAAE,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC1C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC;YAED,KAAK,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,MAAM,EAAE,CAAC;gBACrC,KAAK,CAAC,IAAI,CAAC,SAAS,QAAQ,IAAI,CAAC,CAAC;gBAClC,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;oBACjC,MAAM,aAAa,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,WAAW,IAAI,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC;oBAClF,MAAM,GAAG,GAAG,GAAG,aAAa,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,GAAG,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;oBACxG,KAAK,CAAC,IAAI,CAAC,SAAS,GAAG,SAAS,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC7C,CAAC;gBACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACjB,CAAC;YAED,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;gBAC7B,KAAK,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,gBAAgB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC3F,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC;QAED,0EAA0E;QAC1E,MAAM,SAAS,GAAG,aAAa;aAC5B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAW,CAAC;aACxF,MAAM,CAAC,OAAO,CAAC,CAAC;QAEnB,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;YACnD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvB,KAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;gBACtC,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;oBACxB,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,kBAAkB,CAAC,OAAO,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;oBACtH,MAAM,KAAK,GAAG,CAAC,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACpD,MAAM,IAAI,GAAG,CAAC,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC/E,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,MAAM,IAAI,QAAQ,KAAK,GAAG,IAAI,EAAE,CAAC,CAAC;gBACtF,CAAC;gBACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACjB,CAAC;QACH,CAAC;QAED,0EAA0E;QAC1E,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjD,KAAK,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;YACvD,KAAK,CAAC,IAAI,CAAC,eAAe,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACpE,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF;AArLD,oCAqLC"}

package/dist/utils/modelManager.d.ts

@@ -0,0 +1,37 @@
+interface PipelineTensor {
+    data: Float32Array;
+    dims: number[];
+}
+type EmbedPipeline = (texts: string[], options?: Record<string, unknown>) => Promise<PipelineTensor>;
+export interface DownloadProgress {
+    file: string;
+    progress: number;
+    loaded: number;
+    total: number;
+    status: 'initiate' | 'download' | 'progress' | 'done' | 'ready' | 'loading';
+}
+export type ProgressCallback = (info: DownloadProgress) => void;
+/** True if the pipeline has been loaded into memory this process. */
+export declare function isPipelineReady(): boolean;
+/**
+ * Load (and optionally download) the embedding model.
+ * Returns the pipeline or null if loading fails.
+ * @param onProgress  Called for each download/load progress event.
+ */
+export declare function ensureModel(onProgress?: ProgressCallback): Promise<EmbedPipeline | null>;
+/** Get the cached pipeline (null if not yet loaded). */
+export declare function getCachedPipeline(): EmbedPipeline | null;
+/** Reset the cache (for tests). */
+export declare function resetPipeline(): void;
+/**
+ * Extract individual embedding vectors from a batched pipeline tensor.
+ * The tensor's `data` is a flat Float32Array of shape [batch, dim].
+ */
+export declare function extractEmbeddings(tensor: PipelineTensor): number[][];
+/**
+ * Embed texts using a worker thread when available (production), falling back
+ * to the main-thread pipeline in dev/test environments.
+ */
+export declare function embedTexts(texts: string[]): Promise<number[][]>;
+export {};
+//# sourceMappingURL=modelManager.d.ts.map

package/dist/utils/modelManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelManager.d.ts","sourceRoot":"","sources":["../../src/utils/modelManager.ts"],"names":[],"mappings":"AAKA,UAAU,cAAc;IACtB,IAAI,EAAE,YAAY,CAAC;IACnB,IAAI,EAAE,MAAM,EAAE,CAAC;CAChB;AAED,KAAK,aAAa,GAAG,CACnB,KAAK,EAAE,MAAM,EAAE,EACf,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,cAAc,CAAC,CAAC;AAE7B,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,UAAU,GAAG,UAAU,GAAG,UAAU,GAAG,MAAM,GAAG,OAAO,GAAG,SAAS,CAAC;CAC7E;AAED,MAAM,MAAM,gBAAgB,GAAG,CAAC,IAAI,EAAE,gBAAgB,KAAK,IAAI,CAAC;AAIhE,qEAAqE;AACrE,wBAAgB,eAAe,IAAI,OAAO,CAEzC;AAED;;;;GAIG;AACH,wBAAsB,WAAW,CAAC,UAAU,CAAC,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC,CAkC9F;AAED,wDAAwD;AACxD,wBAAgB,iBAAiB,IAAI,aAAa,GAAG,IAAI,CAExD;AAED,mCAAmC;AACnC,wBAAgB,aAAa,IAAI,IAAI,CAEpC;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,cAAc,GAAG,MAAM,EAAE,EAAE,CAOpE;AAED;;;GAGG;AACH,wBAAsB,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAUrE"}

package/dist/utils/modelManager.js

@@ -0,0 +1,90 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isPipelineReady = isPipelineReady;
+exports.ensureModel = ensureModel;
+exports.getCachedPipeline = getCachedPipeline;
+exports.resetPipeline = resetPipeline;
+exports.extractEmbeddings = extractEmbeddings;
+exports.embedTexts = embedTexts;
+const logger_js_1 = require("./logger.js");
+const workerPool_js_1 = require("./workerPool.js");
+const MODEL_ID = 'Xenova/all-MiniLM-L6-v2';
+let cachedPipeline = null;
+/** True if the pipeline has been loaded into memory this process. */
+function isPipelineReady() {
+    return cachedPipeline !== null;
+}
+/**
+ * Load (and optionally download) the embedding model.
+ * Returns the pipeline or null if loading fails.
+ * @param onProgress  Called for each download/load progress event.
+ */
+async function ensureModel(onProgress) {
+    if (cachedPipeline)
+        return cachedPipeline;
+    try {
+        const { pipeline, env } = await import('@xenova/transformers');
+        // Allow overriding cache dir via env var
+        if (process.env.FIRECODE_MODEL_CACHE) {
+            env.cacheDir = process.env.FIRECODE_MODEL_CACHE;
+        }
+        const progressCb = onProgress
+            ? (info) => {
+                onProgress({
+                    file: String(info.file ?? info.name ?? ''),
+                    progress: typeof info.progress === 'number' ? info.progress : 0,
+                    loaded: typeof info.loaded === 'number' ? info.loaded : 0,
+                    total: typeof info.total === 'number' ? info.total : 0,
+                    status: info.status ?? 'progress',
+                });
+            }
+            : undefined;
+        const pipe = await pipeline('feature-extraction', MODEL_ID, {
+            progress_callback: progressCb,
+        });
+        cachedPipeline = pipe;
+        logger_js_1.logger.info({ model: MODEL_ID }, 'Embedding model loaded');
+        return cachedPipeline;
+    }
+    catch (err) {
+        logger_js_1.logger.warn({ err: String(err) }, 'Could not load embedding model — falling back to text similarity');
+        return null;
+    }
+}
+/** Get the cached pipeline (null if not yet loaded). */
+function getCachedPipeline() {
+    return cachedPipeline;
+}
+/** Reset the cache (for tests). */
+function resetPipeline() {
+    cachedPipeline = null;
+}
+/**
+ * Extract individual embedding vectors from a batched pipeline tensor.
+ * The tensor's `data` is a flat Float32Array of shape [batch, dim].
+ */
+function extractEmbeddings(tensor) {
+    const [batchSize, dim] = tensor.dims;
+    const result = [];
+    for (let i = 0; i < batchSize; i++) {
+        result.push(Array.from(tensor.data.slice(i * dim, (i + 1) * dim)));
+    }
+    return result;
+}
+/**
+ * Embed texts using a worker thread when available (production), falling back
+ * to the main-thread pipeline in dev/test environments.
+ */
+async function embedTexts(texts) {
+    // Try worker thread first — returns null in dev/test mode or after worker crash
+    const workerResult = await (0, workerPool_js_1.workerEmbed)(texts).catch(() => null);
+    if (workerResult !== null)
+        return workerResult;
+    // Main-thread fallback
+    const pipe = await ensureModel();
+    if (!pipe)
+        return texts.map(() => []);
+    const tensor = await pipe(texts, { pooling: 'mean', normalize: true });
+    return extractEmbeddings(tensor);
+}
+//# sourceMappingURL=modelManager.js.map