@phoenixaihub/graphrag 0.1.0

package/dist/db.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"db.js","sourceRoot":"","sources":["../src/db.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,IAAI,CAAC;AA2BpB,MAAM,MAAM,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgCd,CAAC;AAEF,MAAM,OAAO,OAAO;IACV,EAAE,CAAoB;IAE9B,YAAY,MAAc;QACxB,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACjC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAChE,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC/B,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QACrC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACvB,CAAC;IAED,YAAY,CAAC,CAAS;QACpB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,iHAAiH,CAClH,CAAC;QACF,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC;QAChG,OAAO,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IACxC,CAAC;IAED,UAAU,CAAC,CAAO;QAChB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,qEAAqE,CACtE,CAAC;QACF,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC5D,OAAO,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IACxC,CAAC;IAED,WAAW,CAAC,CAAQ;QAClB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,qEAAqE,CACtE,CAAC;QACF,MAAM,aAAa,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC3E,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QAC9D,OAAO,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IACxC,CAAC;IAED,SAAS,CAAC,EAAU;QAClB,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAuB,CAAC;IAC9F,CAAC;IAED,kBAAkB,CAAC,IAAY;QAC7B,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CACpB,0CAA0C,CAC3C,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,CAAa,CAAC;IACjC,CAAC;IAED,cAAc,CAAC,KAAa,EAAE,KAAK,GAAG,CAAC;QACrC,yCAAyC;QACzC,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAClC,MAAM,UAAU,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,4CAA4C,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/F,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QACxD,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CACpB,gCAAgC,UAAU,8BAA8B,CACzE,CAAC,GAAG,CAAC,GAAG,MAAM,EAAE,KAAK,CAAa,CAAC;IACtC,CAAC;IAED,UAAU,CAAC,QAAgB,EAAE,YAA8C,MAAM;QAC/E,MAAM,OAAO,GAAgD,EAAE,CAAC;QAChE,IAAI,SAAS,KAAK,UAAU,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;YACrD,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,qGAAqG,CACtG,CAAC,GAAG,CAAC,QAAQ,CAAyC,CAAC;YACxD,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,SAAS,KAAK,UAAU,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;YACrD,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,qGAAqG,CACtG,CAAC,GAAG,CAAC,QAAQ,CAAyC,CAAC;YACxD,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;QACxE,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,SAAS,CAAC,IAAY;QACpB,OAAO,IAAI,CAAC,EAAE,CAAC,OAAO,CACpB;;;iEAG2D,CAC5D,CAAC,GAAG,CAAC,IAAI,CAAa,CAAC;IAC1B,CAAC;IAED,SAAS,CAAC,QAAgB;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC,GAAG,CAAC,QAAQ,CAA0B,CAAC;QACvH,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACpC,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAC7B,MAAM,YAAY,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClD,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,0CAA0C,YAAY,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC;QACvF,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,yCAAyC,YAAY,sBAAsB,YAAY,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,EAAE,GAAG,GAAG,CAAC,CAAC;QAChI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,0CAA0C,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC5E,CAAC;IAED,QAAQ;QACN,MAAM,QAAQ,GAAI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAC,GAAG,EAAoB,CAAC,CAAC,CAAC;QAClG,MAAM,KAAK,GAAI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,iCAAiC,CAAC,CAAC,GAAG,EAAoB,CAAC,CAAC,CAAC;QAC5F,MAAM,MAAM,GAAI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC,GAAG,EAAoB,CAAC,CAAC,CAAC;QAC9F,MAAM,KAAK,GAAI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,qDAAqD,CAAC,CAAC,GAAG,EAAoB,CAAC,CAAC,CAAC;QAChH,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;IAC5C,CAAC;IAED,KAAK;QACH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;CACF"}

package/dist/embeddings.d.ts

@@ -0,0 +1,4 @@
+export declare function getEmbedding(text: string): Promise<number[]>;
+export declare function getEmbeddings(texts: string[]): Promise<number[][]>;
+export declare function cosineSimilarity(a: number[], b: number[]): number;
+//# sourceMappingURL=embeddings.d.ts.map

package/dist/embeddings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.d.ts","sourceRoot":"","sources":["../src/embeddings.ts"],"names":[],"mappings":"AAWA,wBAAsB,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAOlE;AAED,wBAAsB,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAQxE;AAED,wBAAgB,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,CAQjE"}

package/dist/embeddings.js

@@ -0,0 +1,35 @@
+import OpenAI from 'openai';
+let client = null;
+function getClient() {
+    if (!client) {
+        client = new OpenAI();
+    }
+    return client;
+}
+export async function getEmbedding(text) {
+    const openai = getClient();
+    const resp = await openai.embeddings.create({
+        model: 'text-embedding-3-small',
+        input: text.substring(0, 8000),
+    });
+    return resp.data[0].embedding;
+}
+export async function getEmbeddings(texts) {
+    const openai = getClient();
+    const truncated = texts.map(t => t.substring(0, 8000));
+    const resp = await openai.embeddings.create({
+        model: 'text-embedding-3-small',
+        input: truncated,
+    });
+    return resp.data.map(d => d.embedding);
+}
+export function cosineSimilarity(a, b) {
+    let dot = 0, normA = 0, normB = 0;
+    for (let i = 0; i < a.length; i++) {
+        dot += a[i] * b[i];
+        normA += a[i] * a[i];
+        normB += b[i] * b[i];
+    }
+    return dot / (Math.sqrt(normA) * Math.sqrt(normB));
+}
+//# sourceMappingURL=embeddings.js.map

package/dist/embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.js","sourceRoot":"","sources":["../src/embeddings.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,IAAI,MAAM,GAAkB,IAAI,CAAC;AAEjC,SAAS,SAAS;IAChB,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;IACxB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,IAAY;IAC7C,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;QAC1C,KAAK,EAAE,wBAAwB;QAC/B,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC;KAC/B,CAAC,CAAC;IACH,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAChC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,KAAe;IACjD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;IACvD,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;QAC1C,KAAK,EAAE,wBAAwB;QAC/B,KAAK,EAAE,SAAS;KACjB,CAAC,CAAC;IACH,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,CAAW,EAAE,CAAW;IACvD,IAAI,GAAG,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IACD,OAAO,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;AACrD,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,6 @@
+export { indexRepo } from './commands/index-repo.js';
+export { queryCode } from './commands/query.js';
+export { serve } from './commands/serve.js';
+export { GraphDB } from './db.js';
+export { parseFile } from './parser.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChD,OAAO,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC"}

package/dist/index.js

@@ -0,0 +1,6 @@
+export { indexRepo } from './commands/index-repo.js';
+export { queryCode } from './commands/query.js';
+export { serve } from './commands/serve.js';
+export { GraphDB } from './db.js';
+export { parseFile } from './parser.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChD,OAAO,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC"}

package/dist/parser.d.ts

@@ -0,0 +1,12 @@
+import type { Entity, Edge } from './db.js';
+interface ParseResult {
+    entities: Omit<Entity, 'id'>[];
+    edges: Array<{
+        sourceName: string;
+        targetName: string;
+        relation: Edge['relation'];
+    }>;
+}
+export declare function parseFile(filePath: string, source?: string): ParseResult;
+export {};
+//# sourceMappingURL=parser.d.ts.map

package/dist/parser.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parser.d.ts","sourceRoot":"","sources":["../src/parser.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAI5C,UAAU,WAAW;IACnB,QAAQ,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC;IAC/B,KAAK,EAAE,KAAK,CAAC;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;KAAE,CAAC,CAAC;CACtF;AAyED,wBAAgB,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,WAAW,CA+ExE"}

package/dist/parser.js

@@ -0,0 +1,152 @@
+import fs from 'fs';
+import path from 'path';
+const FUNCTION_PATTERNS = {
+    typescript: [
+        /^(?:export\s+)?(?:async\s+)?function\s+(\w+)/gm,
+        /^(?:export\s+)?const\s+(\w+)\s*=\s*(?:async\s+)?\(/gm,
+        /^(?:export\s+)?const\s+(\w+)\s*=\s*(?:async\s+)?(?:\([^)]*\)|[^=])\s*=>/gm,
+    ],
+    python: [
+        /^def\s+(\w+)\s*\(/gm,
+        /^async\s+def\s+(\w+)\s*\(/gm,
+    ],
+};
+const CLASS_PATTERNS = {
+    typescript: [/^(?:export\s+)?(?:abstract\s+)?class\s+(\w+)/gm],
+    python: [/^class\s+(\w+)/gm],
+};
+const IMPORT_PATTERNS = {
+    typescript: [
+        /import\s+(?:{([^}]+)}|(\w+))\s+from\s+['"]([^'"]+)['"]/gm,
+        /import\s+\*\s+as\s+(\w+)\s+from\s+['"]([^'"]+)['"]/gm,
+    ],
+    python: [
+        /^from\s+(\S+)\s+import\s+(.+)$/gm,
+        /^import\s+(\S+)/gm,
+    ],
+};
+function detectLanguage(filePath) {
+    const ext = path.extname(filePath).toLowerCase();
+    if (['.ts', '.tsx', '.js', '.jsx', '.mjs', '.cjs'].includes(ext))
+        return 'typescript';
+    if (ext === '.py')
+        return 'python';
+    return 'unknown';
+}
+function getLineNumber(source, index) {
+    return source.substring(0, index).split('\n').length;
+}
+function extractBlock(lines, startLine, language) {
+    // Simple brace/indent block extraction
+    const start = startLine - 1;
+    if (language === 'python') {
+        let end = start + 1;
+        const baseIndent = lines[start]?.search(/\S/) ?? 0;
+        while (end < lines.length) {
+            const line = lines[end];
+            if (line.trim() === '') {
+                end++;
+                continue;
+            }
+            const indent = line.search(/\S/);
+            if (indent <= baseIndent)
+                break;
+            end++;
+        }
+        return { endLine: end, code: lines.slice(start, end).join('\n') };
+    }
+    // Brace-based
+    let braces = 0;
+    let found = false;
+    let end = start;
+    for (let i = start; i < lines.length; i++) {
+        for (const ch of lines[i]) {
+            if (ch === '{') {
+                braces++;
+                found = true;
+            }
+            if (ch === '}')
+                braces--;
+        }
+        end = i + 1;
+        if (found && braces <= 0)
+            break;
+    }
+    if (!found)
+        end = Math.min(start + 1, lines.length);
+    return { endLine: end, code: lines.slice(start, end).join('\n') };
+}
+export function parseFile(filePath, source) {
+    const content = source ?? fs.readFileSync(filePath, 'utf-8');
+    const language = detectLanguage(filePath);
+    if (language === 'unknown')
+        return { entities: [], edges: [] };
+    const lines = content.split('\n');
+    const entities = [];
+    const edgesList = [];
+    // File entity
+    entities.push({
+        filePath, name: path.basename(filePath), kind: 'file',
+        startLine: 1, endLine: lines.length, code: content.substring(0, 500),
+        language,
+    });
+    const langKey = language;
+    // Functions
+    for (const pattern of (FUNCTION_PATTERNS[langKey] ?? [])) {
+        const regex = new RegExp(pattern.source, pattern.flags);
+        let match;
+        while ((match = regex.exec(content)) !== null) {
+            const name = match[1];
+            const startLine = getLineNumber(content, match.index);
+            const block = extractBlock(lines, startLine, language);
+            entities.push({
+                filePath, name, kind: 'function',
+                startLine, endLine: block.endLine, code: block.code,
+                language,
+            });
+            edgesList.push({ sourceName: path.basename(filePath), targetName: name, relation: 'contains' });
+        }
+    }
+    // Classes
+    for (const pattern of (CLASS_PATTERNS[langKey] ?? [])) {
+        const regex = new RegExp(pattern.source, pattern.flags);
+        let match;
+        while ((match = regex.exec(content)) !== null) {
+            const name = match[1];
+            const startLine = getLineNumber(content, match.index);
+            const block = extractBlock(lines, startLine, language);
+            entities.push({
+                filePath, name, kind: 'class',
+                startLine, endLine: block.endLine, code: block.code,
+                language,
+            });
+            edgesList.push({ sourceName: path.basename(filePath), targetName: name, relation: 'contains' });
+        }
+    }
+    // Imports
+    for (const pattern of (IMPORT_PATTERNS[langKey] ?? [])) {
+        const regex = new RegExp(pattern.source, pattern.flags);
+        let match;
+        while ((match = regex.exec(content)) !== null) {
+            const startLine = getLineNumber(content, match.index);
+            const importSource = language === 'typescript' ? (match[3] ?? match[2] ?? match[1]) : match[1];
+            const importedNames = language === 'typescript'
+                ? (match[1] ?? match[2] ?? '').split(',').map(s => s.trim()).filter(Boolean)
+                : (match[2] ?? match[1] ?? '').split(',').map(s => s.trim()).filter(Boolean);
+            for (const name of importedNames) {
+                entities.push({
+                    filePath, name: name.replace(/\s+as\s+\w+/, ''), kind: 'import',
+                    startLine, endLine: startLine, code: match[0],
+                    language,
+                });
+                edgesList.push({
+                    sourceName: path.basename(filePath),
+                    targetName: name.replace(/\s+as\s+\w+/, ''),
+                    relation: 'imports',
+                });
+            }
+        }
+    }
+    return { entities, edges: edgesList };
+}
+//# sourceMappingURL=parser.js.map

package/dist/parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parser.js","sourceRoot":"","sources":["../src/parser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAUxB,MAAM,iBAAiB,GAA6B;IAClD,UAAU,EAAE;QACV,gDAAgD;QAChD,sDAAsD;QACtD,2EAA2E;KAC5E;IACD,MAAM,EAAE;QACN,qBAAqB;QACrB,6BAA6B;KAC9B;CACF,CAAC;AAEF,MAAM,cAAc,GAA6B;IAC/C,UAAU,EAAE,CAAC,gDAAgD,CAAC;IAC9D,MAAM,EAAE,CAAC,kBAAkB,CAAC;CAC7B,CAAC;AAEF,MAAM,eAAe,GAA6B;IAChD,UAAU,EAAE;QACV,0DAA0D;QAC1D,sDAAsD;KACvD;IACD,MAAM,EAAE;QACN,kCAAkC;QAClC,mBAAmB;KACpB;CACF,CAAC;AAEF,SAAS,cAAc,CAAC,QAAgB;IACtC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IACjD,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,OAAO,YAAY,CAAC;IACtF,IAAI,GAAG,KAAK,KAAK;QAAE,OAAO,QAAQ,CAAC;IACnC,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,aAAa,CAAC,MAAc,EAAE,KAAa;IAClD,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;AACvD,CAAC;AAED,SAAS,YAAY,CAAC,KAAe,EAAE,SAAiB,EAAE,QAAgB;IACxE,uCAAuC;IACvC,MAAM,KAAK,GAAG,SAAS,GAAG,CAAC,CAAC;IAC5B,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,IAAI,GAAG,GAAG,KAAK,GAAG,CAAC,CAAC;QACpB,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,OAAO,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;YAC1B,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;gBAAC,GAAG,EAAE,CAAC;gBAAC,SAAS;YAAC,CAAC;YAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,MAAM,IAAI,UAAU;gBAAE,MAAM;YAChC,GAAG,EAAE,CAAC;QACR,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACpE,CAAC;IAED,cAAc;IACd,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,IAAI,KAAK,GAAG,KAAK,CAAC;IAClB,IAAI,GAAG,GAAG,KAAK,CAAC;IAChB,KAAK,IAAI,CAAC,GAAG,KAAK,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1C,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1B,IAAI,EAAE,KAAK,GAAG,EAAE,CAAC;gBAAC,MAAM,EAAE,CAAC;gBAAC,KAAK,GAAG,IAAI,CAAC;YAAC,CAAC;YAC3C,IAAI,EAAE,KAAK,GAAG;gBAAE,MAAM,EAAE,CAAC;QAC3B,CAAC;QACD,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;QACZ,IAAI,KAAK,IAAI,MAAM,IAAI,CAAC;YAAE,MAAM;IAClC,CAAC;IACD,IAAI,CAAC,KAAK;QAAE,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IACpD,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;AACpE,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,QAAgB,EAAE,MAAe;IACzD,MAAM,OAAO,GAAG,MAAM,IAAI,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC7D,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IAE/D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,QAAQ,GAAyB,EAAE,CAAC;IAC1C,MAAM,SAAS,GAAyB,EAAE,CAAC;IAE3C,cAAc;IACd,QAAQ,CAAC,IAAI,CAAC;QACZ,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM;QACrD,SAAS,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;QACpE,QAAQ;KACT,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,QAAQ,CAAC;IAEzB,YAAY;IACZ,KAAK,MAAM,OAAO,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;QACzD,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,KAAK,CAAC;QACV,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;YAC9C,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACtB,MAAM,SAAS,GAAG,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACtD,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;YACvD,QAAQ,CAAC,IAAI,CAAC;gBACZ,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU;gBAChC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI;gBACnD,QAAQ;aACT,CAAC,CAAC;YACH,SAAS,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QAClG,CAAC;IACH,CAAC;IAED,UAAU;IACV,KAAK,MAAM,OAAO,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;QACtD,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,KAAK,CAAC;QACV,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;YAC9C,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACtB,MAAM,SAAS,GAAG,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACtD,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;YACvD,QAAQ,CAAC,IAAI,CAAC;gBACZ,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO;gBAC7B,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI;gBACnD,QAAQ;aACT,CAAC,CAAC;YACH,SAAS,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;QAClG,CAAC;IACH,CAAC;IAED,UAAU;IACV,KAAK,MAAM,OAAO,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;QACvD,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,KAAK,CAAC;QACV,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;YAC9C,MAAM,SAAS,GAAG,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACtD,MAAM,YAAY,GAAG,QAAQ,KAAK,YAAY,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAC/F,MAAM,aAAa,GAAG,QAAQ,KAAK,YAAY;gBAC7C,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;gBAC5E,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAE/E,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;gBACjC,QAAQ,CAAC,IAAI,CAAC;oBACZ,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ;oBAC/D,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;oBAC7C,QAAQ;iBACT,CAAC,CAAC;gBACH,SAAS,CAAC,IAAI,CAAC;oBACb,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;oBACnC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC;oBAC3C,QAAQ,EAAE,SAAS;iBACpB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;AACxC,CAAC"}

package/package.json

@@ -0,0 +1,54 @@
+{
+  "name": "@phoenixaihub/graphrag",
+  "version": "0.1.0",
+  "description": "Graph RAG for private codebases — index code into a knowledge graph with semantic search and MCP server. Self-hosted, no code leaves your machine.",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "bin": {
+    "graphrag": "./dist/cli.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "lint": "tsc --noEmit",
+    "test": "node --test dist/**/*.test.js",
+    "prepublishOnly": "npm run build"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "keywords": [
+    "rag",
+    "graph",
+    "codebase",
+    "knowledge-graph",
+    "mcp",
+    "tree-sitter",
+    "semantic-search",
+    "embeddings",
+    "code-intelligence"
+  ],
+  "author": "Phoenix AI Hub",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/phoenix-assistant/graph-rag-private-codebases.git"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "dependencies": {
+    "better-sqlite3": "^11.7.0",
+    "commander": "^13.1.0",
+    "glob": "^11.0.1",
+    "openai": "^4.78.0",
+    "@anthropic-ai/sdk": "^0.39.0",
+    "@modelcontextprotocol/sdk": "^1.12.1"
+  },
+  "devDependencies": {
+    "@types/better-sqlite3": "^7.6.12",
+    "@types/node": "^22.10.0",
+    "typescript": "^5.7.0"
+  }
+}

package/src/cli.ts

@@ -0,0 +1,52 @@
+#!/usr/bin/env node
+import { Command } from 'commander';
+import { indexRepo } from './commands/index-repo.js';
+import { queryCode } from './commands/query.js';
+import { serve } from './commands/serve.js';
+
+const program = new Command();
+
+program
+  .name('graphrag')
+  .description('Graph RAG for private codebases — self-hosted code intelligence')
+  .version('0.1.0');
+
+program
+  .command('index <repo>')
+  .description('Index a codebase into a knowledge graph')
+  .option('--extensions <exts>', 'File extensions to index (comma-separated)', 'ts,tsx,js,jsx,py')
+  .option('--db <path>', 'Database path', '.graphrag/index.db')
+  .option('--embed', 'Generate embeddings (requires OPENAI_API_KEY)', false)
+  .action(async (repo: string, opts) => {
+    await indexRepo(repo, {
+      extensions: opts.extensions.split(','),
+      dbPath: opts.db,
+      embed: opts.embed,
+    });
+  });
+
+program
+  .command('query <question>')
+  .description('Semantic search over indexed codebase with graph context')
+  .option('--db <path>', 'Database path', '.graphrag/index.db')
+  .option('--limit <n>', 'Max results', '5')
+  .action(async (question: string, opts) => {
+    await queryCode(question, {
+      dbPath: opts.db,
+      limit: parseInt(opts.limit, 10),
+    });
+  });
+
+program
+  .command('serve')
+  .description('Start MCP server for code intelligence')
+  .option('--db <path>', 'Database path', '.graphrag/index.db')
+  .option('--port <n>', 'Port number', '3700')
+  .action(async (opts) => {
+    await serve({
+      dbPath: opts.db,
+      port: parseInt(opts.port, 10),
+    });
+  });
+
+program.parse();

package/src/commands/index-repo.ts

@@ -0,0 +1,96 @@
+import path from 'path';
+import { glob } from 'glob';
+import { GraphDB } from '../db.js';
+import { parseFile } from '../parser.js';
+import { getEmbeddings } from '../embeddings.js';
+
+interface IndexOptions {
+  extensions: string[];
+  dbPath: string;
+  embed: boolean;
+}
+
+export async function indexRepo(repoPath: string, opts: IndexOptions): Promise<void> {
+  const absRepo = path.resolve(repoPath);
+  const dbPath = path.isAbsolute(opts.dbPath) ? opts.dbPath : path.join(absRepo, opts.dbPath);
+  const db = new GraphDB(dbPath);
+
+  const patterns = opts.extensions.map(ext => `**/*.${ext}`);
+  const files = await glob(patterns, {
+    cwd: absRepo,
+    ignore: ['**/node_modules/**', '**/dist/**', '**/.git/**', '**/venv/**', '**/__pycache__/**'],
+    absolute: true,
+  });
+
+  console.log(`Indexing ${files.length} files from ${absRepo}`);
+
+  const entityNameToId = new Map<string, number>();
+  let totalEntities = 0;
+  let totalEdges = 0;
+  const pendingEdges: Array<{ sourceName: string; targetName: string; relation: string }> = [];
+  const chunksToEmbed: Array<{ entityId: number; content: string }> = [];
+
+  for (const file of files) {
+    const relPath = path.relative(absRepo, file);
+    db.clearFile(relPath);
+    
+    const result = parseFile(file);
+    
+    for (const entity of result.entities) {
+      const e = { ...entity, filePath: relPath };
+      const id = db.insertEntity(e);
+      entityNameToId.set(e.name, id);
+      totalEntities++;
+
+      // Create chunk for non-import entities
+      if (e.kind !== 'import') {
+        chunksToEmbed.push({ entityId: id, content: `${e.kind} ${e.name}\n${e.code}` });
+      }
+    }
+
+    pendingEdges.push(...result.edges);
+  }
+
+  // Resolve and insert edges
+  for (const edge of pendingEdges) {
+    const sourceId = entityNameToId.get(edge.sourceName);
+    const targetId = entityNameToId.get(edge.targetName);
+    if (sourceId && targetId) {
+      db.insertEdge({ sourceId, targetId, relation: edge.relation as any });
+      totalEdges++;
+    }
+  }
+
+  // Generate embeddings in batches
+  if (opts.embed && process.env.OPENAI_API_KEY) {
+    console.log(`Generating embeddings for ${chunksToEmbed.length} chunks...`);
+    const batchSize = 100;
+    for (let i = 0; i < chunksToEmbed.length; i += batchSize) {
+      const batch = chunksToEmbed.slice(i, i + batchSize);
+      const embeddings = await getEmbeddings(batch.map(c => c.content));
+      for (let j = 0; j < batch.length; j++) {
+        db.insertChunk({
+          entityId: batch[j].entityId,
+          content: batch[j].content,
+          embedding: new Float64Array(embeddings[j]),
+        });
+      }
+      console.log(`  Embedded ${Math.min(i + batchSize, chunksToEmbed.length)}/${chunksToEmbed.length}`);
+    }
+  } else {
+    // Store chunks without embeddings
+    for (const chunk of chunksToEmbed) {
+      db.insertChunk({ entityId: chunk.entityId, content: chunk.content });
+    }
+  }
+
+  const stats = db.getStats();
+  console.log(`\nDone! Indexed:`);
+  console.log(`  Files:    ${stats.files}`);
+  console.log(`  Entities: ${stats.entities}`);
+  console.log(`  Edges:    ${stats.edges}`);
+  console.log(`  Chunks:   ${stats.chunks}`);
+  console.log(`\nDatabase: ${dbPath}`);
+
+  db.close();
+}

package/src/commands/query.ts

@@ -0,0 +1,43 @@
+import path from 'path';
+import { GraphDB } from '../db.js';
+
+interface QueryOptions {
+  dbPath: string;
+  limit: number;
+}
+
+export async function queryCode(question: string, opts: QueryOptions): Promise<void> {
+  const dbPath = path.resolve(opts.dbPath);
+  const db = new GraphDB(dbPath);
+
+  const results = db.searchEntities(question, opts.limit);
+
+  if (results.length === 0) {
+    console.log('No results found.');
+    db.close();
+    return;
+  }
+
+  console.log(`Found ${results.length} results:\n`);
+
+  for (const entity of results) {
+    const id = (entity as any).id as number;
+    console.log(`━━━ ${entity.kind}: ${entity.name} ━━━`);
+    console.log(`File: ${entity.filePath}:${entity.startLine}-${entity.endLine}`);
+    
+    // Get related entities for context
+    const related = db.getRelated(id);
+    if (related.length > 0) {
+      console.log(`Relations:`);
+      for (const r of related.slice(0, 5)) {
+        console.log(`  ${r.relation} → ${r.entity.name} (${r.entity.kind})`);
+      }
+    }
+    
+    console.log(`\n${entity.code.substring(0, 300)}`);
+    if (entity.code.length > 300) console.log('  ...');
+    console.log('');
+  }
+
+  db.close();
+}

package/src/commands/serve.ts

@@ -0,0 +1,90 @@
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { z } from 'zod';
+import path from 'path';
+import { GraphDB } from '../db.js';
+
+interface ServeOptions {
+  dbPath: string;
+  port: number;
+}
+
+export async function serve(opts: ServeOptions): Promise<void> {
+  const dbPath = path.resolve(opts.dbPath);
+  const db = new GraphDB(dbPath);
+
+  const server = new McpServer({
+    name: 'graphrag',
+    version: '0.1.0',
+  });
+
+  server.tool(
+    'search_codebase',
+    'Search the indexed codebase for functions, classes, and code patterns',
+    { query: z.string().describe('Search query'), limit: z.number().optional().describe('Max results (default 5)') },
+    async ({ query, limit }) => {
+      const results = db.searchEntities(query, limit ?? 5);
+      return {
+        content: [{
+          type: 'text' as const,
+          text: JSON.stringify(results.map(e => ({
+            name: e.name,
+            kind: e.kind,
+            file: `${e.filePath}:${e.startLine}-${e.endLine}`,
+            code: e.code.substring(0, 500),
+          })), null, 2),
+        }],
+      };
+    }
+  );
+
+  server.tool(
+    'get_function_context',
+    'Get a function/class with its full graph context (callers, callees, imports)',
+    { name: z.string().describe('Function or class name') },
+    async ({ name }) => {
+      const entities = db.findEntitiesByName(name);
+      if (entities.length === 0) {
+        return { content: [{ type: 'text' as const, text: `No entity found: ${name}` }] };
+      }
+      const entity = entities[0];
+      const id = (entity as any).id as number;
+      const related = db.getRelated(id);
+      return {
+        content: [{
+          type: 'text' as const,
+          text: JSON.stringify({
+            entity: { name: entity.name, kind: entity.kind, file: entity.filePath, code: entity.code },
+            related: related.map(r => ({
+              name: r.entity.name, kind: r.entity.kind, relation: r.relation,
+              file: r.entity.filePath,
+            })),
+          }, null, 2),
+        }],
+      };
+    }
+  );
+
+  server.tool(
+    'find_usages',
+    'Find all usages of a function, class, or variable',
+    { name: z.string().describe('Entity name to find usages of') },
+    async ({ name }) => {
+      const usages = db.getUsages(name);
+      return {
+        content: [{
+          type: 'text' as const,
+          text: JSON.stringify(usages.map(e => ({
+            name: e.name, kind: e.kind,
+            file: `${e.filePath}:${e.startLine}-${e.endLine}`,
+            code: e.code.substring(0, 200),
+          })), null, 2),
+        }],
+      };
+    }
+  );
+
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+  console.error('GraphRAG MCP server running on stdio');
+}