edhindex 1.0.0 → 1.0.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +228 -0
- package/dist/cli/commands/config_cmd.d.ts +2 -0
- package/dist/cli/commands/config_cmd.d.ts.map +1 -0
- package/dist/cli/commands/config_cmd.js +53 -0
- package/dist/cli/commands/config_cmd.js.map +1 -0
- package/dist/cli/commands/doctor_cmd.d.ts +2 -0
- package/dist/cli/commands/doctor_cmd.d.ts.map +1 -0
- package/dist/cli/commands/doctor_cmd.js +32 -0
- package/dist/cli/commands/doctor_cmd.js.map +1 -0
- package/dist/cli/commands/graph.d.ts +2 -0
- package/dist/cli/commands/graph.d.ts.map +1 -0
- package/dist/cli/commands/graph.js +26 -0
- package/dist/cli/commands/graph.js.map +1 -0
- package/dist/cli/commands/index_cmd.d.ts +2 -0
- package/dist/cli/commands/index_cmd.d.ts.map +1 -0
- package/dist/cli/commands/index_cmd.js +110 -0
- package/dist/cli/commands/index_cmd.js.map +1 -0
- package/dist/cli/commands/init.d.ts +2 -0
- package/dist/cli/commands/init.d.ts.map +1 -0
- package/dist/cli/commands/init.js +53 -0
- package/dist/cli/commands/init.js.map +1 -0
- package/dist/cli/commands/knowledge-graph.d.ts +8 -0
- package/dist/cli/commands/knowledge-graph.d.ts.map +1 -0
- package/dist/cli/commands/knowledge-graph.js +76 -0
- package/dist/cli/commands/knowledge-graph.js.map +1 -0
- package/dist/cli/commands/models.d.ts +2 -0
- package/dist/cli/commands/models.d.ts.map +1 -0
- package/dist/cli/commands/models.js +17 -0
- package/dist/cli/commands/models.js.map +1 -0
- package/dist/cli/commands/reset.d.ts +2 -0
- package/dist/cli/commands/reset.d.ts.map +1 -0
- package/dist/cli/commands/reset.js +35 -0
- package/dist/cli/commands/reset.js.map +1 -0
- package/dist/cli/commands/search.d.ts +2 -0
- package/dist/cli/commands/search.d.ts.map +1 -0
- package/dist/cli/commands/search.js +40 -0
- package/dist/cli/commands/search.js.map +1 -0
- package/dist/cli/commands/start.d.ts +2 -0
- package/dist/cli/commands/start.d.ts.map +1 -0
- package/dist/cli/commands/start.js +265 -0
- package/dist/cli/commands/start.js.map +1 -0
- package/dist/cli/commands/status.d.ts +2 -0
- package/dist/cli/commands/status.d.ts.map +1 -0
- package/dist/cli/commands/status.js +47 -0
- package/dist/cli/commands/status.js.map +1 -0
- package/dist/cli/index.d.ts +3 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +121 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/config/index.d.ts +28 -0
- package/dist/config/index.d.ts.map +1 -0
- package/dist/config/index.js +65 -0
- package/dist/config/index.js.map +1 -0
- package/dist/doctor/index.d.ts +8 -0
- package/dist/doctor/index.d.ts.map +1 -0
- package/dist/doctor/index.js +117 -0
- package/dist/doctor/index.js.map +1 -0
- package/dist/embeddings/provider.d.ts +9 -0
- package/dist/embeddings/provider.d.ts.map +1 -0
- package/dist/embeddings/provider.js +2 -0
- package/dist/embeddings/provider.js.map +1 -0
- package/dist/embeddings/transformers.d.ts +26 -0
- package/dist/embeddings/transformers.d.ts.map +1 -0
- package/dist/embeddings/transformers.js +120 -0
- package/dist/embeddings/transformers.js.map +1 -0
- package/dist/graph/build.d.ts +13 -0
- package/dist/graph/build.d.ts.map +1 -0
- package/dist/graph/build.js +33 -0
- package/dist/graph/build.js.map +1 -0
- package/dist/graph/builder.d.ts +24 -0
- package/dist/graph/builder.d.ts.map +1 -0
- package/dist/graph/builder.js +231 -0
- package/dist/graph/builder.js.map +1 -0
- package/dist/graph/render.d.ts +3 -0
- package/dist/graph/render.d.ts.map +1 -0
- package/dist/graph/render.js +60 -0
- package/dist/graph/render.js.map +1 -0
- package/dist/graph/serve.d.ts +6 -0
- package/dist/graph/serve.d.ts.map +1 -0
- package/dist/graph/serve.js +389 -0
- package/dist/graph/serve.js.map +1 -0
- package/dist/graph/service.d.ts +29 -0
- package/dist/graph/service.d.ts.map +1 -0
- package/dist/graph/service.js +73 -0
- package/dist/graph/service.js.map +1 -0
- package/dist/graph/storage.d.ts +33 -0
- package/dist/graph/storage.d.ts.map +1 -0
- package/dist/graph/storage.js +164 -0
- package/dist/graph/storage.js.map +1 -0
- package/dist/graph/svg.d.ts +27 -0
- package/dist/graph/svg.d.ts.map +1 -0
- package/dist/graph/svg.js +212 -0
- package/dist/graph/svg.js.map +1 -0
- package/dist/graph/types.d.ts +34 -0
- package/dist/graph/types.d.ts.map +1 -0
- package/dist/graph/types.js +9 -0
- package/dist/graph/types.js.map +1 -0
- package/dist/indexer/chunker.d.ts +18 -0
- package/dist/indexer/chunker.d.ts.map +1 -0
- package/dist/indexer/chunker.js +139 -0
- package/dist/indexer/chunker.js.map +1 -0
- package/dist/indexer/file-utils.d.ts +17 -0
- package/dist/indexer/file-utils.d.ts.map +1 -0
- package/dist/indexer/file-utils.js +123 -0
- package/dist/indexer/file-utils.js.map +1 -0
- package/dist/indexer/ignore.d.ts +5 -0
- package/dist/indexer/ignore.d.ts.map +1 -0
- package/dist/indexer/ignore.js +96 -0
- package/dist/indexer/ignore.js.map +1 -0
- package/dist/indexer/incremental.d.ts +16 -0
- package/dist/indexer/incremental.d.ts.map +1 -0
- package/dist/indexer/incremental.js +55 -0
- package/dist/indexer/incremental.js.map +1 -0
- package/dist/indexer/indexer.d.ts +36 -0
- package/dist/indexer/indexer.d.ts.map +1 -0
- package/dist/indexer/indexer.js +121 -0
- package/dist/indexer/indexer.js.map +1 -0
- package/dist/indexer/parser.d.ts +30 -0
- package/dist/indexer/parser.d.ts.map +1 -0
- package/dist/indexer/parser.js +225 -0
- package/dist/indexer/parser.js.map +1 -0
- package/dist/logging.d.ts +15 -0
- package/dist/logging.d.ts.map +1 -0
- package/dist/logging.js +42 -0
- package/dist/logging.js.map +1 -0
- package/dist/mcp/server.d.ts +4 -0
- package/dist/mcp/server.d.ts.map +1 -0
- package/dist/mcp/server.js +226 -0
- package/dist/mcp/server.js.map +1 -0
- package/dist/mcp-config.d.ts +4 -0
- package/dist/mcp-config.d.ts.map +1 -0
- package/dist/mcp-config.js +175 -0
- package/dist/mcp-config.js.map +1 -0
- package/dist/progress.d.ts +12 -0
- package/dist/progress.d.ts.map +1 -0
- package/dist/progress.js +16 -0
- package/dist/progress.js.map +1 -0
- package/dist/retrieval/bm25.d.ts +6 -0
- package/dist/retrieval/bm25.d.ts.map +1 -0
- package/dist/retrieval/bm25.js +20 -0
- package/dist/retrieval/bm25.js.map +1 -0
- package/dist/retrieval/reranker.d.ts +11 -0
- package/dist/retrieval/reranker.d.ts.map +1 -0
- package/dist/retrieval/reranker.js +46 -0
- package/dist/retrieval/reranker.js.map +1 -0
- package/dist/retrieval/search.d.ts +35 -0
- package/dist/retrieval/search.d.ts.map +1 -0
- package/dist/retrieval/search.js +95 -0
- package/dist/retrieval/search.js.map +1 -0
- package/dist/security.d.ts +5 -0
- package/dist/security.d.ts.map +1 -0
- package/dist/security.js +42 -0
- package/dist/security.js.map +1 -0
- package/dist/storage/fts.d.ts +30 -0
- package/dist/storage/fts.d.ts.map +1 -0
- package/dist/storage/fts.js +107 -0
- package/dist/storage/fts.js.map +1 -0
- package/dist/storage/metadata.d.ts +46 -0
- package/dist/storage/metadata.d.ts.map +1 -0
- package/dist/storage/metadata.js +133 -0
- package/dist/storage/metadata.js.map +1 -0
- package/dist/vector/lancedb.d.ts +25 -0
- package/dist/vector/lancedb.d.ts.map +1 -0
- package/dist/vector/lancedb.js +93 -0
- package/dist/vector/lancedb.js.map +1 -0
- package/dist/vector/store.d.ts +16 -0
- package/dist/vector/store.d.ts.map +1 -0
- package/dist/vector/store.js +2 -0
- package/dist/vector/store.js.map +1 -0
- package/dist/version.d.ts +12 -0
- package/dist/version.d.ts.map +1 -0
- package/dist/version.js +46 -0
- package/dist/version.js.map +1 -0
- package/dist/watcher/index.d.ts +15 -0
- package/dist/watcher/index.d.ts.map +1 -0
- package/dist/watcher/index.js +65 -0
- package/dist/watcher/index.js.map +1 -0
- package/package.json +35 -4
- package/src/cli/commands/graph.ts +30 -0
- package/src/cli/commands/init.ts +41 -3
- package/src/cli/commands/knowledge-graph.ts +85 -0
- package/src/cli/commands/reset.ts +40 -0
- package/src/cli/commands/start.ts +54 -21
- package/src/cli/index.ts +33 -0
- package/src/config/index.ts +13 -0
- package/src/graph/build.ts +45 -0
- package/src/graph/builder.ts +243 -0
- package/src/graph/render.ts +67 -0
- package/src/graph/serve.ts +399 -0
- package/src/graph/service.ts +84 -0
- package/src/graph/storage.ts +181 -0
- package/src/graph/svg.ts +241 -0
- package/src/graph/types.ts +68 -0
- package/src/mcp/server.ts +212 -85
- package/src/mcp-config.ts +185 -0
- package/src/storage/metadata.ts +9 -1
- package/.edhindexignore +0 -66
- package/BUILD_BRIEF.md +0 -228
- package/tsconfig.json +0 -21
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
import { join } from 'node:path';
|
|
2
|
+
import { existsSync } from 'node:fs';
|
|
3
|
+
import { GraphService } from '../../graph/service.js';
|
|
4
|
+
import { startGraphServer, writeViewerFile } from '../../graph/serve.js';
|
|
5
|
+
import chalk from 'chalk';
|
|
6
|
+
import { spawn } from 'node:child_process';
|
|
7
|
+
export async function kgCommand(rootPath, flags) {
|
|
8
|
+
const indexDir = join(rootPath, '.edhindex');
|
|
9
|
+
if (!existsSync(indexDir)) {
|
|
10
|
+
console.log(chalk.yellow('No index found. Run "edhindex start" first.'));
|
|
11
|
+
return;
|
|
12
|
+
}
|
|
13
|
+
if (flags.write) {
|
|
14
|
+
console.log(chalk.dim('Writing graph viewer...'));
|
|
15
|
+
const outFile = join(rootPath, '.edhindex', 'knowledge-graph.html');
|
|
16
|
+
writeViewerFile(rootPath, outFile);
|
|
17
|
+
console.log(chalk.green(`\n Viewer written: ${outFile}`));
|
|
18
|
+
console.log(chalk.dim('\n Open in VS Code:'));
|
|
19
|
+
console.log(chalk.dim(' 1. File > Open File... > select the file'));
|
|
20
|
+
console.log(chalk.dim(' 2. Or drag into VS Code and click "Preview"'));
|
|
21
|
+
console.log(chalk.dim(' 3. Or right-click → "Open with Live Server"'));
|
|
22
|
+
console.log(chalk.dim('\n Or open in browser:'));
|
|
23
|
+
console.log(chalk.dim(` file://${outFile}`));
|
|
24
|
+
return;
|
|
25
|
+
}
|
|
26
|
+
if (flags.serve) {
|
|
27
|
+
const port = flags.port ? parseInt(flags.port, 10) : 0;
|
|
28
|
+
console.log(chalk.dim('Starting graph server...'));
|
|
29
|
+
const { port: actualPort, close } = await startGraphServer(rootPath, port);
|
|
30
|
+
const url = `http://127.0.0.1:${actualPort}`;
|
|
31
|
+
console.log(chalk.green(`\n Knowledge Graph: ${url}`));
|
|
32
|
+
console.log(chalk.dim('\n Press Ctrl+C to stop.\n'));
|
|
33
|
+
// Try to open browser
|
|
34
|
+
const cmd = process.platform === 'darwin' ? 'open' : process.platform === 'win32' ? 'start' : 'xdg-open';
|
|
35
|
+
try {
|
|
36
|
+
const child = spawn(cmd, [url], { detached: true, stdio: 'ignore' });
|
|
37
|
+
child.unref();
|
|
38
|
+
}
|
|
39
|
+
catch { /* browser open not critical */ }
|
|
40
|
+
// Keep alive
|
|
41
|
+
process.on('SIGINT', () => { close(); process.exit(0); });
|
|
42
|
+
process.on('SIGTERM', () => { close(); process.exit(0); });
|
|
43
|
+
await new Promise(() => { }); // hang
|
|
44
|
+
return;
|
|
45
|
+
}
|
|
46
|
+
try {
|
|
47
|
+
const svc = new GraphService(rootPath);
|
|
48
|
+
if (flags.rebuild) {
|
|
49
|
+
console.log(chalk.dim('Building knowledge graph...'));
|
|
50
|
+
svc.build();
|
|
51
|
+
}
|
|
52
|
+
const stats = svc.getStats();
|
|
53
|
+
svc.close();
|
|
54
|
+
console.log(chalk.bold('\nKnowledge Graph'));
|
|
55
|
+
console.log(chalk.dim('─'.repeat(40)));
|
|
56
|
+
console.log(` Nodes: ${chalk.cyan(stats.nodeCount)}`);
|
|
57
|
+
console.log(` Edges: ${chalk.cyan(stats.edgeCount)}`);
|
|
58
|
+
if (Object.keys(stats.typeCounts).length > 0) {
|
|
59
|
+
console.log(chalk.dim('\n By type:'));
|
|
60
|
+
for (const [type, count] of Object.entries(stats.typeCounts).sort((a, b) => b[1] - a[1])) {
|
|
61
|
+
console.log(` ${type.padEnd(14)} ${count}`);
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
console.log(chalk.dim('\n Commands:'));
|
|
65
|
+
console.log(chalk.dim(' edhindex kg serve — Interactive graph in browser'));
|
|
66
|
+
console.log(chalk.dim(' edhindex graph — Static SVG graph'));
|
|
67
|
+
console.log(chalk.dim('\n MCP tools:'));
|
|
68
|
+
console.log(chalk.dim(' get_graph — Full graph data'));
|
|
69
|
+
console.log(chalk.dim(' get_node — Node details + neighbors'));
|
|
70
|
+
console.log(chalk.dim(' search_graph — Search nodes by name'));
|
|
71
|
+
}
|
|
72
|
+
catch (e) {
|
|
73
|
+
console.error(chalk.red('Failed:'), e instanceof Error ? e.message : e);
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
//# sourceMappingURL=knowledge-graph.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"knowledge-graph.js","sourceRoot":"","sources":["../../../src/cli/commands/knowledge-graph.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACzE,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAE3C,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,QAAgB,EAAE,KAA8F;IAC9I,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC7C,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,6CAA6C,CAAC,CAAC,CAAC;QACzE,OAAO;IACT,CAAC;IAED,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAChB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,sBAAsB,CAAC,CAAC;QACpE,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,uBAAuB,OAAO,EAAE,CAAC,CAAC,CAAC;QAC3D,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC,CAAC;QAC/C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC,CAAC;QACvE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC,CAAC;QAC1E,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC,CAAC;QAC1E,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC,CAAC;QAClD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,OAAO,EAAE,CAAC,CAAC,CAAC;QAChD,OAAO;IACT,CAAC;IAED,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAChB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC,CAAC;QACnD,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,MAAM,gBAAgB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC3E,MAAM,GAAG,GAAG,oBAAoB,UAAU,EAAE,CAAC;QAC7C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,wBAAwB,GAAG,EAAE,CAAC,CAAC,CAAC;QACxD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC,CAAC;QAEtD,sBAAsB;QACtB,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC;QACzG,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YACrE,KAAK,CAAC,KAAK,EAAE,CAAC;QAChB,CAAC;QAAC,MAAM,CAAC,CAAC,+BAA+B,CAAC,CAAC;QAE3C,aAAa;QACb,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1D,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,MAAM,IAAI,OAAO,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC,OAAO;QACpC,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QAEvC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC,CAAC;YACtD,GAAG,CAAC,KAAK,EAAE,CAAC;QACd,CAAC;QAED,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC7B,GAAG,CAAC,KAAK,EAAE,CAAC;QAEZ,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAC7C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,YAAY,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,YAAY,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAEvD,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC;YACvC,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzF,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,0DAA0D,CAAC,CAAC,CAAC;QACnF,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC,CAAC;QAEvE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC;QACzC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC,CAAC;QACtE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,sDAAsD,CAAC,CAAC,CAAC;QAC/E,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC,CAAC;IAC7E,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1E,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../../src/cli/commands/models.ts"],"names":[],"mappings":"AAGA,wBAAgB,aAAa,SAe5B"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { MODEL_CONFIGS } from '../../config/index.js';
|
|
2
|
+
import chalk from 'chalk';
|
|
3
|
+
export function modelsCommand() {
|
|
4
|
+
console.log(chalk.bold('Available embedding models:\n'));
|
|
5
|
+
for (const [tier, cfg] of Object.entries(MODEL_CONFIGS)) {
|
|
6
|
+
const tag = tier === 'balanced' ? ' (default)' : '';
|
|
7
|
+
console.log(` ${chalk.cyan(tier)}${chalk.dim(tag)}`);
|
|
8
|
+
console.log(` Model: ${cfg.displayName}`);
|
|
9
|
+
console.log(` Size: ${cfg.size}`);
|
|
10
|
+
console.log(` Dims: ${cfg.dimensions}`);
|
|
11
|
+
console.log(` HF ID: ${cfg.modelId}`);
|
|
12
|
+
console.log('');
|
|
13
|
+
}
|
|
14
|
+
console.log(chalk.dim('Switch models with: edhindex config model <tier>'));
|
|
15
|
+
console.log(chalk.dim('Note: Changing models requires a full re-index.'));
|
|
16
|
+
}
|
|
17
|
+
//# sourceMappingURL=models.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"models.js","sourceRoot":"","sources":["../../../src/cli/commands/models.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,MAAM,UAAU,aAAa;IAC3B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC,CAAC;IAEzD,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;QACxD,MAAM,GAAG,GAAG,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC;QACpD,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtD,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;QAChD,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;QACzC,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;QAC/C,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAC5C,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC,CAAC;IAC3E,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC,CAAC;AAC5E,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"reset.d.ts","sourceRoot":"","sources":["../../../src/cli/commands/reset.ts"],"names":[],"mappings":"AAuBA,wBAAsB,YAAY,CAAC,QAAQ,EAAE,MAAM,iBAgBlD"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
import { join } from 'node:path';
|
|
2
|
+
import { existsSync, rmSync } from 'node:fs';
|
|
3
|
+
import { createInterface } from 'node:readline';
|
|
4
|
+
import chalk from 'chalk';
|
|
5
|
+
function isInteractive() {
|
|
6
|
+
return process.stdin.isTTY && process.stdout.isTTY;
|
|
7
|
+
}
|
|
8
|
+
function confirm(message) {
|
|
9
|
+
return new Promise((resolve) => {
|
|
10
|
+
if (!isInteractive()) {
|
|
11
|
+
resolve(false);
|
|
12
|
+
return;
|
|
13
|
+
}
|
|
14
|
+
const rl = createInterface({ input: process.stdin, output: process.stdout });
|
|
15
|
+
rl.question(`${message} (y/N) `, (answer) => {
|
|
16
|
+
rl.close();
|
|
17
|
+
resolve(answer.trim().toLowerCase() === 'y');
|
|
18
|
+
});
|
|
19
|
+
});
|
|
20
|
+
}
|
|
21
|
+
export async function resetCommand(rootPath) {
|
|
22
|
+
const indexDir = join(rootPath, '.edhindex');
|
|
23
|
+
if (!existsSync(indexDir)) {
|
|
24
|
+
console.log(chalk.yellow('No EDHIndex index found in this directory.'));
|
|
25
|
+
return;
|
|
26
|
+
}
|
|
27
|
+
const ok = await confirm(chalk.red('Delete the entire EDHIndex index (config + vectors + metadata)?'));
|
|
28
|
+
if (!ok) {
|
|
29
|
+
console.log(chalk.dim('Cancelled.'));
|
|
30
|
+
return;
|
|
31
|
+
}
|
|
32
|
+
rmSync(indexDir, { recursive: true, force: true });
|
|
33
|
+
console.log(chalk.green(`Deleted: ${indexDir}`));
|
|
34
|
+
}
|
|
35
|
+
//# sourceMappingURL=reset.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"reset.js","sourceRoot":"","sources":["../../../src/cli/commands/reset.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,SAAS,aAAa;IACpB,OAAO,OAAO,CAAC,KAAK,CAAC,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;AACrD,CAAC;AAED,SAAS,OAAO,CAAC,OAAe;IAC9B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YACrB,OAAO,CAAC,KAAK,CAAC,CAAC;YACf,OAAO;QACT,CAAC;QACD,MAAM,EAAE,GAAG,eAAe,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7E,EAAE,CAAC,QAAQ,CAAC,GAAG,OAAO,SAAS,EAAE,CAAC,MAAM,EAAE,EAAE;YAC1C,EAAE,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,KAAK,GAAG,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,QAAgB;IACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAE7C,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,4CAA4C,CAAC,CAAC,CAAC;QACxE,OAAO;IACT,CAAC;IAED,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,iEAAiE,CAAC,CAAC,CAAC;IACvG,IAAI,CAAC,EAAE,EAAE,CAAC;QACR,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC;QACrC,OAAO;IACT,CAAC;IAED,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,QAAQ,EAAE,CAAC,CAAC,CAAC;AACnD,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"search.d.ts","sourceRoot":"","sources":["../../../src/cli/commands/search.ts"],"names":[],"mappings":"AAUA,wBAAsB,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,iBAiCtF"}
|
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
import { join } from 'node:path';
|
|
2
|
+
import { loadConfig, MODEL_CONFIGS } from '../../config/index.js';
|
|
3
|
+
import { setLogLevel, LogLevel } from '../../logging.js';
|
|
4
|
+
import { MetadataStore } from '../../storage/metadata.js';
|
|
5
|
+
import { FTSStore } from '../../storage/fts.js';
|
|
6
|
+
import { TransformersEmbeddingProvider } from '../../embeddings/transformers.js';
|
|
7
|
+
import { LanceDBStore } from '../../vector/lancedb.js';
|
|
8
|
+
import { SearchEngine } from '../../retrieval/search.js';
|
|
9
|
+
import chalk from 'chalk';
|
|
10
|
+
export async function searchCommand(rootPath, query, maxResults) {
|
|
11
|
+
const indexDir = join(rootPath, '.edhindex');
|
|
12
|
+
setLogLevel(LogLevel.Silent);
|
|
13
|
+
const config = loadConfig(indexDir);
|
|
14
|
+
const modelCfg = MODEL_CONFIGS[config.model];
|
|
15
|
+
const metadataStore = new MetadataStore(indexDir);
|
|
16
|
+
const ftsStore = new FTSStore(indexDir);
|
|
17
|
+
const vectorStore = new LanceDBStore(indexDir, modelCfg.dimensions);
|
|
18
|
+
await vectorStore.init();
|
|
19
|
+
const embedder = new TransformersEmbeddingProvider(config.model);
|
|
20
|
+
await embedder.load();
|
|
21
|
+
const engine = new SearchEngine(ftsStore, metadataStore, vectorStore, embedder, null, false);
|
|
22
|
+
const results = await engine.search({ query, maxResults });
|
|
23
|
+
if (results.length === 0) {
|
|
24
|
+
console.log(chalk.yellow('No results found.'));
|
|
25
|
+
}
|
|
26
|
+
else {
|
|
27
|
+
for (const r of results) {
|
|
28
|
+
const tag = r.matchType === 'keyword' ? chalk.blue('KW') : r.matchType === 'vector' ? chalk.magenta('VEC') : chalk.green('HYB');
|
|
29
|
+
console.log(`${tag} ${chalk.cyan(`${r.file}:${r.startLine}-${r.endLine}`)} (${r.language})`);
|
|
30
|
+
if (r.symbol)
|
|
31
|
+
console.log(` ${chalk.dim('Symbol:')} ${r.symbol}${r.kind ? ` (${r.kind})` : ''}`);
|
|
32
|
+
console.log(` ${chalk.dim('Score:')} ${r.score.toFixed(4)}`);
|
|
33
|
+
console.log('');
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
metadataStore.close();
|
|
37
|
+
ftsStore.close();
|
|
38
|
+
await vectorStore.close();
|
|
39
|
+
}
|
|
40
|
+
//# sourceMappingURL=search.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"search.js","sourceRoot":"","sources":["../../../src/cli/commands/search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EAAE,6BAA6B,EAAE,MAAM,kCAAkC,CAAC;AACjF,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,QAAgB,EAAE,KAAa,EAAE,UAAkB;IACrF,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC7C,WAAW,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAE7B,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IACpC,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAE7C,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,QAAQ,CAAC,CAAC;IAClD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACxC,MAAM,WAAW,GAAG,IAAI,YAAY,CAAC,QAAQ,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;IACpE,MAAM,WAAW,CAAC,IAAI,EAAE,CAAC;IAEzB,MAAM,QAAQ,GAAG,IAAI,6BAA6B,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACjE,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAEtB,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,QAAQ,EAAE,aAAa,EAAE,WAAW,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;IAC7F,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;IAE3D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,CAAC;IACjD,CAAC;SAAM,CAAC;QACN,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACxB,MAAM,GAAG,GAAG,CAAC,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAChI,OAAO,CAAC,GAAG,CAAC,GAAG,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC;YAC7F,IAAI,CAAC,CAAC,MAAM;gBAAE,OAAO,CAAC,GAAG,CAAC,OAAO,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACpG,OAAO,CAAC,GAAG,CAAC,OAAO,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAChE,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAED,aAAa,CAAC,KAAK,EAAE,CAAC;IACtB,QAAQ,CAAC,KAAK,EAAE,CAAC;IACjB,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;AAC5B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"start.d.ts","sourceRoot":"","sources":["../../../src/cli/commands/start.ts"],"names":[],"mappings":"AAqBA,wBAAsB,YAAY,CAAC,QAAQ,EAAE,MAAM,iBAwQlD"}
|
|
@@ -0,0 +1,265 @@
|
|
|
1
|
+
import { join } from 'node:path';
|
|
2
|
+
import { existsSync, mkdirSync, writeFileSync } from 'node:fs';
|
|
3
|
+
import { loadConfig, MODEL_CONFIGS } from '../../config/index.js';
|
|
4
|
+
import { logger, setLogLevel, LogLevel } from '../../logging.js';
|
|
5
|
+
import { onProgress } from '../../progress.js';
|
|
6
|
+
import { loadVersion, saveVersion, getExpectedVersion, needsRebuild } from '../../version.js';
|
|
7
|
+
import { MetadataStore } from '../../storage/metadata.js';
|
|
8
|
+
import { FTSStore } from '../../storage/fts.js';
|
|
9
|
+
import { TransformersEmbeddingProvider, TransformersReranker } from '../../embeddings/transformers.js';
|
|
10
|
+
import { LanceDBStore } from '../../vector/lancedb.js';
|
|
11
|
+
import { Indexer } from '../../indexer/indexer.js';
|
|
12
|
+
import { SearchEngine } from '../../retrieval/search.js';
|
|
13
|
+
import { createMCPServer } from '../../mcp/server.js';
|
|
14
|
+
import { FileWatcher } from '../../watcher/index.js';
|
|
15
|
+
import { buildGraphData } from '../../graph/build.js';
|
|
16
|
+
import { generateSVG } from '../../graph/svg.js';
|
|
17
|
+
import { GraphBuilder } from '../../graph/builder.js';
|
|
18
|
+
import { GraphStore } from '../../graph/storage.js';
|
|
19
|
+
import cliProgress from 'cli-progress';
|
|
20
|
+
import chalk from 'chalk';
|
|
21
|
+
export async function startCommand(rootPath) {
|
|
22
|
+
const indexDir = join(rootPath, '.edhindex');
|
|
23
|
+
if (!existsSync(indexDir)) {
|
|
24
|
+
mkdirSync(indexDir, { recursive: true });
|
|
25
|
+
}
|
|
26
|
+
const config = loadConfig(indexDir);
|
|
27
|
+
const modelCfg = MODEL_CONFIGS[config.model];
|
|
28
|
+
const logLevelMap = {
|
|
29
|
+
silent: LogLevel.Silent,
|
|
30
|
+
info: LogLevel.Info,
|
|
31
|
+
verbose: LogLevel.Verbose,
|
|
32
|
+
debug: LogLevel.Debug,
|
|
33
|
+
};
|
|
34
|
+
setLogLevel(logLevelMap[config.logLevel] || LogLevel.Info);
|
|
35
|
+
console.error(chalk.bold('EDHIndex v1.0.0'));
|
|
36
|
+
console.error(`Workspace: ${rootPath}`);
|
|
37
|
+
console.error(`Model: ${modelCfg.displayName} (${modelCfg.dimensions}d)`);
|
|
38
|
+
console.error(`Languages: ${config.languages.join(', ')}`);
|
|
39
|
+
console.error('');
|
|
40
|
+
// Check index version
|
|
41
|
+
const version = loadVersion(indexDir);
|
|
42
|
+
if (needsRebuild(version, config.model)) {
|
|
43
|
+
if (version) {
|
|
44
|
+
console.error(chalk.yellow('Index version mismatch — full rebuild required.'));
|
|
45
|
+
console.error(chalk.dim(` Expected model: ${config.model}, existing: ${version.embeddingModel}`));
|
|
46
|
+
}
|
|
47
|
+
else {
|
|
48
|
+
console.error(chalk.dim('No existing index found — building from scratch.'));
|
|
49
|
+
}
|
|
50
|
+
}
|
|
51
|
+
else {
|
|
52
|
+
console.error(chalk.dim('Existing index is compatible — incremental update.'));
|
|
53
|
+
}
|
|
54
|
+
// Setup progress bar
|
|
55
|
+
let progressBar = null;
|
|
56
|
+
const unsub = onProgress((event) => {
|
|
57
|
+
if (progressBar) {
|
|
58
|
+
if (event.phase === 'scanning' && event.total) {
|
|
59
|
+
progressBar.setTotal(event.total);
|
|
60
|
+
progressBar.update(event.current || 0);
|
|
61
|
+
}
|
|
62
|
+
else if (event.phase === 'embedding' && event.percent !== undefined) {
|
|
63
|
+
progressBar.setTotal(100);
|
|
64
|
+
progressBar.update(event.percent);
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
});
|
|
68
|
+
// Initialize stores
|
|
69
|
+
console.error(chalk.dim('Initializing storage...'));
|
|
70
|
+
const metadataStore = new MetadataStore(indexDir);
|
|
71
|
+
const ftsStore = new FTSStore(indexDir);
|
|
72
|
+
// Initialize vector store
|
|
73
|
+
const vectorStore = new LanceDBStore(indexDir, modelCfg.dimensions);
|
|
74
|
+
await vectorStore.init();
|
|
75
|
+
// Initialize embedding provider
|
|
76
|
+
const embedder = new TransformersEmbeddingProvider(config.model);
|
|
77
|
+
console.error(chalk.dim(`Loading embedding model (${modelCfg.displayName})...`));
|
|
78
|
+
await embedder.load();
|
|
79
|
+
// Initialize reranker if enabled
|
|
80
|
+
let reranker = null;
|
|
81
|
+
if (config.rerank) {
|
|
82
|
+
console.error(chalk.dim('Loading reranker model...'));
|
|
83
|
+
reranker = new TransformersReranker();
|
|
84
|
+
await reranker.load();
|
|
85
|
+
}
|
|
86
|
+
// Initialize indexer
|
|
87
|
+
const indexer = new Indexer({
|
|
88
|
+
config,
|
|
89
|
+
rootPath,
|
|
90
|
+
indexDir,
|
|
91
|
+
onIndex: async (result) => {
|
|
92
|
+
for (const chunk of result.chunks) {
|
|
93
|
+
metadataStore.upsertChunk({
|
|
94
|
+
id: chunk.id,
|
|
95
|
+
file: chunk.file,
|
|
96
|
+
language: chunk.language,
|
|
97
|
+
symbol: chunk.symbol,
|
|
98
|
+
kind: chunk.kind,
|
|
99
|
+
parent: chunk.parent,
|
|
100
|
+
hash: chunk.hash,
|
|
101
|
+
git_commit: null,
|
|
102
|
+
start_line: chunk.startLine,
|
|
103
|
+
end_line: chunk.endLine,
|
|
104
|
+
imports: chunk.imports.join('\n'),
|
|
105
|
+
exports: chunk.exports.join('\n'),
|
|
106
|
+
embedding_model: embedder.name(),
|
|
107
|
+
last_indexed: new Date().toISOString(),
|
|
108
|
+
checksum: chunk.hash,
|
|
109
|
+
});
|
|
110
|
+
ftsStore.insertChunk(chunk.id, chunk.content, chunk.file, chunk.symbol, chunk.language);
|
|
111
|
+
}
|
|
112
|
+
if (result.chunks.length > 0) {
|
|
113
|
+
const texts = result.chunks.map(c => c.content);
|
|
114
|
+
const embeddings = await embedder.embed(texts);
|
|
115
|
+
const vectors = result.chunks.map((c, i) => ({
|
|
116
|
+
id: c.id,
|
|
117
|
+
values: embeddings[i],
|
|
118
|
+
metadata: {
|
|
119
|
+
file: c.file,
|
|
120
|
+
symbol: c.symbol,
|
|
121
|
+
kind: c.kind,
|
|
122
|
+
text: c.content,
|
|
123
|
+
language: c.language,
|
|
124
|
+
},
|
|
125
|
+
}));
|
|
126
|
+
await vectorStore.insert(vectors);
|
|
127
|
+
}
|
|
128
|
+
metadataStore.upsertFileHash(result.relativePath, result.hash);
|
|
129
|
+
},
|
|
130
|
+
onDelete: async (chunkIds) => {
|
|
131
|
+
for (const id of chunkIds) {
|
|
132
|
+
ftsStore.deleteChunk(id);
|
|
133
|
+
}
|
|
134
|
+
await vectorStore.delete(chunkIds);
|
|
135
|
+
},
|
|
136
|
+
});
|
|
137
|
+
await indexer.initialize();
|
|
138
|
+
// Determine if full or incremental
|
|
139
|
+
const existingFileHashes = metadataStore.getAllFileHashes();
|
|
140
|
+
const existingChunkHashes = metadataStore.getAllChunkHashes();
|
|
141
|
+
const doFullRebuild = needsRebuild(version, config.model) || existingFileHashes.size === 0;
|
|
142
|
+
// Index
|
|
143
|
+
console.error('');
|
|
144
|
+
progressBar = new cliProgress.SingleBar({
|
|
145
|
+
format: '{phase} | {bar} | {percentage}% | {value}/{total}',
|
|
146
|
+
barCompleteChar: '\u2588',
|
|
147
|
+
barIncompleteChar: '\u2591',
|
|
148
|
+
hideCursor: true,
|
|
149
|
+
});
|
|
150
|
+
if (doFullRebuild) {
|
|
151
|
+
progressBar.start(100, 0, { phase: 'Scanning' });
|
|
152
|
+
console.error(chalk.dim('Running full index...'));
|
|
153
|
+
const chunks = await indexer.runFullIndex();
|
|
154
|
+
progressBar.stop();
|
|
155
|
+
console.error(chalk.green(`Indexed ${chunks.length} chunks.`));
|
|
156
|
+
// Save version
|
|
157
|
+
const newVersion = getExpectedVersion(config.model);
|
|
158
|
+
saveVersion(indexDir, newVersion);
|
|
159
|
+
}
|
|
160
|
+
else {
|
|
161
|
+
progressBar.start(100, 0, { phase: 'Scanning' });
|
|
162
|
+
console.error(chalk.dim('Running incremental index...'));
|
|
163
|
+
const result = await indexer.runIncremental(existingFileHashes, existingChunkHashes);
|
|
164
|
+
progressBar.stop();
|
|
165
|
+
console.error(chalk.green(`Indexed ${result.chunks.length} chunks, removed ${result.removed.length} files.`));
|
|
166
|
+
// Update version
|
|
167
|
+
const existingVersion = version;
|
|
168
|
+
existingVersion.lastIndexed = new Date().toISOString();
|
|
169
|
+
saveVersion(indexDir, existingVersion);
|
|
170
|
+
}
|
|
171
|
+
unsub();
|
|
172
|
+
// Build search engine
|
|
173
|
+
const searchEngine = new SearchEngine(ftsStore, metadataStore, vectorStore, embedder, reranker, config.rerank);
|
|
174
|
+
// Start file watcher if enabled
|
|
175
|
+
let watcher = null;
|
|
176
|
+
if (config.watch) {
|
|
177
|
+
watcher = new FileWatcher(rootPath, async (filePath) => {
|
|
178
|
+
logger.info(`File changed: ${filePath}`);
|
|
179
|
+
const { scanFiles } = await import('../../indexer/file-utils.js');
|
|
180
|
+
const files = scanFiles(rootPath, config.languages);
|
|
181
|
+
const file = files.find(f => f.relativePath === filePath);
|
|
182
|
+
if (file) {
|
|
183
|
+
const oldChunks = metadataStore.getChunksForFile(filePath);
|
|
184
|
+
const oldIds = oldChunks.map(c => c.id);
|
|
185
|
+
if (oldIds.length > 0) {
|
|
186
|
+
for (const id of oldIds)
|
|
187
|
+
ftsStore.deleteChunk(id);
|
|
188
|
+
await vectorStore.delete(oldIds);
|
|
189
|
+
}
|
|
190
|
+
metadataStore.deleteChunksForFile(filePath);
|
|
191
|
+
metadataStore.deleteFileHash(filePath);
|
|
192
|
+
const chunks = await indexer.processFile(file.path, file.relativePath, file.language);
|
|
193
|
+
for (const chunk of chunks) {
|
|
194
|
+
metadataStore.upsertChunk({
|
|
195
|
+
id: chunk.id, file: chunk.file, language: chunk.language, symbol: chunk.symbol,
|
|
196
|
+
kind: chunk.kind, parent: chunk.parent, hash: chunk.hash, git_commit: null,
|
|
197
|
+
start_line: chunk.startLine, end_line: chunk.endLine,
|
|
198
|
+
imports: chunk.imports.join('\n'), exports: chunk.exports.join('\n'),
|
|
199
|
+
embedding_model: embedder.name(), last_indexed: new Date().toISOString(), checksum: chunk.hash,
|
|
200
|
+
});
|
|
201
|
+
ftsStore.insertChunk(chunk.id, chunk.content, chunk.file, chunk.symbol, chunk.language);
|
|
202
|
+
}
|
|
203
|
+
if (chunks.length > 0) {
|
|
204
|
+
const embeddings = await embedder.embed(chunks.map(c => c.content));
|
|
205
|
+
await vectorStore.insert(chunks.map((c, i) => ({
|
|
206
|
+
id: c.id, values: embeddings[i],
|
|
207
|
+
metadata: { file: c.file, symbol: c.symbol, kind: c.kind, text: c.content, language: c.language },
|
|
208
|
+
})));
|
|
209
|
+
}
|
|
210
|
+
metadataStore.upsertFileHash(filePath, file.hash);
|
|
211
|
+
logger.info(`Re-indexed: ${filePath}`);
|
|
212
|
+
}
|
|
213
|
+
});
|
|
214
|
+
await watcher.start();
|
|
215
|
+
console.error(chalk.green('File watcher started.'));
|
|
216
|
+
}
|
|
217
|
+
// Generate dependency graph SVG
|
|
218
|
+
try {
|
|
219
|
+
const { nodes, edges, fileCount, edgeCount } = buildGraphData(metadataStore);
|
|
220
|
+
if (fileCount > 0) {
|
|
221
|
+
const svg = generateSVG(rootPath, nodes, edges);
|
|
222
|
+
writeFileSync(join(rootPath, 'edhindex-graph.svg'), svg, 'utf-8');
|
|
223
|
+
logger.verbose(`SVG graph: ${fileCount} files, ${edgeCount} edges`);
|
|
224
|
+
}
|
|
225
|
+
}
|
|
226
|
+
catch (e) {
|
|
227
|
+
logger.debug('Failed to generate graph SVG:', e);
|
|
228
|
+
}
|
|
229
|
+
// Build knowledge graph
|
|
230
|
+
if (config.graphEnabled) {
|
|
231
|
+
try {
|
|
232
|
+
logger.info('Building knowledge graph...');
|
|
233
|
+
const graphStore = new GraphStore(indexDir);
|
|
234
|
+
const builder = new GraphBuilder(rootPath, metadataStore, graphStore);
|
|
235
|
+
builder.buildFull();
|
|
236
|
+
const { nodeCount, edgeCount } = { nodeCount: graphStore.getNodeCount(), edgeCount: graphStore.getEdgeCount() };
|
|
237
|
+
graphStore.close();
|
|
238
|
+
console.error(chalk.dim(`\n Knowledge Graph: ${nodeCount} nodes, ${edgeCount} edges`));
|
|
239
|
+
console.error(chalk.dim(` View in browser: edhindex kg --serve`));
|
|
240
|
+
console.error(chalk.dim(` View in VS Code: edhindex kg --write`));
|
|
241
|
+
}
|
|
242
|
+
catch (e) {
|
|
243
|
+
logger.debug('Failed to build knowledge graph:', e);
|
|
244
|
+
}
|
|
245
|
+
}
|
|
246
|
+
// Start MCP server
|
|
247
|
+
console.error(chalk.green('Starting MCP server on stdio...'));
|
|
248
|
+
const mcpServer = createMCPServer(searchEngine, rootPath);
|
|
249
|
+
const transport = await import('@modelcontextprotocol/sdk/server/stdio.js');
|
|
250
|
+
const stdioTransport = new transport.StdioServerTransport();
|
|
251
|
+
await mcpServer.connect(stdioTransport);
|
|
252
|
+
// Handle shutdown
|
|
253
|
+
const shutdown = async () => {
|
|
254
|
+
console.error(chalk.yellow('\nShutting down...'));
|
|
255
|
+
if (watcher)
|
|
256
|
+
await watcher.stop();
|
|
257
|
+
vectorStore.close();
|
|
258
|
+
metadataStore.close();
|
|
259
|
+
ftsStore.close();
|
|
260
|
+
process.exit(0);
|
|
261
|
+
};
|
|
262
|
+
process.on('SIGINT', shutdown);
|
|
263
|
+
process.on('SIGTERM', shutdown);
|
|
264
|
+
}
|
|
265
|
+
//# sourceMappingURL=start.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"start.js","sourceRoot":"","sources":["../../../src/cli/commands/start.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC/D,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAClE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,QAAQ,EAAe,MAAM,kBAAkB,CAAC;AAC9E,OAAO,EAAgB,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC7D,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,kBAAkB,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAC9F,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EAAE,6BAA6B,EAAE,oBAAoB,EAAE,MAAM,kCAAkC,CAAC;AACvG,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,WAAW,MAAM,cAAc,CAAC;AACvC,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,QAAgB;IACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC7C,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IACpC,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAE7C,MAAM,WAAW,GAA6B;QAC5C,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,KAAK,EAAE,QAAQ,CAAC,KAAK;KACtB,CAAC;IACF,WAAW,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC;IAE3D,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;IAC7C,OAAO,CAAC,KAAK,CAAC,cAAc,QAAQ,EAAE,CAAC,CAAC;IACxC,OAAO,CAAC,KAAK,CAAC,UAAU,QAAQ,CAAC,WAAW,KAAK,QAAQ,CAAC,UAAU,IAAI,CAAC,CAAC;IAC1E,OAAO,CAAC,KAAK,CAAC,cAAc,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC3D,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAElB,sBAAsB;IACtB,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IACtC,IAAI,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACxC,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,iDAAiD,CAAC,CAAC,CAAC;YAC/E,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,qBAAqB,MAAM,CAAC,KAAK,eAAe,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QACrG,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC,CAAC;IACjF,CAAC;IAED,qBAAqB;IACrB,IAAI,WAAW,GAAiC,IAAI,CAAC;IACrD,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,KAAK,EAAE,EAAE;QACjC,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,KAAK,CAAC,KAAK,KAAK,UAAU,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC9C,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAClC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;YACzC,CAAC;iBAAM,IAAI,KAAK,CAAC,KAAK,KAAK,WAAW,IAAI,KAAK,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;gBACtE,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;gBAC1B,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,oBAAoB;IACpB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC,CAAC;IACpD,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,QAAQ,CAAC,CAAC;IAClD,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAExC,0BAA0B;IAC1B,MAAM,WAAW,GAAG,IAAI,YAAY,CAAC,QAAQ,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;IACpE,MAAM,WAAW,CAAC,IAAI,EAAE,CAAC;IAEzB,gCAAgC;IAChC,MAAM,QAAQ,GAAG,IAAI,6BAA6B,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACjE,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,4BAA4B,QAAQ,CAAC,WAAW,MAAM,CAAC,CAAC,CAAC;IACjF,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAEtB,iCAAiC;IACjC,IAAI,QAAQ,GAAgC,IAAI,CAAC;IACjD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QAClB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC,CAAC;QACtD,QAAQ,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACtC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IACxB,CAAC;IAED,qBAAqB;IACrB,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC;QAC1B,MAAM;QACN,QAAQ;QACR,QAAQ;QACR,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACxB,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClC,aAAa,CAAC,WAAW,CAAC;oBACxB,EAAE,EAAE,KAAK,CAAC,EAAE;oBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,UAAU,EAAE,IAAI;oBAChB,UAAU,EAAE,KAAK,CAAC,SAAS;oBAC3B,QAAQ,EAAE,KAAK,CAAC,OAAO;oBACvB,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;oBACjC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;oBACjC,eAAe,EAAE,QAAQ,CAAC,IAAI,EAAE;oBAChC,YAAY,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACtC,QAAQ,EAAE,KAAK,CAAC,IAAI;iBACrB,CAAC,CAAC;gBAEH,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC1F,CAAC;YAED,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAChD,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC/C,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC3C,EAAE,EAAE,CAAC,CAAC,EAAE;oBACR,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;oBACrB,QAAQ,EAAE;wBACR,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,MAAM,EAAE,CAAC,CAAC,MAAM;wBAChB,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,IAAI,EAAE,CAAC,CAAC,OAAO;wBACf,QAAQ,EAAE,CAAC,CAAC,QAAQ;qBACrB;iBACF,CAAC,CAAC,CAAC;gBACJ,MAAM,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACpC,CAAC;YAED,aAAa,CAAC,cAAc,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QACjE,CAAC;QACD,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;YAC3B,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;gBAC1B,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAC3B,CAAC;YACD,MAAM,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;IAE3B,mCAAmC;IACnC,MAAM,kBAAkB,GAAG,aAAa,CAAC,gBAAgB,EAAE,CAAC;IAC5D,MAAM,mBAAmB,GAAG,aAAa,CAAC,iBAAiB,EAAE,CAAC;IAC9D,MAAM,aAAa,GAAG,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,kBAAkB,CAAC,IAAI,KAAK,CAAC,CAAC;IAE3F,QAAQ;IACR,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAClB,WAAW,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC;QACtC,MAAM,EAAE,mDAAmD;QAC3D,eAAe,EAAE,QAAQ;QACzB,iBAAiB,EAAE,QAAQ;QAC3B,UAAU,EAAE,IAAI;KACjB,CAAC,CAAC;IAEH,IAAI,aAAa,EAAE,CAAC;QAClB,WAAW,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QACjD,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,EAAE,CAAC;QAC5C,WAAW,CAAC,IAAI,EAAE,CAAC;QACnB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,MAAM,CAAC,MAAM,UAAU,CAAC,CAAC,CAAC;QAE/D,eAAe;QACf,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACpD,WAAW,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IACpC,CAAC;SAAM,CAAC;QACN,WAAW,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QACjD,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC,CAAC;QACzD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,kBAAkB,EAAE,mBAAmB,CAAC,CAAC;QACrF,WAAW,CAAC,IAAI,EAAE,CAAC;QACnB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,MAAM,CAAC,MAAM,CAAC,MAAM,oBAAoB,MAAM,CAAC,OAAO,CAAC,MAAM,SAAS,CAAC,CAAC,CAAC;QAE9G,iBAAiB;QACjB,MAAM,eAAe,GAAG,OAAQ,CAAC;QACjC,eAAe,CAAC,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACvD,WAAW,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,EAAE,CAAC;IAER,sBAAsB;IACtB,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,QAAQ,EAAE,aAAa,EAAE,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAE/G,gCAAgC;IAChC,IAAI,OAAO,GAAuB,IAAI,CAAC;IACvC,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,OAAO,GAAG,IAAI,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;YACrD,MAAM,CAAC,IAAI,CAAC,iBAAiB,QAAQ,EAAE,CAAC,CAAC;YACzC,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,6BAA6B,CAAC,CAAC;YAClE,MAAM,KAAK,GAAG,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;YACpD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC;YAC1D,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,SAAS,GAAG,aAAa,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;gBAC3D,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACxC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtB,KAAK,MAAM,EAAE,IAAI,MAAM;wBAAE,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;oBAClD,MAAM,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACnC,CAAC;gBACD,aAAa,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;gBAC5C,aAAa,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;gBAEvC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACtF,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC3B,aAAa,CAAC,WAAW,CAAC;wBACxB,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM;wBAC9E,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,UAAU,EAAE,IAAI;wBAC1E,UAAU,EAAE,KAAK,CAAC,SAAS,EAAE,QAAQ,EAAE,KAAK,CAAC,OAAO;wBACpD,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;wBACpE,eAAe,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,YAAY,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,IAAI;qBAC/F,CAAC,CAAC;oBACH,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;gBAC1F,CAAC;gBACD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtB,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;oBACpE,MAAM,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC7C,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;wBAC/B,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE;qBAClG,CAAC,CAAC,CAAC,CAAC;gBACP,CAAC;gBACD,aAAa,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;gBAClD,MAAM,CAAC,IAAI,CAAC,eAAe,QAAQ,EAAE,CAAC,CAAC;YACzC,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;QACtB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,gCAAgC;IAChC,IAAI,CAAC;QACH,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,cAAc,CAAC,aAAa,CAAC,CAAC;QAC7E,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,MAAM,GAAG,GAAG,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;YAChD,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,oBAAoB,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;YAClE,MAAM,CAAC,OAAO,CAAC,cAAc,SAAS,WAAW,SAAS,QAAQ,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,wBAAwB;IACxB,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QACxB,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC3C,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC5C,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,QAAQ,EAAE,aAAa,EAAE,UAAU,CAAC,CAAC;YACtE,OAAO,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,SAAS,EAAE,UAAU,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,UAAU,CAAC,YAAY,EAAE,EAAE,CAAC;YAChH,UAAU,CAAC,KAAK,EAAE,CAAC;YACnB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,wBAAwB,SAAS,WAAW,SAAS,QAAQ,CAAC,CAAC,CAAC;YACxF,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC,CAAC;YACpE,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC,CAAC;QACtE,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,mBAAmB;IACnB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC,CAAC;IAC9D,MAAM,SAAS,GAAG,eAAe,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IAE1D,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,2CAA2C,CAAC,CAAC;IAC5E,MAAM,cAAc,GAAG,IAAI,SAAS,CAAC,oBAAoB,EAAE,CAAC;IAC5D,MAAM,SAAS,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IAExC,kBAAkB;IAClB,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;QAC1B,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,CAAC;QAClD,IAAI,OAAO;YAAE,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QAClC,WAAW,CAAC,KAAK,EAAE,CAAC;QACpB,aAAa,CAAC,KAAK,EAAE,CAAC;QACtB,QAAQ,CAAC,KAAK,EAAE,CAAC;QACjB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC;IAEF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC/B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;AAClC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"status.d.ts","sourceRoot":"","sources":["../../../src/cli/commands/status.ts"],"names":[],"mappings":"AASA,wBAAsB,aAAa,CAAC,QAAQ,EAAE,MAAM,iBA4CnD"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
import { join } from 'node:path';
|
|
2
|
+
import { existsSync } from 'node:fs';
|
|
3
|
+
import { loadConfig, MODEL_CONFIGS } from '../../config/index.js';
|
|
4
|
+
import { loadVersion } from '../../version.js';
|
|
5
|
+
import { MetadataStore } from '../../storage/metadata.js';
|
|
6
|
+
import { LanceDBStore } from '../../vector/lancedb.js';
|
|
7
|
+
import chalk from 'chalk';
|
|
8
|
+
export async function statusCommand(rootPath) {
|
|
9
|
+
const indexDir = join(rootPath, '.edhindex');
|
|
10
|
+
if (!existsSync(indexDir)) {
|
|
11
|
+
console.log(chalk.yellow('No .edhindex directory found. Run "edhindex init" first.'));
|
|
12
|
+
return;
|
|
13
|
+
}
|
|
14
|
+
const config = loadConfig(indexDir);
|
|
15
|
+
const version = loadVersion(indexDir);
|
|
16
|
+
console.log(chalk.bold('EDHIndex Status'));
|
|
17
|
+
console.log('');
|
|
18
|
+
if (version) {
|
|
19
|
+
console.log(` Index: ${chalk.green('Present')}`);
|
|
20
|
+
console.log(` Schema: v${version.schema}`);
|
|
21
|
+
console.log(` Model tier: ${version.embeddingModel}`);
|
|
22
|
+
console.log(` Created: ${version.createdAt}`);
|
|
23
|
+
console.log(` Last indexed: ${version.lastIndexed}`);
|
|
24
|
+
}
|
|
25
|
+
else {
|
|
26
|
+
console.log(` Index: ${chalk.yellow('No version file')}`);
|
|
27
|
+
}
|
|
28
|
+
console.log(` Config: ${chalk.cyan(JSON.stringify(config))}`);
|
|
29
|
+
const metadataStore = new MetadataStore(indexDir);
|
|
30
|
+
const chunkCount = metadataStore.getChunkCount();
|
|
31
|
+
const fileCount = metadataStore.getFileCount();
|
|
32
|
+
metadataStore.close();
|
|
33
|
+
console.log(` Chunks: ${chunkCount}`);
|
|
34
|
+
console.log(` Files: ${fileCount}`);
|
|
35
|
+
console.log(` Languages: ${config.languages.join(', ')}`);
|
|
36
|
+
try {
|
|
37
|
+
const vectorStore = new LanceDBStore(indexDir, MODEL_CONFIGS[config.model].dimensions);
|
|
38
|
+
await vectorStore.init();
|
|
39
|
+
const vecCount = await vectorStore.count();
|
|
40
|
+
console.log(` Vectors: ${vecCount}`);
|
|
41
|
+
await vectorStore.close();
|
|
42
|
+
}
|
|
43
|
+
catch {
|
|
44
|
+
console.log(` Vectors: ${chalk.yellow('unavailable')}`);
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
//# sourceMappingURL=status.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"status.js","sourceRoot":"","sources":["../../../src/cli/commands/status.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE1D,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,QAAgB;IAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAE7C,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,0DAA0D,CAAC,CAAC,CAAC;QACtF,OAAO;IACT,CAAC;IAED,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IACpC,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IAEtC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;IAC3C,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAEhB,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,CAAC,GAAG,CAAC,mBAAmB,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACzD,OAAO,CAAC,GAAG,CAAC,oBAAoB,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAClD,OAAO,CAAC,GAAG,CAAC,mBAAmB,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;QACzD,OAAO,CAAC,GAAG,CAAC,mBAAmB,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;QACpD,OAAO,CAAC,GAAG,CAAC,mBAAmB,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;IACxD,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,mBAAmB,KAAK,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;IACpE,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,mBAAmB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC;IAErE,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,QAAQ,CAAC,CAAC;IAClD,MAAM,UAAU,GAAG,aAAa,CAAC,aAAa,EAAE,CAAC;IACjD,MAAM,SAAS,GAAG,aAAa,CAAC,YAAY,EAAE,CAAC;IAC/C,aAAa,CAAC,KAAK,EAAE,CAAC;IAEtB,OAAO,CAAC,GAAG,CAAC,mBAAmB,UAAU,EAAE,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,mBAAmB,SAAS,EAAE,CAAC,CAAC;IAC5C,OAAO,CAAC,GAAG,CAAC,mBAAmB,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAE9D,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,IAAI,YAAY,CAAC,QAAQ,EAAE,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,CAAC,CAAC;QACvF,MAAM,WAAW,CAAC,IAAI,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,mBAAmB,QAAQ,EAAE,CAAC,CAAC;QAC3C,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,GAAG,CAAC,mBAAmB,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;IAChE,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":""}
|