@monoes/cli 1.1.0 → 1.2.1

package/dist/src/mcp-tools/graphify-tools.js

@@ -1,222 +1,450 @@
 /**
- * Graphify MCP Tools (compiled)
- * Bridges graphify's knowledge graph into monobrain's MCP tool surface.
+ * Graphify MCP Tools
+ *
+ * Bridges @monobrain/graph's knowledge graph into monobrain's MCP tool surface.
+ * Agents can query the codebase knowledge graph without reading files —
+ * god_nodes(), query_graph(), shortest_path() give structural understanding
+ * in milliseconds vs. reading dozens of source files.
+ *
+ * Graph is built automatically on `monobrain init` and stored at
+ * .monobrain/graph/graph.json (legacy: graphify-out/graph.json).
+ * Rebuild manually: call graphify_build via MCP.
  */
-import { spawnSync } from 'child_process';
 import { existsSync, readFileSync } from 'fs';
 import { join, resolve } from 'path';
 import { getProjectCwd } from './types.js';
-
+// ── Path helpers ──────────────────────────────────────────────────────────────
+/** Resolve graph path: prefer native monobrain path, fall back to legacy graphify path. */
 function getGraphPath(cwd) {
-    return resolve(join(cwd, 'graphify-out', 'graph.json'));
+    const nativePath = resolve(join(cwd, '.monobrain', 'graph', 'graph.json'));
+    const legacyPath = resolve(join(cwd, 'graphify-out', 'graph.json'));
+    if (existsSync(nativePath))
+        return nativePath;
+    if (existsSync(legacyPath))
+        return legacyPath;
+    return nativePath; // return expected path even if not yet built
 }
-
-function isGraphifyInstalled() {
-    try {
-        const result = spawnSync('python', ['-c', 'import graphify; print("ok")'], {
-            timeout: 5000, encoding: 'utf-8',
-        });
-        return result.status === 0 && result.stdout.includes('ok');
-    }
-    catch { return false; }
-}
-
 function graphExists(cwd) {
     return existsSync(getGraphPath(cwd));
 }
-
-function runGraphifyPython(snippet, cwd) {
+/**
+ * Load the knowledge graph.
+ * Tries @monobrain/graph's loadGraph first; falls back to parsing raw JSON.
+ */
+async function loadKnowledgeGraph(cwd) {
     const graphPath = getGraphPath(cwd);
-    const script = `
-import json, sys
-sys.path.insert(0, '${cwd.replace(/'/g, "\\'")}')
-graph_path = '${graphPath.replace(/'/g, "\\'")}'
-${snippet}
-`.trim();
-    const result = spawnSync('python', ['-c', script], {
-        timeout: 15000,
-        encoding: 'utf-8',
-        cwd,
-    });
-    if (result.status !== 0) {
-        throw new Error(result.stderr?.trim() || 'graphify python call failed');
+    let rawNodes = [];
+    let rawEdges = [];
+    try {
+        // Prefer @monobrain/graph's loader which handles format normalization.
+        const { loadGraph } = await import('@monoes/graph');
+        const loaded = loadGraph(graphPath);
+        rawNodes = loaded.nodes;
+        rawEdges = loaded.edges;
+    }
+    catch {
+        // Fallback: parse JSON directly
+        const data = JSON.parse(readFileSync(graphPath, 'utf-8'));
+        rawNodes = data.nodes || [];
+        rawEdges = data.links || data.edges || [];
+    }
+    // Build in-memory graph structures
+    const nodes = new Map();
+    for (const n of rawNodes) {
+        nodes.set(n.id, n);
+    }
+    const adj = new Map();
+    const radj = new Map();
+    const degree = new Map();
+    for (const n of rawNodes) {
+        adj.set(n.id, []);
+        radj.set(n.id, []);
+        degree.set(n.id, 0);
     }
-    const output = result.stdout?.trim();
-    if (!output) throw new Error('No output from graphify');
-    return JSON.parse(output);
+    for (const e of rawEdges) {
+        const src = e.source;
+        const tgt = e.target;
+        adj.get(src)?.push(tgt);
+        radj.get(tgt)?.push(src);
+        degree.set(src, (degree.get(src) ?? 0) + 1);
+        degree.set(tgt, (degree.get(tgt) ?? 0) + 1);
+    }
+    return { nodes, adj, radj, edges: rawEdges, degree, graphPath };
+}
+// ── Shared output helpers ─────────────────────────────────────────────────────
+function nodeOut(g, id) {
+    const d = g.nodes.get(id) ?? { id };
+    return {
+        id,
+        label: d.label ?? id,
+        file: d.source_file ?? '',
+        location: d.source_location ?? '',
+        community: d.community ?? null,
+        degree: g.degree.get(id) ?? 0,
+        file_type: d.file_type ?? '',
+    };
 }
-
-const LOAD_GRAPH_SNIPPET = `
-import json
-from pathlib import Path
-import networkx as nx
-from networkx.readwrite import json_graph
-data = json.loads(Path(graph_path).read_text())
-try:
-    G = json_graph.node_link_graph(data, edges='links')
-except TypeError:
-    G = json_graph.node_link_graph(data)
-`;
-
+/** Score a node against search terms. */
+function scoreNode(g, id, terms) {
+    const d = g.nodes.get(id);
+    if (!d)
+        return 0;
+    const label = (d.label ?? id).toLowerCase();
+    const file = (d.source_file ?? '').toLowerCase();
+    return terms.reduce((s, t) => {
+        if (label.includes(t))
+            s += 1;
+        if (file.includes(t))
+            s += 0.5;
+        return s;
+    }, 0);
+}
+// ── Tool Definitions ──────────────────────────────────────────────────────────
+/**
+ * Build or rebuild the knowledge graph for a directory.
+ */
 export const graphifyBuildTool = {
     name: 'graphify_build',
-    description: 'Build (or rebuild) the knowledge graph for a project directory. Extracts AST structure from code files, semantic relationships from docs, and clusters them into communities. Run this first before using other graphify tools.',
+    description: 'Build (or rebuild) the knowledge graph for a project directory. ' +
+        'Extracts AST structure from code files, semantic relationships from docs, ' +
+        'and clusters them into communities. Run this first before using other graphify tools. ' +
+        'Code-only changes are fast (tree-sitter, no LLM). Doc changes require an LLM call.',
     category: 'graphify',
     tags: ['knowledge-graph', 'codebase', 'architecture', 'analysis'],
     inputSchema: {
         type: 'object',
         properties: {
-            path: { type: 'string', description: 'Path to analyse (defaults to current project root)' },
-            codeOnly: { type: 'boolean', description: 'Only re-extract changed code files — fast rebuild', default: false },
+            path: {
+                type: 'string',
+                description: 'Path to analyse (defaults to current project root)',
+            },
+            codeOnly: {
+                type: 'boolean',
+                description: 'Only re-extract changed code files — no LLM, fast rebuild',
+                default: false,
+            },
         },
     },
     handler: async (params) => {
         const cwd = getProjectCwd();
         const targetPath = params.path || cwd;
-        if (!isGraphifyInstalled()) {
-            return { error: true, message: 'graphify not installed. Run: pip install graphifyy[mcp]', hint: 'After installing, call graphify_build again.' };
-        }
         try {
-            const result = spawnSync('python', ['-m', 'graphify', targetPath, ...(params.codeOnly ? ['--update'] : [])], {
-                timeout: 120000, encoding: 'utf-8', cwd,
+            const { buildGraph } = await import('@monoes/graph');
+            const outputDir = join(targetPath, '.monobrain', 'graph');
+            const result = await buildGraph(targetPath, {
+                codeOnly: Boolean(params.codeOnly),
+                outputDir,
             });
-            const graphPath = getGraphPath(cwd);
-            const built = existsSync(graphPath);
             return {
-                success: result.status === 0,
-                graphPath: built ? graphPath : null,
-                output: result.stdout?.trim().slice(-500) || '',
-                error: result.status !== 0 ? result.stderr?.trim().slice(-500) : undefined,
-                message: built ? `Knowledge graph built at ${graphPath}` : 'Build may have failed — graph.json not found',
+                success: true,
+                graphPath: result.graphPath,
+                filesProcessed: result.filesProcessed,
+                nodes: result.analysis.stats.nodes,
+                edges: result.analysis.stats.edges,
+                message: `Knowledge graph built at ${result.graphPath}`,
+            };
+        }
+        catch (err) {
+            return {
+                error: true,
+                message: String(err),
+                hint: '@monobrain/graph package not available — ensure it is installed and built.',
             };
         }
-        catch (err) { return { error: true, message: String(err) }; }
     },
 };
-
+/**
+ * Query the knowledge graph with natural language.
+ */
 export const graphifyQueryTool = {
     name: 'graphify_query',
-    description: 'Search the knowledge graph with a natural language question or keywords. Returns relevant nodes and edges as structured context — use this instead of reading many files when you want to understand how components relate.',
+    description: 'Search the knowledge graph with a natural language question or keywords. ' +
+        'Returns relevant nodes and edges as structured context — use this instead of reading ' +
+        'many files when you want to understand how components relate. ' +
+        'BFS mode gives broad context; DFS traces a specific call path.',
     category: 'graphify',
     tags: ['knowledge-graph', 'search', 'architecture', 'codebase'],
     inputSchema: {
         type: 'object',
         properties: {
-            question: { type: 'string', description: 'Natural language question or keyword' },
-            mode: { type: 'string', enum: ['bfs', 'dfs'], default: 'bfs' },
-            depth: { type: 'integer', default: 3 },
-            tokenBudget: { type: 'integer', default: 2000 },
+            question: {
+                type: 'string',
+                description: 'Natural language question or keyword (e.g. "authentication flow", "how does caching work")',
+            },
+            mode: {
+                type: 'string',
+                enum: ['bfs', 'dfs'],
+                default: 'bfs',
+                description: 'bfs = broad context, dfs = trace specific path',
+            },
+            depth: {
+                type: 'integer',
+                default: 3,
+                description: 'Traversal depth (1–6)',
+            },
+            tokenBudget: {
+                type: 'integer',
+                default: 2000,
+                description: 'Approximate max output tokens',
+            },
         },
         required: ['question'],
     },
     handler: async (params) => {
         const cwd = getProjectCwd();
-        if (!graphExists(cwd)) return { error: true, message: 'No graph found. Run graphify_build first.', hint: `Expected: ${getGraphPath(cwd)}` };
+        if (!graphExists(cwd)) {
+            return {
+                error: true,
+                message: 'No graph found. Run graphify_build first.',
+                hint: `Expected: ${getGraphPath(cwd)}`,
+            };
+        }
+        const question = params.question;
+        const mode = params.mode || 'bfs';
+        const depth = params.depth || 3;
         try {
-            return runGraphifyPython(`
-${LOAD_GRAPH_SNIPPET}
-question = ${JSON.stringify(params.question)}
-terms = [t.lower() for t in question.split() if len(t) > 2]
-mode = ${JSON.stringify(params.mode || 'bfs')}
-depth = ${params.depth || 3}
-scored = []
-for nid, d in G.nodes(data=True):
-    label = d.get('label', '').lower()
-    source = d.get('source_file', '').lower()
-    score = sum(1 for t in terms if t in label) + sum(0.5 for t in terms if t in source)
-    if score > 0:
-        scored.append((score, nid))
-scored.sort(reverse=True)
-start_nodes = [nid for _, nid in scored[:5]]
-if not start_nodes:
-    start_nodes = sorted(G.nodes(), key=lambda n: G.degree(n), reverse=True)[:3]
-visited = set(start_nodes)
-frontier = set(start_nodes)
-edges_seen = []
-if mode == 'bfs':
-    for _ in range(depth):
-        next_frontier = set()
-        for n in frontier:
-            for nbr in G.neighbors(n):
-                if nbr not in visited:
-                    next_frontier.add(nbr)
-                    edges_seen.append((n, nbr))
-        visited.update(next_frontier)
-        frontier = next_frontier
-else:
-    stack = [(n, 0) for n in reversed(start_nodes)]
-    while stack:
-        node, d = stack.pop()
-        if node in visited or d > depth: continue
-        visited.add(node)
-        for nbr in G.neighbors(node):
-            if nbr not in visited:
-                stack.append((nbr, d + 1))
-                edges_seen.append((node, nbr))
-nodes_out = [{'id': nid, 'label': G.nodes[nid].get('label', nid), 'file': G.nodes[nid].get('source_file', ''), 'location': G.nodes[nid].get('source_location', ''), 'community': G.nodes[nid].get('community'), 'degree': G.degree(nid), 'file_type': G.nodes[nid].get('file_type', '')} for nid in sorted(visited, key=lambda n: G.degree(n), reverse=True)]
-edges_out = [{'from': G.nodes[u].get('label', u), 'to': G.nodes[v].get('label', v), 'relation': G.edges.get((u, v), {}).get('relation', ''), 'confidence': G.edges.get((u, v), {}).get('confidence', '')} for u, v in edges_seen if u in visited and v in visited]
-print(json.dumps({'question': question, 'mode': mode, 'depth': depth, 'nodes': nodes_out[:60], 'edges': edges_out[:80], 'total_nodes': len(visited), 'total_edges': len(edges_seen)}))
-`, cwd);
+            const g = await loadKnowledgeGraph(cwd);
+            const terms = question.toLowerCase().split(/\s+/).filter(t => t.length > 2);
+            // Score nodes; fall back to highest-degree nodes if no match
+            let startNodes = [];
+            if (terms.length > 0) {
+                const scored = [];
+                for (const id of g.nodes.keys()) {
+                    const s = scoreNode(g, id, terms);
+                    if (s > 0)
+                        scored.push([s, id]);
+                }
+                scored.sort((a, b) => b[0] - a[0]);
+                startNodes = scored.slice(0, 5).map(([, id]) => id);
+            }
+            if (startNodes.length === 0) {
+                startNodes = [...g.nodes.keys()]
+                    .sort((a, b) => (g.degree.get(b) ?? 0) - (g.degree.get(a) ?? 0))
+                    .slice(0, 3);
+            }
+            const visited = new Set(startNodes);
+            const edgesSeen = [];
+            if (mode === 'bfs') {
+                let frontier = new Set(startNodes);
+                for (let d = 0; d < depth; d++) {
+                    const next = new Set();
+                    for (const n of frontier) {
+                        for (const nbr of g.adj.get(n) ?? []) {
+                            if (!visited.has(nbr)) {
+                                next.add(nbr);
+                                edgesSeen.push([n, nbr]);
+                            }
+                        }
+                    }
+                    for (const n of next)
+                        visited.add(n);
+                    frontier = next;
+                }
+            }
+            else {
+                // DFS
+                const stack = startNodes.map(n => [n, 0]);
+                while (stack.length > 0) {
+                    const [node, d] = stack.pop();
+                    if (visited.has(node) && d > 0)
+                        continue;
+                    if (d > depth)
+                        continue;
+                    visited.add(node);
+                    for (const nbr of g.adj.get(node) ?? []) {
+                        if (!visited.has(nbr)) {
+                            stack.push([nbr, d + 1]);
+                            edgesSeen.push([node, nbr]);
+                        }
+                    }
+                }
+            }
+            const nodesOut = [...visited]
+                .sort((a, b) => (g.degree.get(b) ?? 0) - (g.degree.get(a) ?? 0))
+                .slice(0, 60)
+                .map(id => nodeOut(g, id));
+            // Build edge lookup for attribute access
+            const edgeLookup = new Map();
+            for (const e of g.edges) {
+                edgeLookup.set(`${e.source}__${e.target}`, e);
+            }
+            const edgesOut = edgesSeen
+                .filter(([u, v]) => visited.has(u) && visited.has(v))
+                .slice(0, 80)
+                .map(([u, v]) => {
+                const e = edgeLookup.get(`${u}__${v}`) ?? {};
+                return {
+                    from: (g.nodes.get(u)?.label ?? u),
+                    to: (g.nodes.get(v)?.label ?? v),
+                    relation: e.relation ?? '',
+                    confidence: e.confidence ?? '',
+                };
+            });
+            return {
+                question,
+                mode,
+                depth,
+                nodes: nodesOut,
+                edges: edgesOut,
+                total_nodes: visited.size,
+                total_edges: edgesSeen.length,
+            };
+        }
+        catch (err) {
+            return { error: true, message: String(err) };
         }
-        catch (err) { return { error: true, message: String(err) }; }
     },
 };
-
+/**
+ * Get the most connected (god) nodes — the core abstractions.
+ */
 export const graphifyGodNodesTool = {
     name: 'graphify_god_nodes',
-    description: 'Return the most connected nodes in the knowledge graph — the core abstractions and central concepts of the codebase. Use this at the start of any architectural analysis.',
+    description: 'Return the most connected nodes in the knowledge graph — the core abstractions ' +
+        'and central concepts of the codebase. Use this at the start of any architectural analysis ' +
+        'to understand what the most important components are before diving into details.',
     category: 'graphify',
     tags: ['knowledge-graph', 'architecture', 'abstractions', 'codebase'],
-    inputSchema: { type: 'object', properties: { topN: { type: 'integer', default: 15 } } },
+    inputSchema: {
+        type: 'object',
+        properties: {
+            topN: {
+                type: 'integer',
+                default: 15,
+                description: 'Number of god nodes to return',
+            },
+        },
+    },
     handler: async (params) => {
         const cwd = getProjectCwd();
-        if (!graphExists(cwd)) return { error: true, message: 'No graph found. Run graphify_build first.' };
+        if (!graphExists(cwd)) {
+            return {
+                error: true,
+                message: 'No graph found. Run graphify_build first.',
+            };
+        }
+        const topN = params.topN || 15;
         try {
-            return runGraphifyPython(`
-${LOAD_GRAPH_SNIPPET}
-top_n = ${params.topN || 15}
-degree_map = dict(G.degree())
-sorted_nodes = sorted(degree_map, key=degree_map.get, reverse=True)[:top_n]
-nodes_out = [{'label': G.nodes[nid].get('label', nid), 'degree': degree_map[nid], 'file': G.nodes[nid].get('source_file', ''), 'location': G.nodes[nid].get('source_location', ''), 'community': G.nodes[nid].get('community'), 'file_type': G.nodes[nid].get('file_type', ''), 'neighbors': [G.nodes[n].get('label', n) for n in G.neighbors(nid)][:8]} for nid in sorted_nodes]
-print(json.dumps({'god_nodes': nodes_out, 'total_nodes': G.number_of_nodes()}))
-`, cwd);
+            const g = await loadKnowledgeGraph(cwd);
+            const sortedIds = [...g.nodes.keys()]
+                .sort((a, b) => (g.degree.get(b) ?? 0) - (g.degree.get(a) ?? 0))
+                .slice(0, topN);
+            const godNodes = sortedIds.map(id => {
+                const d = g.nodes.get(id) ?? { id };
+                const neighbors = (g.adj.get(id) ?? [])
+                    .slice(0, 8)
+                    .map(nid => g.nodes.get(nid)?.label ?? nid);
+                return {
+                    label: d.label ?? id,
+                    degree: g.degree.get(id) ?? 0,
+                    file: d.source_file ?? '',
+                    location: d.source_location ?? '',
+                    community: d.community ?? null,
+                    file_type: d.file_type ?? '',
+                    neighbors,
+                };
+            });
+            return { god_nodes: godNodes, total_nodes: g.nodes.size };
+        }
+        catch (err) {
+            return { error: true, message: String(err) };
         }
-        catch (err) { return { error: true, message: String(err) }; }
     },
 };
-
+/**
+ * Get full details for a specific node.
+ */
 export const graphifyGetNodeTool = {
     name: 'graphify_get_node',
-    description: 'Get all details for a specific concept/node in the knowledge graph: source location, community, all relationships, and confidence levels.',
+    description: 'Get all details for a specific concept/node in the knowledge graph: ' +
+        'its source location, community, all relationships, and confidence levels. ' +
+        'Use this when you need to deeply understand one specific component.',
     category: 'graphify',
     tags: ['knowledge-graph', 'node', 'details'],
-    inputSchema: { type: 'object', properties: { label: { type: 'string', description: 'Node label or ID (case-insensitive)' } }, required: ['label'] },
+    inputSchema: {
+        type: 'object',
+        properties: {
+            label: {
+                type: 'string',
+                description: 'Node label or ID to look up (case-insensitive)',
+            },
+        },
+        required: ['label'],
+    },
     handler: async (params) => {
         const cwd = getProjectCwd();
-        if (!graphExists(cwd)) return { error: true, message: 'No graph found. Run graphify_build first.' };
+        if (!graphExists(cwd)) {
+            return { error: true, message: 'No graph found. Run graphify_build first.' };
+        }
         try {
-            return runGraphifyPython(`
-${LOAD_GRAPH_SNIPPET}
-term = ${JSON.stringify(params.label.toLowerCase())}
-matches = [nid for nid, d in G.nodes(data=True) if term in d.get('label', '').lower() or term == nid.lower()]
-if not matches:
-    print(json.dumps({'error': 'Node not found', 'searched': term}))
-else:
-    nid = matches[0]
-    d = dict(G.nodes[nid])
-    edges = []
-    for u, v, ed in G.edges(nid, data=True):
-        edges.append({'direction': 'outgoing', 'to': G.nodes[v].get('label', v), 'relation': ed.get('relation', ''), 'confidence': ed.get('confidence', ''), 'confidence_score': ed.get('confidence_score')})
-    print(json.dumps({'id': nid, 'label': d.get('label', nid), 'file': d.get('source_file', ''), 'location': d.get('source_location', ''), 'community': d.get('community'), 'file_type': d.get('file_type', ''), 'degree': G.degree(nid), 'edges': edges[:40], 'all_matches': [G.nodes[m].get('label', m) for m in matches[:10]]}))
-`, cwd);
+            const g = await loadKnowledgeGraph(cwd);
+            const term = params.label.toLowerCase();
+            const matches = [...g.nodes.entries()]
+                .filter(([id, d]) => (d.label ?? id).toLowerCase().includes(term) || id.toLowerCase() === term)
+                .sort(([aId], [bId]) => (g.degree.get(bId) ?? 0) - (g.degree.get(aId) ?? 0))
+                .map(([id]) => id);
+            if (matches.length === 0) {
+                return { error: 'Node not found', searched: term };
+            }
+            const id = matches[0];
+            const d = g.nodes.get(id) ?? { id };
+            // Build edge lookup
+            const edgeLookup = new Map();
+            for (const e of g.edges) {
+                edgeLookup.set(`${e.source}__${e.target}`, e);
+            }
+            const outEdges = (g.adj.get(id) ?? []).slice(0, 40).map(tgt => {
+                const e = edgeLookup.get(`${id}__${tgt}`) ?? {};
+                return {
+                    direction: 'outgoing',
+                    to: g.nodes.get(tgt)?.label ?? tgt,
+                    relation: e.relation ?? '',
+                    confidence: e.confidence ?? '',
+                    confidence_score: e.confidence_score ?? null,
+                };
+            });
+            const inEdges = (g.radj.get(id) ?? []).slice(0, 40).map(src => {
+                const e = edgeLookup.get(`${src}__${id}`) ?? {};
+                return {
+                    direction: 'incoming',
+                    from: g.nodes.get(src)?.label ?? src,
+                    relation: e.relation ?? '',
+                    confidence: e.confidence ?? '',
+                };
+            });
+            // Strip well-known fields from attributes output to avoid duplication
+            const knownKeys = new Set(['label', 'source_file', 'source_location', 'community', 'file_type', 'id']);
+            const attributes = {};
+            for (const [k, v] of Object.entries(d)) {
+                if (!knownKeys.has(k))
+                    attributes[k] = v;
+            }
+            return {
+                id,
+                label: d.label ?? id,
+                file: d.source_file ?? '',
+                location: d.source_location ?? '',
+                community: d.community ?? null,
+                file_type: d.file_type ?? '',
+                degree: g.degree.get(id) ?? 0,
+                attributes,
+                edges: [...outEdges, ...inEdges],
+                all_matches: matches.slice(0, 10).map(m => g.nodes.get(m)?.label ?? m),
+            };
+        }
+        catch (err) {
+            return { error: true, message: String(err) };
         }
-        catch (err) { return { error: true, message: String(err) }; }
     },
 };
-
+/**
+ * Find shortest path between two concepts.
+ */
 export const graphifyShortestPathTool = {
     name: 'graphify_shortest_path',
-    description: 'Find the shortest relationship path between two concepts in the knowledge graph. Reveals coupling chains between any two components.',
+    description: 'Find the shortest relationship path between two concepts in the knowledge graph. ' +
+        'Use this to trace how component A depends on or relates to component B, ' +
+        'revealing hidden coupling chains (e.g. "how does the router connect to the database?").',
     category: 'graphify',
     tags: ['knowledge-graph', 'path', 'dependencies', 'coupling'],
     inputSchema: {
@@ -224,154 +452,379 @@ export const graphifyShortestPathTool = {
         properties: {
             source: { type: 'string', description: 'Source concept label' },
             target: { type: 'string', description: 'Target concept label' },
-            maxHops: { type: 'integer', default: 8 },
+            maxHops: { type: 'integer', default: 8, description: 'Maximum hops to search' },
         },
         required: ['source', 'target'],
     },
     handler: async (params) => {
         const cwd = getProjectCwd();
-        if (!graphExists(cwd)) return { error: true, message: 'No graph found. Run graphify_build first.' };
+        if (!graphExists(cwd)) {
+            return { error: true, message: 'No graph found. Run graphify_build first.' };
+        }
         try {
-            return runGraphifyPython(`
-${LOAD_GRAPH_SNIPPET}
-import networkx as nx
-def find_node(G, term):
-    t = term.lower()
-    matches = [nid for nid, d in G.nodes(data=True) if t in d.get('label', '').lower() or t == nid.lower()]
-    return sorted(matches, key=lambda n: G.degree(n), reverse=True)
-src_nodes = find_node(G, ${JSON.stringify(params.source)})
-tgt_nodes = find_node(G, ${JSON.stringify(params.target)})
-max_hops = ${params.maxHops || 8}
-if not src_nodes:
-    print(json.dumps({'error': f'Source not found: ${params.source}'}))
-elif not tgt_nodes:
-    print(json.dumps({'error': f'Target not found: ${params.target}'}))
-else:
-    UG = G.to_undirected() if G.is_directed() else G
-    best_path = None
-    best_len = max_hops + 1
-    for src in src_nodes[:3]:
-        for tgt in tgt_nodes[:3]:
-            try:
-                p = nx.shortest_path(UG, src, tgt)
-                if len(p) < best_len:
-                    best_len = len(p)
-                    best_path = p
-            except nx.NetworkXNoPath: pass
-    if best_path is None:
-        print(json.dumps({'found': False, 'message': 'No path found within hop limit'}))
-    else:
-        steps = []
-        for i, nid in enumerate(best_path):
-            d = G.nodes[nid]
-            step = {'label': d.get('label', nid), 'file': d.get('source_file', ''), 'location': d.get('source_location', '')}
-            if i < len(best_path) - 1:
-                edge = G.edges.get((nid, best_path[i+1]), G.edges.get((best_path[i+1], nid), {}))
-                step['next_relation'] = edge.get('relation', '')
-                step['confidence'] = edge.get('confidence', '')
-            steps.append(step)
-        print(json.dumps({'found': True, 'hops': len(best_path) - 1, 'path': steps}))
-`, cwd);
+            const g = await loadKnowledgeGraph(cwd);
+            const maxHops = params.maxHops || 8;
+            /** Find node ids matching a search term, sorted by degree descending. */
+            function findNodes(term) {
+                const t = term.toLowerCase();
+                return [...g.nodes.entries()]
+                    .filter(([id, d]) => (d.label ?? id).toLowerCase().includes(t) || id.toLowerCase() === t)
+                    .sort(([aId], [bId]) => (g.degree.get(bId) ?? 0) - (g.degree.get(aId) ?? 0))
+                    .map(([id]) => id);
+            }
+            const srcNodes = findNodes(params.source);
+            const tgtNodes = findNodes(params.target);
+            if (srcNodes.length === 0) {
+                return { error: true, message: `Source not found: ${params.source}` };
+            }
+            if (tgtNodes.length === 0) {
+                return { error: true, message: `Target not found: ${params.target}` };
+            }
+            // BFS on undirected graph (use both adj and radj as neighbours)
+            function bfsPath(start, end) {
+                const prev = new Map();
+                const queue = [start];
+                const visited = new Set([start]);
+                while (queue.length > 0) {
+                    const cur = queue.shift();
+                    if (cur === end) {
+                        // Reconstruct path
+                        const path = [];
+                        let node = end;
+                        while (node !== undefined) {
+                            path.unshift(node);
+                            node = prev.get(node);
+                        }
+                        return path.length - 1 <= maxHops ? path : null;
+                    }
+                    // Treat edges as undirected
+                    const nbrs = [...(g.adj.get(cur) ?? []), ...(g.radj.get(cur) ?? [])];
+                    for (const nbr of nbrs) {
+                        if (!visited.has(nbr)) {
+                            visited.add(nbr);
+                            prev.set(nbr, cur);
+                            if (queue.length < 100000)
+                                queue.push(nbr);
+                        }
+                    }
+                }
+                return null;
+            }
+            let bestPath = null;
+            for (const src of srcNodes.slice(0, 3)) {
+                for (const tgt of tgtNodes.slice(0, 3)) {
+                    const p = bfsPath(src, tgt);
+                    if (p && (!bestPath || p.length < bestPath.length)) {
+                        bestPath = p;
+                    }
+                }
+            }
+            if (!bestPath) {
+                return {
+                    found: false,
+                    message: `No path within ${maxHops} hops between "${params.source}" and "${params.target}"`,
+                };
+            }
+            // Build edge lookup
+            const edgeLookup = new Map();
+            for (const e of g.edges) {
+                edgeLookup.set(`${e.source}__${e.target}`, e);
+                edgeLookup.set(`${e.target}__${e.source}`, e); // bidirectional lookup
+            }
+            const steps = bestPath.map((id, i) => {
+                const d = g.nodes.get(id) ?? { id };
+                const step = {
+                    label: d.label ?? id,
+                    file: d.source_file ?? '',
+                    location: d.source_location ?? '',
+                };
+                if (i < bestPath.length - 1) {
+                    const nextId = bestPath[i + 1];
+                    const e = edgeLookup.get(`${id}__${nextId}`) ?? edgeLookup.get(`${nextId}__${id}`) ?? {};
+                    step.next_relation = e.relation ?? '';
+                    step.confidence = e.confidence ?? '';
+                }
+                return step;
+            });
+            return { found: true, hops: bestPath.length - 1, path: steps };
+        }
+        catch (err) {
+            return { error: true, message: String(err) };
         }
-        catch (err) { return { error: true, message: String(err) }; }
     },
 };
-
+/**
+ * Get all nodes in a community (cluster of related components).
+ */
 export const graphifyGetCommunityTool = {
     name: 'graphify_community',
-    description: 'Get all nodes in a specific community cluster. Communities are groups of tightly related components. Use graphify_stats first to see community count.',
+    description: 'Get all nodes in a specific community cluster. Communities are groups of ' +
+        'tightly related components detected by graph clustering. Use graph_stats first to ' +
+        'see community count, then explore communities to understand subsystem boundaries.',
     category: 'graphify',
     tags: ['knowledge-graph', 'community', 'clusters', 'subsystems'],
-    inputSchema: { type: 'object', properties: { communityId: { type: 'integer', description: 'Community ID (0 = largest)' } }, required: ['communityId'] },
+    inputSchema: {
+        type: 'object',
+        properties: {
+            communityId: {
+                type: 'integer',
+                description: 'Community ID (0 = largest community)',
+            },
+        },
+        required: ['communityId'],
+    },
     handler: async (params) => {
         const cwd = getProjectCwd();
-        if (!graphExists(cwd)) return { error: true, message: 'No graph found. Run graphify_build first.' };
+        if (!graphExists(cwd)) {
+            return { error: true, message: 'No graph found. Run graphify_build first.' };
+        }
         try {
-            return runGraphifyPython(`
-${LOAD_GRAPH_SNIPPET}
-cid = ${params.communityId}
-members = [{'label': d.get('label', nid), 'file': d.get('source_file', ''), 'location': d.get('source_location', ''), 'degree': G.degree(nid), 'file_type': d.get('file_type', '')} for nid, d in G.nodes(data=True) if d.get('community') == cid]
-members.sort(key=lambda x: x['degree'], reverse=True)
-external_edges = []
-for nid, d in G.nodes(data=True):
-    if d.get('community') == cid:
-        for nbr in G.neighbors(nid):
-            if G.nodes[nbr].get('community') != cid:
-                ed = G.edges.get((nid, nbr), {})
-                external_edges.append({'from': d.get('label', nid), 'to': G.nodes[nbr].get('label', nbr), 'to_community': G.nodes[nbr].get('community'), 'relation': ed.get('relation', '')})
-print(json.dumps({'community_id': cid, 'member_count': len(members), 'members': members[:50], 'external_connections': external_edges[:30]}))
-`, cwd);
+            const g = await loadKnowledgeGraph(cwd);
+            const cid = params.communityId;
+            const members = [...g.nodes.entries()]
+                .filter(([, d]) => d.community === cid)
+                .sort(([aId], [bId]) => (g.degree.get(bId) ?? 0) - (g.degree.get(aId) ?? 0))
+                .slice(0, 50)
+                .map(([id, d]) => ({
+                label: d.label ?? id,
+                file: d.source_file ?? '',
+                location: d.source_location ?? '',
+                degree: g.degree.get(id) ?? 0,
+                file_type: d.file_type ?? '',
+            }));
+            // Build edge lookup
+            const edgeLookup = new Map();
+            for (const e of g.edges) {
+                edgeLookup.set(`${e.source}__${e.target}`, e);
+            }
+            const externalEdges = [];
+            for (const [id, d] of g.nodes.entries()) {
+                if (d.community !== cid)
+                    continue;
+                for (const nbr of g.adj.get(id) ?? []) {
+                    const nbrD = g.nodes.get(nbr);
+                    if (nbrD?.community !== cid) {
+                        const e = edgeLookup.get(`${id}__${nbr}`) ?? {};
+                        externalEdges.push({
+                            from: d.label ?? id,
+                            to: nbrD?.label ?? nbr,
+                            to_community: nbrD?.community ?? null,
+                            relation: e.relation ?? '',
+                        });
+                    }
+                }
+            }
+            return {
+                community_id: cid,
+                member_count: members.length,
+                members,
+                external_connections: externalEdges.slice(0, 30),
+            };
+        }
+        catch (err) {
+            return { error: true, message: String(err) };
         }
-        catch (err) { return { error: true, message: String(err) }; }
     },
 };
-
+/**
+ * Get graph statistics: node/edge counts, communities, confidence breakdown.
+ */
 export const graphifyStatsTool = {
     name: 'graphify_stats',
-    description: 'Get summary statistics for the knowledge graph: node count, edge count, community count, confidence breakdown, and top god nodes. Use this first to understand graph size.',
+    description: 'Get summary statistics for the knowledge graph: node count, edge count, ' +
+        'community count, confidence breakdown (EXTRACTED/INFERRED/AMBIGUOUS), ' +
+        'and top god nodes. Use this first to understand graph size and structure.',
     category: 'graphify',
     tags: ['knowledge-graph', 'stats', 'overview'],
-    inputSchema: { type: 'object', properties: {} },
+    inputSchema: {
+        type: 'object',
+        properties: {},
+    },
     handler: async (_params) => {
         const cwd = getProjectCwd();
-        if (!graphExists(cwd)) return { error: true, message: 'No graph found. Run graphify_build first.', hint: `Expected: ${getGraphPath(cwd)}` };
-        if (!isGraphifyInstalled()) {
-            try {
-                const raw = JSON.parse(readFileSync(getGraphPath(cwd), 'utf-8'));
-                const nodes = raw.nodes || [];
-                const edges = raw.links || raw.edges || [];
-                return { nodes: nodes.length, edges: edges.length, note: 'graphify Python package not installed — basic stats only', install: 'pip install graphifyy[mcp]' };
-            }
-            catch (err) { return { error: true, message: String(err) }; }
+        if (!graphExists(cwd)) {
+            return {
+                error: true,
+                message: 'No graph found. Run graphify_build first.',
+                hint: `Expected: ${getGraphPath(cwd)}`,
+            };
         }
         try {
-            return runGraphifyPython(`
-${LOAD_GRAPH_SNIPPET}
-from collections import Counter
-communities = {}
-for nid, d in G.nodes(data=True):
-    c = d.get('community')
-    if c is not None:
-        communities.setdefault(int(c), []).append(nid)
-confidence_counts = Counter(d.get('confidence', 'UNKNOWN') for u, v, d in G.edges(data=True))
-relation_counts = Counter(d.get('relation', 'unknown') for u, v, d in G.edges(data=True))
-file_types = Counter(d.get('file_type', 'unknown') for nid, d in G.nodes(data=True))
-degree_map = dict(G.degree())
-top_nodes = sorted(degree_map, key=degree_map.get, reverse=True)[:5]
-print(json.dumps({'nodes': G.number_of_nodes(), 'edges': G.number_of_edges(), 'communities': len(communities), 'community_sizes': {str(k): len(v) for k, v in sorted(communities.items(), key=lambda x: len(x[1]), reverse=True)[:10]}, 'confidence': dict(confidence_counts), 'top_relations': dict(relation_counts.most_common(10)), 'file_types': dict(file_types), 'graph_path': graph_path, 'top_god_nodes': [G.nodes[n].get('label', n) for n in top_nodes], 'is_directed': G.is_directed()}))
-`, cwd);
+            const g = await loadKnowledgeGraph(cwd);
+            // Community sizes
+            const communities = new Map();
+            for (const d of g.nodes.values()) {
+                if (d.community != null) {
+                    communities.set(d.community, (communities.get(d.community) ?? 0) + 1);
+                }
+            }
+            const communitySizes = {};
+            [...communities.entries()]
+                .sort((a, b) => b[1] - a[1])
+                .slice(0, 10)
+                .forEach(([cid, count]) => { communitySizes[String(cid)] = count; });
+            // Confidence and relation counts
+            const confidenceCounts = {};
+            const relationCounts = {};
+            const fileTypeCounts = {};
+            for (const e of g.edges) {
+                const conf = e.confidence ?? 'UNKNOWN';
+                confidenceCounts[conf] = (confidenceCounts[conf] ?? 0) + 1;
+                const rel = e.relation ?? 'unknown';
+                relationCounts[rel] = (relationCounts[rel] ?? 0) + 1;
+            }
+            for (const d of g.nodes.values()) {
+                const ft = d.file_type ?? 'unknown';
+                fileTypeCounts[ft] = (fileTypeCounts[ft] ?? 0) + 1;
+            }
+            const topRelations = Object.entries(relationCounts)
+                .sort((a, b) => b[1] - a[1])
+                .slice(0, 10)
+                .reduce((acc, [k, v]) => { acc[k] = v; return acc; }, {});
+            const topGodNodes = [...g.nodes.keys()]
+                .sort((a, b) => (g.degree.get(b) ?? 0) - (g.degree.get(a) ?? 0))
+                .slice(0, 5)
+                .map(id => g.nodes.get(id)?.label ?? id);
+            return {
+                nodes: g.nodes.size,
+                edges: g.edges.length,
+                communities: communities.size,
+                community_sizes: communitySizes,
+                confidence: confidenceCounts,
+                top_relations: topRelations,
+                file_types: fileTypeCounts,
+                graph_path: g.graphPath,
+                top_god_nodes: topGodNodes,
+                is_directed: true,
+            };
+        }
+        catch (err) {
+            return { error: true, message: String(err) };
         }
-        catch (err) { return { error: true, message: String(err) }; }
     },
 };
-
+/**
+ * Find surprising cross-community connections (architectural insights).
+ */
 export const graphifySurprisesTool = {
     name: 'graphify_surprises',
-    description: 'Find surprising connections between components in different communities with strong relationships. These unexpected couplings often reveal hidden dependencies or important architectural patterns.',
+    description: 'Find surprising connections between components that are in different communities ' +
+        'but have strong relationships. These unexpected couplings often reveal hidden dependencies, ' +
+        'design smells, or important architectural patterns worth understanding.',
     category: 'graphify',
     tags: ['knowledge-graph', 'architecture', 'coupling', 'surprises'],
-    inputSchema: { type: 'object', properties: { topN: { type: 'integer', default: 10 } } },
+    inputSchema: {
+        type: 'object',
+        properties: {
+            topN: {
+                type: 'integer',
+                default: 10,
+                description: 'Number of surprising connections to return',
+            },
+        },
+    },
     handler: async (params) => {
         const cwd = getProjectCwd();
-        if (!graphExists(cwd)) return { error: true, message: 'No graph found. Run graphify_build first.' };
+        if (!graphExists(cwd)) {
+            return { error: true, message: 'No graph found. Run graphify_build first.' };
+        }
         try {
-            return runGraphifyPython(`
-${LOAD_GRAPH_SNIPPET}
-top_n = ${params.topN || 10}
-surprises = []
-for u, v, d in G.edges(data=True):
-    cu = G.nodes[u].get('community')
-    cv = G.nodes[v].get('community')
-    if cu is not None and cv is not None and cu != cv:
-        surprises.append({'score': G.degree(u) * G.degree(v), 'from': G.nodes[u].get('label', u), 'from_community': cu, 'from_file': G.nodes[u].get('source_file', ''), 'to': G.nodes[v].get('label', v), 'to_community': cv, 'to_file': G.nodes[v].get('source_file', ''), 'relation': d.get('relation', ''), 'confidence': d.get('confidence', '')})
-surprises.sort(key=lambda x: x['score'], reverse=True)
-print(json.dumps({'surprises': surprises[:top_n], 'total_cross_community_edges': len(surprises)}))
-`, cwd);
+            const g = await loadKnowledgeGraph(cwd);
+            const topN = params.topN || 10;
+            const surprises = [];
+            for (const e of g.edges) {
+                const uD = g.nodes.get(e.source);
+                const vD = g.nodes.get(e.target);
+                const cu = uD?.community ?? null;
+                const cv = vD?.community ?? null;
+                if (cu != null && cv != null && cu !== cv) {
+                    const score = (g.degree.get(e.source) ?? 0) * (g.degree.get(e.target) ?? 0);
+                    surprises.push({
+                        score,
+                        from: uD?.label ?? e.source,
+                        from_community: cu,
+                        from_file: uD?.source_file ?? '',
+                        to: vD?.label ?? e.target,
+                        to_community: cv,
+                        to_file: vD?.source_file ?? '',
+                        relation: e.relation ?? '',
+                        confidence: e.confidence ?? '',
+                    });
+                }
+            }
+            surprises.sort((a, b) => b.score - a.score);
+            return {
+                surprises: surprises.slice(0, topN),
+                total_cross_community_edges: surprises.length,
+            };
+        }
+        catch (err) {
+            return { error: true, message: String(err) };
+        }
+    },
+};
+/**
+ * Generate or open the interactive HTML visualization of the knowledge graph.
+ */
+export const graphifyVisualizeTool = {
+    name: 'graphify_visualize',
+    description: 'Generate (and optionally open) the interactive HTML knowledge graph explorer. ' +
+        'Produces graph.html alongside graph.json — a self-contained visualization with force-directed ' +
+        'layout, community coloring, god-node panel, sidebar details, minimap, and search. ' +
+        'No internet connection or server required — just open the HTML file in a browser.',
+    category: 'graphify',
+    tags: ['knowledge-graph', 'visualization', 'html', 'browser'],
+    inputSchema: {
+        type: 'object',
+        properties: {
+            open: {
+                type: 'boolean',
+                description: 'Open the HTML file in the default browser after generating (macOS/Linux)',
+                default: false,
+            },
+            path: {
+                type: 'string',
+                description: 'Project path (defaults to current project root)',
+            },
+        },
+    },
+    handler: async (params) => {
+        const cwd = getProjectCwd();
+        const targetPath = params.path || cwd;
+        if (!graphExists(targetPath)) {
+            return {
+                error: true,
+                message: 'No graph found. Run graphify_build first.',
+                hint: `Expected: ${getGraphPath(targetPath)}`,
+            };
+        }
+        try {
+            const graphPath = getGraphPath(targetPath);
+            const { readFileSync } = await import('fs');
+            const { join, dirname } = await import('path');
+            const outputDir = dirname(graphPath);
+            const { exportHTML } = await import('@monoes/graph');
+            const raw = JSON.parse(readFileSync(graphPath, 'utf-8'));
+            const htmlPath = exportHTML(raw, outputDir);
+            if (params.open) {
+                const { spawn } = await import('child_process');
+                const opener = process.platform === 'darwin' ? 'open'
+                    : process.platform === 'win32' ? 'start' : 'xdg-open';
+                spawn(opener, [htmlPath], { detached: true, stdio: 'ignore' }).unref();
+            }
+            return {
+                success: true,
+                htmlPath,
+                message: `Interactive visualization generated at ${htmlPath}`,
+                hint: params.open ? 'Opening in browser…' : `Open in browser: open "${htmlPath}"`,
+            };
+        }
+        catch (err) {
+            return { error: true, message: String(err) };
         }
-        catch (err) { return { error: true, message: String(err) }; }
     },
 };
-
+// ── Exports ───────────────────────────────────────────────────────────────────
 export const graphifyTools = [
     graphifyBuildTool,
     graphifyQueryTool,
@@ -381,6 +834,7 @@ export const graphifyTools = [
     graphifyGetCommunityTool,
     graphifyStatsTool,
     graphifySurprisesTool,
+    graphifyVisualizeTool,
 ];
-
 export default graphifyTools;
+//# sourceMappingURL=graphify-tools.js.map