edhindex 1.0.0 → 1.0.1
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 +101 -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 +9 -3
- 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 @@
|
|
|
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":""}
|
|
@@ -0,0 +1,121 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import { Command } from 'commander';
|
|
3
|
+
import { join } from 'node:path';
|
|
4
|
+
import { initCommand } from './commands/init.js';
|
|
5
|
+
import { startCommand } from './commands/start.js';
|
|
6
|
+
import { indexCommand } from './commands/index_cmd.js';
|
|
7
|
+
import { searchCommand } from './commands/search.js';
|
|
8
|
+
import { statusCommand } from './commands/status.js';
|
|
9
|
+
import { configCommand } from './commands/config_cmd.js';
|
|
10
|
+
import { modelsCommand } from './commands/models.js';
|
|
11
|
+
import { doctorCommand } from './commands/doctor_cmd.js';
|
|
12
|
+
import { resetCommand } from './commands/reset.js';
|
|
13
|
+
import { graphCommand } from './commands/graph.js';
|
|
14
|
+
import { kgCommand } from './commands/knowledge-graph.js';
|
|
15
|
+
import { setLogLevel, LogLevel } from '../logging.js';
|
|
16
|
+
const program = new Command();
|
|
17
|
+
program
|
|
18
|
+
.name('edhindex')
|
|
19
|
+
.description('Local-first hybrid code search engine with MCP')
|
|
20
|
+
.version('1.0.0')
|
|
21
|
+
.option('-v, --verbose', 'Enable verbose logging')
|
|
22
|
+
.option('--debug', 'Enable debug logging')
|
|
23
|
+
.hook('preAction', (thisCommand) => {
|
|
24
|
+
if (thisCommand.opts().debug)
|
|
25
|
+
setLogLevel(LogLevel.Debug);
|
|
26
|
+
else if (thisCommand.opts().verbose)
|
|
27
|
+
setLogLevel(LogLevel.Verbose);
|
|
28
|
+
});
|
|
29
|
+
program
|
|
30
|
+
.command('init')
|
|
31
|
+
.description('Initialize EDHIndex in the current directory')
|
|
32
|
+
.action(async () => {
|
|
33
|
+
const rootPath = process.cwd();
|
|
34
|
+
await initCommand(rootPath);
|
|
35
|
+
});
|
|
36
|
+
program
|
|
37
|
+
.command('start')
|
|
38
|
+
.description('Build/update index and start the MCP server')
|
|
39
|
+
.action(async () => {
|
|
40
|
+
const rootPath = process.cwd();
|
|
41
|
+
await startCommand(rootPath);
|
|
42
|
+
});
|
|
43
|
+
program
|
|
44
|
+
.command('index')
|
|
45
|
+
.description('Build or update the search index')
|
|
46
|
+
.action(async () => {
|
|
47
|
+
const rootPath = process.cwd();
|
|
48
|
+
await indexCommand(rootPath);
|
|
49
|
+
});
|
|
50
|
+
program
|
|
51
|
+
.command('search')
|
|
52
|
+
.description('Search the indexed codebase from the CLI')
|
|
53
|
+
.argument('<query>', 'Search query')
|
|
54
|
+
.option('-n, --max-results <number>', 'Max results', '10')
|
|
55
|
+
.action(async (query, opts) => {
|
|
56
|
+
const rootPath = process.cwd();
|
|
57
|
+
await searchCommand(rootPath, query, parseInt(opts.maxResults, 10) || 10);
|
|
58
|
+
});
|
|
59
|
+
program
|
|
60
|
+
.command('status')
|
|
61
|
+
.description('Show index status')
|
|
62
|
+
.action(async () => {
|
|
63
|
+
const rootPath = process.cwd();
|
|
64
|
+
await statusCommand(rootPath);
|
|
65
|
+
});
|
|
66
|
+
program
|
|
67
|
+
.command('config')
|
|
68
|
+
.description('View or update configuration')
|
|
69
|
+
.argument('[key]', 'Config key')
|
|
70
|
+
.argument('[value]', 'Config value')
|
|
71
|
+
.action((key, value) => {
|
|
72
|
+
const rootPath = process.cwd();
|
|
73
|
+
const changes = {};
|
|
74
|
+
if (key && value !== undefined) {
|
|
75
|
+
changes[key] = value;
|
|
76
|
+
}
|
|
77
|
+
configCommand(rootPath, changes);
|
|
78
|
+
});
|
|
79
|
+
program
|
|
80
|
+
.command('models')
|
|
81
|
+
.description('List available embedding models')
|
|
82
|
+
.action(() => {
|
|
83
|
+
modelsCommand();
|
|
84
|
+
});
|
|
85
|
+
program
|
|
86
|
+
.command('doctor')
|
|
87
|
+
.description('Run diagnostics')
|
|
88
|
+
.action(async () => {
|
|
89
|
+
const rootPath = process.cwd();
|
|
90
|
+
const indexDir = join(rootPath, '.edhindex');
|
|
91
|
+
await doctorCommand(rootPath, indexDir);
|
|
92
|
+
});
|
|
93
|
+
program
|
|
94
|
+
.command('reset')
|
|
95
|
+
.description('Delete the index and all stored data')
|
|
96
|
+
.action(async () => {
|
|
97
|
+
const rootPath = process.cwd();
|
|
98
|
+
await resetCommand(rootPath);
|
|
99
|
+
});
|
|
100
|
+
program
|
|
101
|
+
.command('graph')
|
|
102
|
+
.description('Generate a dependency graph (SVG)')
|
|
103
|
+
.option('-o, --output <path>', 'Output file path')
|
|
104
|
+
.action(async (opts) => {
|
|
105
|
+
const rootPath = process.cwd();
|
|
106
|
+
await graphCommand(rootPath, opts.output);
|
|
107
|
+
});
|
|
108
|
+
program
|
|
109
|
+
.command('kg')
|
|
110
|
+
.description('Knowledge graph — build, stats, serve, write, and query')
|
|
111
|
+
.option('-r, --rebuild', 'Rebuild graph from index')
|
|
112
|
+
.option('-s, --stats', 'Show graph statistics')
|
|
113
|
+
.option('--serve', 'Start interactive graph viewer (opens browser)')
|
|
114
|
+
.option('-w, --write', 'Write graph viewer to .edhindex/knowledge-graph.html')
|
|
115
|
+
.option('-p, --port <number>', 'Port for graph viewer (default: random)')
|
|
116
|
+
.action(async (opts) => {
|
|
117
|
+
const rootPath = process.cwd();
|
|
118
|
+
await kgCommand(rootPath, opts);
|
|
119
|
+
});
|
|
120
|
+
program.parse(process.argv);
|
|
121
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACzD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAC1D,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAEtD,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAE9B,OAAO;KACJ,IAAI,CAAC,UAAU,CAAC;KAChB,WAAW,CAAC,gDAAgD,CAAC;KAC7D,OAAO,CAAC,OAAO,CAAC;KAChB,MAAM,CAAC,eAAe,EAAE,wBAAwB,CAAC;KACjD,MAAM,CAAC,SAAS,EAAE,sBAAsB,CAAC;KACzC,IAAI,CAAC,WAAW,EAAE,CAAC,WAAW,EAAE,EAAE;IACjC,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC,KAAK;QAAE,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;SACrD,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC,OAAO;QAAE,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AACrE,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,MAAM,CAAC;KACf,WAAW,CAAC,8CAA8C,CAAC;KAC3D,MAAM,CAAC,KAAK,IAAI,EAAE;IACjB,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,WAAW,CAAC,QAAQ,CAAC,CAAC;AAC9B,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,OAAO,CAAC;KAChB,WAAW,CAAC,6CAA6C,CAAC;KAC1D,MAAM,CAAC,KAAK,IAAI,EAAE;IACjB,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,YAAY,CAAC,QAAQ,CAAC,CAAC;AAC/B,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,OAAO,CAAC;KAChB,WAAW,CAAC,kCAAkC,CAAC;KAC/C,MAAM,CAAC,KAAK,IAAI,EAAE;IACjB,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,YAAY,CAAC,QAAQ,CAAC,CAAC;AAC/B,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,QAAQ,CAAC;KACjB,WAAW,CAAC,0CAA0C,CAAC;KACvD,QAAQ,CAAC,SAAS,EAAE,cAAc,CAAC;KACnC,MAAM,CAAC,4BAA4B,EAAE,aAAa,EAAE,IAAI,CAAC;KACzD,MAAM,CAAC,KAAK,EAAE,KAAa,EAAE,IAA4B,EAAE,EAAE;IAC5D,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,aAAa,CAAC,QAAQ,EAAE,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;AAC5E,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,QAAQ,CAAC;KACjB,WAAW,CAAC,mBAAmB,CAAC;KAChC,MAAM,CAAC,KAAK,IAAI,EAAE;IACjB,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,aAAa,CAAC,QAAQ,CAAC,CAAC;AAChC,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,QAAQ,CAAC;KACjB,WAAW,CAAC,8BAA8B,CAAC;KAC3C,QAAQ,CAAC,OAAO,EAAE,YAAY,CAAC;KAC/B,QAAQ,CAAC,SAAS,EAAE,cAAc,CAAC;KACnC,MAAM,CAAC,CAAC,GAAY,EAAE,KAAc,EAAE,EAAE;IACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,IAAI,GAAG,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QAC/B,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IACvB,CAAC;IACD,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;AACnC,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,QAAQ,CAAC;KACjB,WAAW,CAAC,iCAAiC,CAAC;KAC9C,MAAM,CAAC,GAAG,EAAE;IACX,aAAa,EAAE,CAAC;AAClB,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,QAAQ,CAAC;KACjB,WAAW,CAAC,iBAAiB,CAAC;KAC9B,MAAM,CAAC,KAAK,IAAI,EAAE;IACjB,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC7C,MAAM,aAAa,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,OAAO,CAAC;KAChB,WAAW,CAAC,sCAAsC,CAAC;KACnD,MAAM,CAAC,KAAK,IAAI,EAAE;IACjB,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,YAAY,CAAC,QAAQ,CAAC,CAAC;AAC/B,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,OAAO,CAAC;KAChB,WAAW,CAAC,mCAAmC,CAAC;KAChD,MAAM,CAAC,qBAAqB,EAAE,kBAAkB,CAAC;KACjD,MAAM,CAAC,KAAK,EAAE,IAAyB,EAAE,EAAE;IAC1C,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;AAC5C,CAAC,CAAC,CAAC;AAEL,OAAO;KACJ,OAAO,CAAC,IAAI,CAAC;KACb,WAAW,CAAC,yDAAyD,CAAC;KACtE,MAAM,CAAC,eAAe,EAAE,0BAA0B,CAAC;KACnD,MAAM,CAAC,aAAa,EAAE,uBAAuB,CAAC;KAC9C,MAAM,CAAC,SAAS,EAAE,gDAAgD,CAAC;KACnE,MAAM,CAAC,aAAa,EAAE,sDAAsD,CAAC;KAC7E,MAAM,CAAC,qBAAqB,EAAE,yCAAyC,CAAC;KACxE,MAAM,CAAC,KAAK,EAAE,IAA6F,EAAE,EAAE;IAC9G,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC/B,MAAM,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;AAClC,CAAC,CAAC,CAAC;AAEL,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
export type ModelTier = 'light' | 'balanced' | 'max';
|
|
2
|
+
export type LogLevelConfig = 'silent' | 'info' | 'verbose' | 'debug';
|
|
3
|
+
export type GraphLayoutConfig = 'force' | 'hierarchical' | 'circular' | 'grid';
|
|
4
|
+
export interface Config {
|
|
5
|
+
model: ModelTier;
|
|
6
|
+
languages: string[];
|
|
7
|
+
watch: boolean;
|
|
8
|
+
rerank: boolean;
|
|
9
|
+
maxResults: number;
|
|
10
|
+
logLevel: LogLevelConfig;
|
|
11
|
+
graphEnabled: boolean;
|
|
12
|
+
graphLayout: GraphLayoutConfig;
|
|
13
|
+
graphAnimations: boolean;
|
|
14
|
+
graphShowFolders: boolean;
|
|
15
|
+
graphShowFiles: boolean;
|
|
16
|
+
graphShowFunctions: boolean;
|
|
17
|
+
}
|
|
18
|
+
export declare function defaultConfig(): Config;
|
|
19
|
+
export declare function loadConfig(indexDir: string): Config;
|
|
20
|
+
export declare function saveConfig(indexDir: string, config: Config): void;
|
|
21
|
+
export declare const MODEL_CONFIGS: Record<ModelTier, {
|
|
22
|
+
modelId: string;
|
|
23
|
+
dimensions: number;
|
|
24
|
+
displayName: string;
|
|
25
|
+
size: string;
|
|
26
|
+
}>;
|
|
27
|
+
export declare const RERANKER_MODEL = "Xenova/ms-marco-MiniLM-L-6-v2";
|
|
28
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/config/index.ts"],"names":[],"mappings":"AAIA,MAAM,MAAM,SAAS,GAAG,OAAO,GAAG,UAAU,GAAG,KAAK,CAAC;AACrD,MAAM,MAAM,cAAc,GAAG,QAAQ,GAAG,MAAM,GAAG,SAAS,GAAG,OAAO,CAAC;AACrE,MAAM,MAAM,iBAAiB,GAAG,OAAO,GAAG,cAAc,GAAG,UAAU,GAAG,MAAM,CAAC;AAE/E,MAAM,WAAW,MAAM;IACrB,KAAK,EAAE,SAAS,CAAC;IACjB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,OAAO,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,cAAc,CAAC;IACzB,YAAY,EAAE,OAAO,CAAC;IACtB,WAAW,EAAE,iBAAiB,CAAC;IAC/B,eAAe,EAAE,OAAO,CAAC;IACzB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,cAAc,EAAE,OAAO,CAAC;IACxB,kBAAkB,EAAE,OAAO,CAAC;CAC7B;AAiBD,wBAAgB,aAAa,IAAI,MAAM,CAEtC;AAED,wBAAgB,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAenD;AAED,wBAAgB,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,QAK1D;AAED,eAAO,MAAM,aAAa,EAAE,MAAM,CAAC,SAAS,EAAE;IAAE,OAAO,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CAmBvH,CAAC;AAEF,eAAO,MAAM,cAAc,kCAAkC,CAAC"}
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'node:fs';
|
|
2
|
+
import { join } from 'node:path';
|
|
3
|
+
import { logger } from '../logging.js';
|
|
4
|
+
const DEFAULT_CONFIG = {
|
|
5
|
+
model: 'balanced',
|
|
6
|
+
languages: ['ts', 'js', 'go', 'py'],
|
|
7
|
+
watch: true,
|
|
8
|
+
rerank: true,
|
|
9
|
+
maxResults: 10,
|
|
10
|
+
logLevel: 'info',
|
|
11
|
+
graphEnabled: true,
|
|
12
|
+
graphLayout: 'force',
|
|
13
|
+
graphAnimations: true,
|
|
14
|
+
graphShowFolders: true,
|
|
15
|
+
graphShowFiles: true,
|
|
16
|
+
graphShowFunctions: true,
|
|
17
|
+
};
|
|
18
|
+
export function defaultConfig() {
|
|
19
|
+
return { ...DEFAULT_CONFIG };
|
|
20
|
+
}
|
|
21
|
+
export function loadConfig(indexDir) {
|
|
22
|
+
const path = join(indexDir, 'config.json');
|
|
23
|
+
if (!existsSync(path)) {
|
|
24
|
+
const cfg = defaultConfig();
|
|
25
|
+
saveConfig(indexDir, cfg);
|
|
26
|
+
return cfg;
|
|
27
|
+
}
|
|
28
|
+
try {
|
|
29
|
+
const raw = readFileSync(path, 'utf-8');
|
|
30
|
+
const parsed = JSON.parse(raw);
|
|
31
|
+
return { ...defaultConfig(), ...parsed };
|
|
32
|
+
}
|
|
33
|
+
catch (e) {
|
|
34
|
+
logger.error('Failed to load config, using defaults:', e);
|
|
35
|
+
return defaultConfig();
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
export function saveConfig(indexDir, config) {
|
|
39
|
+
if (!existsSync(indexDir)) {
|
|
40
|
+
mkdirSync(indexDir, { recursive: true });
|
|
41
|
+
}
|
|
42
|
+
writeFileSync(join(indexDir, 'config.json'), JSON.stringify(config, null, 2), 'utf-8');
|
|
43
|
+
}
|
|
44
|
+
export const MODEL_CONFIGS = {
|
|
45
|
+
light: {
|
|
46
|
+
modelId: 'Xenova/all-MiniLM-L6-v2',
|
|
47
|
+
dimensions: 384,
|
|
48
|
+
displayName: 'all-MiniLM-L6-v2',
|
|
49
|
+
size: '~46MB',
|
|
50
|
+
},
|
|
51
|
+
balanced: {
|
|
52
|
+
modelId: 'Xenova/bge-base-en-v1.5',
|
|
53
|
+
dimensions: 768,
|
|
54
|
+
displayName: 'bge-base-en-v1.5',
|
|
55
|
+
size: '~109MB',
|
|
56
|
+
},
|
|
57
|
+
max: {
|
|
58
|
+
modelId: 'Xenova/bge-m3',
|
|
59
|
+
dimensions: 1024,
|
|
60
|
+
displayName: 'BGE-M3',
|
|
61
|
+
size: '~1.1GB',
|
|
62
|
+
},
|
|
63
|
+
};
|
|
64
|
+
export const RERANKER_MODEL = 'Xenova/ms-marco-MiniLM-L-6-v2';
|
|
65
|
+
//# sourceMappingURL=index.js.map
|