@phoenixaihub/graphrag 0.1.0

package/.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,17 @@
+---
+name: Bug report
+about: Report a bug
+title: '[Bug] '
+labels: bug
+---
+
+**Describe the bug**
+
+**To reproduce**
+
+**Expected behavior**
+
+**Environment**
+- OS:
+- Node.js version:
+- graphrag version:

package/.github/ISSUE_TEMPLATE/feature_request.md

@@ -0,0 +1,12 @@
+---
+name: Feature request
+about: Suggest an idea
+title: '[Feature] '
+labels: enhancement
+---
+
+**Problem**
+
+**Proposed solution**
+
+**Alternatives considered**

package/.github/pull_request_template.md

@@ -0,0 +1,5 @@
+## Description
+
+## Changes
+
+## Testing

package/.github/workflows/ci.yml

@@ -0,0 +1,22 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        node-version: [18, 20, 22]
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-node@v4
+        with:
+          node-version: ${{ matrix.node-version }}
+      - run: npm ci
+      - run: npm run build
+      - run: npm run lint

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Phoenix AI Hub
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,103 @@
+# @phoenixaihub/graphrag
+
+[![npm](https://img.shields.io/npm/v/@phoenixaihub/graphrag)](https://www.npmjs.com/package/@phoenixaihub/graphrag)
+[![CI](https://github.com/phoenix-assistant/graph-rag-private-codebases/actions/workflows/ci.yml/badge.svg)](https://github.com/phoenix-assistant/graph-rag-private-codebases/actions/workflows/ci.yml)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+**Graph RAG for private codebases** — index your code into a knowledge graph with semantic search and an MCP server. Self-hosted, no code leaves your machine.
+
+## Features
+
+- 🔍 **Index** TypeScript, JavaScript, and Python codebases into a knowledge graph
+- 🧠 **Semantic search** over code with graph-aware context expansion
+- 🔗 **Graph relationships** — functions, classes, imports, exports, call chains
+- 🤖 **MCP server** — plug into Claude, Cursor, or any MCP-compatible client
+- 🏠 **Self-hosted** — SQLite-backed, runs locally, zero cloud dependencies
+- ⚡ **Optional embeddings** via OpenAI `text-embedding-3-small`
+
+## Quick Start
+
+```bash
+npm install -g @phoenixaihub/graphrag
+
+# Index a repo
+graphrag index ./my-project
+
+# Search code
+graphrag query "authentication middleware"
+
+# Start MCP server
+graphrag serve --db ./my-project/.graphrag/index.db
+```
+
+## MCP Configuration
+
+Add to your Claude Desktop / Cursor config:
+
+```json
+{
+  "mcpServers": {
+    "graphrag": {
+      "command": "graphrag",
+      "args": ["serve", "--db", "/path/to/project/.graphrag/index.db"]
+    }
+  }
+}
+```
+
+### Available MCP Tools
+
+| Tool | Description |
+|------|-------------|
+| `search_codebase` | Semantic search across indexed code entities |
+| `get_function_context` | Get a function/class with full graph context (callers, callees, imports) |
+| `find_usages` | Find all usages of a function, class, or variable |
+
+## With Embeddings
+
+Set `OPENAI_API_KEY` and use `--embed`:
+
+```bash
+export OPENAI_API_KEY=sk-...
+graphrag index ./my-project --embed
+```
+
+## How It Works
+
+1. **Parse** — Extracts functions, classes, imports/exports using regex-based parsing
+2. **Graph** — Builds entity relationship graph (contains, imports, calls, extends)
+3. **Store** — Persists to SQLite with full-text search indexes
+4. **Search** — Text search + graph traversal for context-rich results
+5. **Serve** — MCP server exposes tools for AI-assisted code exploration
+
+## Comparison
+
+| Feature | GraphRAG | Sourcegraph | Codeium |
+|---------|----------|-------------|---------|
+| Self-hosted | ✅ Local SQLite | ❌ Server required | ❌ Cloud |
+| Privacy | ✅ No code leaves machine | ⚠️ Depends on deployment | ❌ Cloud-processed |
+| MCP support | ✅ Built-in | ❌ | ❌ |
+| Setup time | < 1 min | Hours | Minutes |
+| Cost | Free | $$$$ | Free tier limited |
+| Graph relationships | ✅ | ✅ | ❌ |
+| Languages | TS/JS/Python | 30+ | 70+ |
+
+## Programmatic API
+
+```typescript
+import { GraphDB, parseFile } from '@phoenixaihub/graphrag';
+
+const db = new GraphDB('.graphrag/index.db');
+const result = parseFile('./src/main.ts');
+
+for (const entity of result.entities) {
+  db.insertEntity(entity);
+}
+
+const results = db.searchEntities('handler', 5);
+console.log(results);
+```
+
+## License
+
+MIT

package/dist/cli.d.ts

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+export {};
+//# sourceMappingURL=cli.d.ts.map

package/dist/cli.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.d.ts","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":""}

package/dist/cli.js

@@ -0,0 +1,47 @@
+#!/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, 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, 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();
+//# sourceMappingURL=cli.js.map

package/dist/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChD,OAAO,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAE5C,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAE9B,OAAO;KACJ,IAAI,CAAC,UAAU,CAAC;KAChB,WAAW,CAAC,iEAAiE,CAAC;KAC9E,OAAO,CAAC,OAAO,CAAC,CAAC;AAEpB,OAAO;KACJ,OAAO,CAAC,cAAc,CAAC;KACvB,WAAW,CAAC,yCAAyC,CAAC;KACtD,MAAM,CAAC,qBAAqB,EAAE,4CAA4C,EAAE,kBAAkB,CAAC;KAC/F,MAAM,CAAC,aAAa,EAAE,eAAe,EAAE,oBAAoB,CAAC;KAC5D,MAAM,CAAC,SAAS,EAAE,+CAA+C,EAAE,KAAK,CAAC;KACzE,MAAM,CAAC,KAAK,EAAE,IAAY,EAAE,IAAI,EAAE,EAAE;IACnC,MAAM,SAAS,CAAC,IAAI,EAAE;QACpB,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC;QACtC,MAAM,EAAE,IAAI,CAAC,EAAE;QACf,KAAK,EAAE,IAAI,CAAC,KAAK;KAClB,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,kBAAkB,CAAC;KAC3B,WAAW,CAAC,0DAA0D,CAAC;KACvE,MAAM,CAAC,aAAa,EAAE,eAAe,EAAE,oBAAoB,CAAC;KAC5D,MAAM,CAAC,aAAa,EAAE,aAAa,EAAE,GAAG,CAAC;KACzC,MAAM,CAAC,KAAK,EAAE,QAAgB,EAAE,IAAI,EAAE,EAAE;IACvC,MAAM,SAAS,CAAC,QAAQ,EAAE;QACxB,MAAM,EAAE,IAAI,CAAC,EAAE;QACf,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC;KAChC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,OAAO,CAAC;KAChB,WAAW,CAAC,wCAAwC,CAAC;KACrD,MAAM,CAAC,aAAa,EAAE,eAAe,EAAE,oBAAoB,CAAC;KAC5D,MAAM,CAAC,YAAY,EAAE,aAAa,EAAE,MAAM,CAAC;KAC3C,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;IACrB,MAAM,KAAK,CAAC;QACV,MAAM,EAAE,IAAI,CAAC,EAAE;QACf,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;KAC9B,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEL,OAAO,CAAC,KAAK,EAAE,CAAC"}

package/dist/commands/index-repo.d.ts

@@ -0,0 +1,8 @@
+interface IndexOptions {
+    extensions: string[];
+    dbPath: string;
+    embed: boolean;
+}
+export declare function indexRepo(repoPath: string, opts: IndexOptions): Promise<void>;
+export {};
+//# sourceMappingURL=index-repo.d.ts.map

package/dist/commands/index-repo.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index-repo.d.ts","sourceRoot":"","sources":["../../src/commands/index-repo.ts"],"names":[],"mappings":"AAMA,UAAU,YAAY;IACpB,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,OAAO,CAAC;CAChB;AAED,wBAAsB,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,CAmFnF"}

package/dist/commands/index-repo.js

@@ -0,0 +1,79 @@
+import path from 'path';
+import { glob } from 'glob';
+import { GraphDB } from '../db.js';
+import { parseFile } from '../parser.js';
+import { getEmbeddings } from '../embeddings.js';
+export async function indexRepo(repoPath, opts) {
+    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();
+    let totalEntities = 0;
+    let totalEdges = 0;
+    const pendingEdges = [];
+    const chunksToEmbed = [];
+    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 });
+            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();
+}
+//# sourceMappingURL=index-repo.js.map

package/dist/commands/index-repo.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index-repo.js","sourceRoot":"","sources":["../../src/commands/index-repo.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,OAAO,EAAE,MAAM,UAAU,CAAC;AACnC,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAQjD,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,QAAgB,EAAE,IAAkB;IAClE,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvC,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IAC5F,MAAM,EAAE,GAAG,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;IAE/B,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,GAAG,EAAE,CAAC,CAAC;IAC3D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE;QACjC,GAAG,EAAE,OAAO;QACZ,MAAM,EAAE,CAAC,oBAAoB,EAAE,YAAY,EAAE,YAAY,EAAE,YAAY,EAAE,mBAAmB,CAAC;QAC7F,QAAQ,EAAE,IAAI;KACf,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,YAAY,KAAK,CAAC,MAAM,eAAe,OAAO,EAAE,CAAC,CAAC;IAE9D,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;IACjD,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,MAAM,YAAY,GAAwE,EAAE,CAAC;IAC7F,MAAM,aAAa,GAAiD,EAAE,CAAC;IAEvE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC7C,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAEtB,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;QAE/B,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,CAAC,GAAG,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;YAC3C,MAAM,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9B,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC/B,aAAa,EAAE,CAAC;YAEhB,uCAAuC;YACvC,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACxB,aAAa,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAClF,CAAC;QACH,CAAC;QAED,YAAY,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IAED,2BAA2B;IAC3B,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACrD,IAAI,QAAQ,IAAI,QAAQ,EAAE,CAAC;YACzB,EAAE,CAAC,UAAU,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAe,EAAE,CAAC,CAAC;YACtE,UAAU,EAAE,CAAC;QACf,CAAC;IACH,CAAC;IAED,iCAAiC;IACjC,IAAI,IAAI,CAAC,KAAK,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;QAC7C,OAAO,CAAC,GAAG,CAAC,6BAA6B,aAAa,CAAC,MAAM,YAAY,CAAC,CAAC;QAC3E,MAAM,SAAS,GAAG,GAAG,CAAC;QACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YACzD,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;YACpD,MAAM,UAAU,GAAG,MAAM,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAClE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,EAAE,CAAC,WAAW,CAAC;oBACb,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ;oBAC3B,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO;oBACzB,SAAS,EAAE,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;iBAC3C,CAAC,CAAC;YACL,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,EAAE,aAAa,CAAC,MAAM,CAAC,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;QACrG,CAAC;IACH,CAAC;SAAM,CAAC;QACN,kCAAkC;QAClC,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;YAClC,EAAE,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED,MAAM,KAAK,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;IAC5B,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;IAChC,OAAO,CAAC,GAAG,CAAC,eAAe,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;IAC1C,OAAO,CAAC,GAAG,CAAC,eAAe,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,eAAe,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;IAC1C,OAAO,CAAC,GAAG,CAAC,eAAe,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAC3C,OAAO,CAAC,GAAG,CAAC,eAAe,MAAM,EAAE,CAAC,CAAC;IAErC,EAAE,CAAC,KAAK,EAAE,CAAC;AACb,CAAC"}

package/dist/commands/query.d.ts

@@ -0,0 +1,7 @@
+interface QueryOptions {
+    dbPath: string;
+    limit: number;
+}
+export declare function queryCode(question: string, opts: QueryOptions): Promise<void>;
+export {};
+//# sourceMappingURL=query.d.ts.map

package/dist/commands/query.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query.d.ts","sourceRoot":"","sources":["../../src/commands/query.ts"],"names":[],"mappings":"AAGA,UAAU,YAAY;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;CACf;AAED,wBAAsB,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,CAkCnF"}

package/dist/commands/query.js

@@ -0,0 +1,32 @@
+import path from 'path';
+import { GraphDB } from '../db.js';
+export async function queryCode(question, opts) {
+    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.id;
+        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();
+}
+//# sourceMappingURL=query.js.map

package/dist/commands/query.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query.js","sourceRoot":"","sources":["../../src/commands/query.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,OAAO,EAAE,MAAM,UAAU,CAAC;AAOnC,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,QAAgB,EAAE,IAAkB;IAClE,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACzC,MAAM,EAAE,GAAG,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;IAE/B,MAAM,OAAO,GAAG,EAAE,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IAExD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACjC,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,OAAO;IACT,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,SAAS,OAAO,CAAC,MAAM,aAAa,CAAC,CAAC;IAElD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,EAAE,GAAI,MAAc,CAAC,EAAY,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,OAAO,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,CAAC;QACtD,OAAO,CAAC,GAAG,CAAC,SAAS,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QAE9E,mCAAmC;QACnC,MAAM,OAAO,GAAG,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YAC1B,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;gBACpC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,QAAQ,MAAM,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC;YACvE,CAAC;QACH,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,KAAK,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAClD,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,GAAG;YAAE,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACnD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClB,CAAC;IAED,EAAE,CAAC,KAAK,EAAE,CAAC;AACb,CAAC"}

package/dist/commands/serve.d.ts

@@ -0,0 +1,7 @@
+interface ServeOptions {
+    dbPath: string;
+    port: number;
+}
+export declare function serve(opts: ServeOptions): Promise<void>;
+export {};
+//# sourceMappingURL=serve.d.ts.map

package/dist/commands/serve.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"serve.d.ts","sourceRoot":"","sources":["../../src/commands/serve.ts"],"names":[],"mappings":"AAMA,UAAU,YAAY;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;CACd;AAED,wBAAsB,KAAK,CAAC,IAAI,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,CA8E7D"}

package/dist/commands/serve.js

@@ -0,0 +1,65 @@
+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';
+export async function serve(opts) {
+    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',
+                    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', text: `No entity found: ${name}` }] };
+        }
+        const entity = entities[0];
+        const id = entity.id;
+        const related = db.getRelated(id);
+        return {
+            content: [{
+                    type: 'text',
+                    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',
+                    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');
+}
+//# sourceMappingURL=serve.js.map

package/dist/commands/serve.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"serve.js","sourceRoot":"","sources":["../../src/commands/serve.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,OAAO,EAAE,MAAM,UAAU,CAAC;AAOnC,MAAM,CAAC,KAAK,UAAU,KAAK,CAAC,IAAkB;IAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACzC,MAAM,EAAE,GAAG,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;IAE/B,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;QAC3B,IAAI,EAAE,UAAU;QAChB,OAAO,EAAE,OAAO;KACjB,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CACT,iBAAiB,EACjB,uEAAuE,EACvE,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,EAChH,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE;QACzB,MAAM,OAAO,GAAG,EAAE,CAAC,cAAc,CAAC,KAAK,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;QACrD,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;wBACrC,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,IAAI,EAAE,GAAG,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,EAAE;wBACjD,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;qBAC/B,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;iBACd,CAAC;SACH,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,sBAAsB,EACtB,8EAA8E,EAC9E,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,EACvD,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;QACjB,MAAM,QAAQ,GAAG,EAAE,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAC7C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,oBAAoB,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC;QACpF,CAAC;QACD,MAAM,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,EAAE,GAAI,MAAc,CAAC,EAAY,CAAC;QACxC,MAAM,OAAO,GAAG,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAClC,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE;wBAC1F,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;4BACzB,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ;4BAC9D,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,QAAQ;yBACxB,CAAC,CAAC;qBACJ,EAAE,IAAI,EAAE,CAAC,CAAC;iBACZ,CAAC;SACH,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,aAAa,EACb,mDAAmD,EACnD,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+BAA+B,CAAC,EAAE,EAC9D,KAAK,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;QACjB,MAAM,MAAM,GAAG,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAClC,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;wBACpC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI;wBAC1B,IAAI,EAAE,GAAG,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,EAAE;wBACjD,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC;qBAC/B,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;iBACd,CAAC;SACH,CAAC;IACJ,CAAC,CACF,CAAC;IAEF,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,OAAO,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;AACxD,CAAC"}

package/dist/db.d.ts

@@ -0,0 +1,46 @@
+export interface Entity {
+    id?: number;
+    filePath: string;
+    name: string;
+    kind: 'file' | 'function' | 'class' | 'method' | 'variable' | 'import' | 'export';
+    startLine: number;
+    endLine: number;
+    code: string;
+    language: string;
+}
+export interface Edge {
+    id?: number;
+    sourceId: number;
+    targetId: number;
+    relation: 'contains' | 'imports' | 'exports' | 'calls' | 'extends' | 'implements';
+}
+export interface Chunk {
+    id?: number;
+    entityId: number;
+    content: string;
+    embedding?: Float64Array | null;
+}
+export declare class GraphDB {
+    private db;
+    constructor(dbPath: string);
+    insertEntity(e: Entity): number;
+    insertEdge(e: Edge): number;
+    insertChunk(c: Chunk): number;
+    getEntity(id: number): Entity | undefined;
+    findEntitiesByName(name: string): Entity[];
+    searchEntities(query: string, limit?: number): Entity[];
+    getRelated(entityId: number, direction?: 'outgoing' | 'incoming' | 'both'): Array<{
+        entity: Entity;
+        relation: string;
+    }>;
+    getUsages(name: string): Entity[];
+    clearFile(filePath: string): void;
+    getStats(): {
+        entities: number;
+        edges: number;
+        chunks: number;
+        files: number;
+    };
+    close(): void;
+}
+//# sourceMappingURL=db.d.ts.map

package/dist/db.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"db.d.ts","sourceRoot":"","sources":["../src/db.ts"],"names":[],"mappings":"AAIA,MAAM,WAAW,MAAM;IACrB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,GAAG,UAAU,GAAG,OAAO,GAAG,QAAQ,GAAG,UAAU,GAAG,QAAQ,GAAG,QAAQ,CAAC;IAClF,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,IAAI;IACnB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,GAAG,OAAO,GAAG,SAAS,GAAG,YAAY,CAAC;CACnF;AAED,MAAM,WAAW,KAAK;IACpB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,YAAY,GAAG,IAAI,CAAC;CACjC;AAoCD,qBAAa,OAAO;IAClB,OAAO,CAAC,EAAE,CAAoB;gBAElB,MAAM,EAAE,MAAM;IAQ1B,YAAY,CAAC,CAAC,EAAE,MAAM,GAAG,MAAM;IAQ/B,UAAU,CAAC,CAAC,EAAE,IAAI,GAAG,MAAM;IAQ3B,WAAW,CAAC,CAAC,EAAE,KAAK,GAAG,MAAM;IAS7B,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;IAIzC,kBAAkB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;IAM1C,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,SAAI,GAAG,MAAM,EAAE;IAWlD,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,GAAE,UAAU,GAAG,UAAU,GAAG,MAAe,GAAG,KAAK,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IAiB/H,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;IASjC,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAUjC,QAAQ,IAAI;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE;IAQ9E,KAAK,IAAI,IAAI;CAGd"}

package/dist/db.js

@@ -0,0 +1,117 @@
+import Database from 'better-sqlite3';
+import path from 'path';
+import fs from 'fs';
+const SCHEMA = `
+CREATE TABLE IF NOT EXISTS entities (
+  id INTEGER PRIMARY KEY AUTOINCREMENT,
+  file_path TEXT NOT NULL,
+  name TEXT NOT NULL,
+  kind TEXT NOT NULL,
+  start_line INTEGER NOT NULL,
+  end_line INTEGER NOT NULL,
+  code TEXT NOT NULL,
+  language TEXT NOT NULL
+);
+
+CREATE TABLE IF NOT EXISTS edges (
+  id INTEGER PRIMARY KEY AUTOINCREMENT,
+  source_id INTEGER NOT NULL REFERENCES entities(id),
+  target_id INTEGER NOT NULL REFERENCES entities(id),
+  relation TEXT NOT NULL
+);
+
+CREATE TABLE IF NOT EXISTS chunks (
+  id INTEGER PRIMARY KEY AUTOINCREMENT,
+  entity_id INTEGER NOT NULL REFERENCES entities(id),
+  content TEXT NOT NULL,
+  embedding BLOB
+);
+
+CREATE INDEX IF NOT EXISTS idx_entities_name ON entities(name);
+CREATE INDEX IF NOT EXISTS idx_entities_kind ON entities(kind);
+CREATE INDEX IF NOT EXISTS idx_entities_file ON entities(file_path);
+CREATE INDEX IF NOT EXISTS idx_edges_source ON edges(source_id);
+CREATE INDEX IF NOT EXISTS idx_edges_target ON edges(target_id);
+CREATE INDEX IF NOT EXISTS idx_chunks_entity ON chunks(entity_id);
+`;
+export class GraphDB {
+    db;
+    constructor(dbPath) {
+        const dir = path.dirname(dbPath);
+        if (!fs.existsSync(dir))
+            fs.mkdirSync(dir, { recursive: true });
+        this.db = new Database(dbPath);
+        this.db.pragma('journal_mode = WAL');
+        this.db.exec(SCHEMA);
+    }
+    insertEntity(e) {
+        const stmt = this.db.prepare('INSERT INTO entities (file_path, name, kind, start_line, end_line, code, language) VALUES (?, ?, ?, ?, ?, ?, ?)');
+        const result = stmt.run(e.filePath, e.name, e.kind, e.startLine, e.endLine, e.code, e.language);
+        return Number(result.lastInsertRowid);
+    }
+    insertEdge(e) {
+        const stmt = this.db.prepare('INSERT INTO edges (source_id, target_id, relation) VALUES (?, ?, ?)');
+        const result = stmt.run(e.sourceId, e.targetId, e.relation);
+        return Number(result.lastInsertRowid);
+    }
+    insertChunk(c) {
+        const stmt = this.db.prepare('INSERT INTO chunks (entity_id, content, embedding) VALUES (?, ?, ?)');
+        const embeddingBlob = c.embedding ? Buffer.from(c.embedding.buffer) : null;
+        const result = stmt.run(c.entityId, c.content, embeddingBlob);
+        return Number(result.lastInsertRowid);
+    }
+    getEntity(id) {
+        return this.db.prepare('SELECT * FROM entities WHERE id = ?').get(id);
+    }
+    findEntitiesByName(name) {
+        return this.db.prepare('SELECT * FROM entities WHERE name LIKE ?').all(`%${name}%`);
+    }
+    searchEntities(query, limit = 5) {
+        // Text-based search across name and code
+        const terms = query.toLowerCase().split(/\s+/).filter(Boolean);
+        if (terms.length === 0)
+            return [];
+        const conditions = terms.map(() => '(LOWER(name) LIKE ? OR LOWER(code) LIKE ?)').join(' AND ');
+        const params = terms.flatMap(t => [`%${t}%`, `%${t}%`]);
+        return this.db.prepare(`SELECT * FROM entities WHERE ${conditions} ORDER BY kind, name LIMIT ?`).all(...params, limit);
+    }
+    getRelated(entityId, direction = 'both') {
+        const results = [];
+        if (direction === 'outgoing' || direction === 'both') {
+            const rows = this.db.prepare('SELECT e.*, ed.relation FROM entities e JOIN edges ed ON e.id = ed.target_id WHERE ed.source_id = ?').all(entityId);
+            results.push(...rows.map(r => ({ entity: r, relation: r.relation })));
+        }
+        if (direction === 'incoming' || direction === 'both') {
+            const rows = this.db.prepare('SELECT e.*, ed.relation FROM entities e JOIN edges ed ON e.id = ed.source_id WHERE ed.target_id = ?').all(entityId);
+            results.push(...rows.map(r => ({ entity: r, relation: r.relation })));
+        }
+        return results;
+    }
+    getUsages(name) {
+        return this.db.prepare(`SELECT DISTINCT e2.* FROM entities e1
+       JOIN edges ed ON e1.id = ed.target_id
+       JOIN entities e2 ON e2.id = ed.source_id
+       WHERE e1.name = ? AND ed.relation IN ('calls', 'imports')`).all(name);
+    }
+    clearFile(filePath) {
+        const entities = this.db.prepare('SELECT id FROM entities WHERE file_path = ?').all(filePath);
+        const ids = entities.map(e => e.id);
+        if (ids.length === 0)
+            return;
+        const placeholders = ids.map(() => '?').join(',');
+        this.db.prepare(`DELETE FROM chunks WHERE entity_id IN (${placeholders})`).run(...ids);
+        this.db.prepare(`DELETE FROM edges WHERE source_id IN (${placeholders}) OR target_id IN (${placeholders})`).run(...ids, ...ids);
+        this.db.prepare(`DELETE FROM entities WHERE file_path = ?`).run(filePath);
+    }
+    getStats() {
+        const entities = this.db.prepare('SELECT COUNT(*) as c FROM entities').get().c;
+        const edges = this.db.prepare('SELECT COUNT(*) as c FROM edges').get().c;
+        const chunks = this.db.prepare('SELECT COUNT(*) as c FROM chunks').get().c;
+        const files = this.db.prepare('SELECT COUNT(DISTINCT file_path) as c FROM entities').get().c;
+        return { entities, edges, chunks, files };
+    }
+    close() {
+        this.db.close();
+    }
+}
+//# sourceMappingURL=db.js.map