codevault 1.6.1 → 1.7.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/dist/chunking/token-counter.d.ts.map +1 -1
- package/dist/chunking/token-counter.js +2 -31
- package/dist/chunking/token-counter.js.map +1 -1
- package/dist/cli.js +3 -5
- package/dist/cli.js.map +1 -1
- package/dist/codemap/io.d.ts +1 -0
- package/dist/codemap/io.d.ts.map +1 -1
- package/dist/codemap/io.js +16 -0
- package/dist/codemap/io.js.map +1 -1
- package/dist/config/apply-env.d.ts +8 -0
- package/dist/config/apply-env.d.ts.map +1 -1
- package/dist/config/apply-env.js +54 -56
- package/dist/config/apply-env.js.map +1 -1
- package/dist/config/resolver.d.ts +30 -0
- package/dist/config/resolver.d.ts.map +1 -0
- package/dist/config/resolver.js +29 -0
- package/dist/config/resolver.js.map +1 -0
- package/dist/core/IndexerEngine.d.ts +24 -0
- package/dist/core/IndexerEngine.d.ts.map +1 -0
- package/dist/core/IndexerEngine.js +372 -0
- package/dist/core/IndexerEngine.js.map +1 -0
- package/dist/core/SearchService.d.ts +25 -0
- package/dist/core/SearchService.d.ts.map +1 -0
- package/dist/core/SearchService.js +455 -0
- package/dist/core/SearchService.js.map +1 -0
- package/dist/core/batch-indexer.d.ts.map +1 -1
- package/dist/core/batch-indexer.js +7 -4
- package/dist/core/batch-indexer.js.map +1 -1
- package/dist/core/indexer.d.ts +1 -1
- package/dist/core/indexer.d.ts.map +1 -1
- package/dist/core/indexer.js +4 -598
- package/dist/core/indexer.js.map +1 -1
- package/dist/core/indexing/chunk-pipeline.d.ts +39 -0
- package/dist/core/indexing/chunk-pipeline.d.ts.map +1 -0
- package/dist/core/indexing/chunk-pipeline.js +210 -0
- package/dist/core/indexing/chunk-pipeline.js.map +1 -0
- package/dist/core/indexing/file-scanner.d.ts +11 -0
- package/dist/core/indexing/file-scanner.d.ts.map +1 -0
- package/dist/core/indexing/file-scanner.js +49 -0
- package/dist/core/indexing/file-scanner.js.map +1 -0
- package/dist/core/search.d.ts +1 -1
- package/dist/core/search.d.ts.map +1 -1
- package/dist/core/search.js +11 -540
- package/dist/core/search.js.map +1 -1
- package/dist/database/db.d.ts +2 -3
- package/dist/database/db.d.ts.map +1 -1
- package/dist/database/db.js +0 -40
- package/dist/database/db.js.map +1 -1
- package/dist/indexer/watch.d.ts.map +1 -1
- package/dist/indexer/watch.js +3 -1
- package/dist/indexer/watch.js.map +1 -1
- package/dist/mcp/handlers/context.d.ts +15 -0
- package/dist/mcp/handlers/context.d.ts.map +1 -0
- package/dist/mcp/handlers/context.js +31 -0
- package/dist/mcp/handlers/context.js.map +1 -0
- package/dist/mcp/handlers/index.d.ts +5 -0
- package/dist/mcp/handlers/index.d.ts.map +1 -0
- package/dist/mcp/handlers/index.js +5 -0
- package/dist/mcp/handlers/index.js.map +1 -0
- package/dist/mcp/handlers/project.d.ts +41 -0
- package/dist/mcp/handlers/project.d.ts.map +1 -0
- package/dist/mcp/handlers/project.js +76 -0
- package/dist/mcp/handlers/project.js.map +1 -0
- package/dist/mcp/handlers/search.d.ts +27 -0
- package/dist/mcp/handlers/search.d.ts.map +1 -0
- package/dist/mcp/handlers/search.js +108 -0
- package/dist/mcp/handlers/search.js.map +1 -0
- package/dist/mcp/handlers/synthesis.d.ts +15 -0
- package/dist/mcp/handlers/synthesis.d.ts.map +1 -0
- package/dist/mcp/handlers/synthesis.js +58 -0
- package/dist/mcp/handlers/synthesis.js.map +1 -0
- package/dist/mcp-server.d.ts +10 -0
- package/dist/mcp-server.d.ts.map +1 -1
- package/dist/mcp-server.js +223 -471
- package/dist/mcp-server.js.map +1 -1
- package/dist/providers/chat-llm.d.ts +7 -2
- package/dist/providers/chat-llm.d.ts.map +1 -1
- package/dist/providers/chat-llm.js +23 -10
- package/dist/providers/chat-llm.js.map +1 -1
- package/dist/providers/index.d.ts +2 -1
- package/dist/providers/index.d.ts.map +1 -1
- package/dist/providers/index.js +2 -2
- package/dist/providers/index.js.map +1 -1
- package/dist/providers/openai.d.ts +5 -1
- package/dist/providers/openai.d.ts.map +1 -1
- package/dist/providers/openai.js +16 -6
- package/dist/providers/openai.js.map +1 -1
- package/dist/ranking/api-reranker.d.ts.map +1 -1
- package/dist/ranking/api-reranker.js +4 -1
- package/dist/ranking/api-reranker.js.map +1 -1
- package/dist/ranking/symbol-boost.js +2 -2
- package/dist/ranking/symbol-boost.js.map +1 -1
- package/dist/synthesis/conversational-synthesizer.d.ts.map +1 -1
- package/dist/synthesis/conversational-synthesizer.js +5 -2
- package/dist/synthesis/conversational-synthesizer.js.map +1 -1
- package/dist/synthesis/synthesizer.d.ts.map +1 -1
- package/dist/synthesis/synthesizer.js +5 -2
- package/dist/synthesis/synthesizer.js.map +1 -1
- package/dist/tests/rate-limiter.test.d.ts +2 -0
- package/dist/tests/rate-limiter.test.d.ts.map +1 -0
- package/dist/tests/rate-limiter.test.js +11 -0
- package/dist/tests/rate-limiter.test.js.map +1 -0
- package/dist/tests/search-normalization.test.d.ts +2 -0
- package/dist/tests/search-normalization.test.d.ts.map +1 -0
- package/dist/tests/search-normalization.test.js +9 -0
- package/dist/tests/search-normalization.test.js.map +1 -0
- package/dist/tests/semantic-chunker.test.d.ts +2 -0
- package/dist/tests/semantic-chunker.test.d.ts.map +1 -0
- package/dist/tests/semantic-chunker.test.js +48 -0
- package/dist/tests/semantic-chunker.test.js.map +1 -0
- package/dist/tests/simple-lru.test.d.ts +2 -0
- package/dist/tests/simple-lru.test.d.ts.map +1 -0
- package/dist/tests/simple-lru.test.js +21 -0
- package/dist/tests/simple-lru.test.js.map +1 -0
- package/dist/tests/symbol-boost.test.d.ts +2 -0
- package/dist/tests/symbol-boost.test.d.ts.map +1 -0
- package/dist/tests/symbol-boost.test.js +21 -0
- package/dist/tests/symbol-boost.test.js.map +1 -0
- package/dist/utils/simple-lru.d.ts +10 -0
- package/dist/utils/simple-lru.d.ts.map +1 -0
- package/dist/utils/simple-lru.js +38 -0
- package/dist/utils/simple-lru.js.map +1 -0
- package/package.json +3 -2
|
@@ -0,0 +1,372 @@
|
|
|
1
|
+
import crypto from 'crypto';
|
|
2
|
+
import fs from 'fs';
|
|
3
|
+
import path from 'path';
|
|
4
|
+
import { createEmbeddingProvider, getModelProfile, getSizeLimits } from '../providers/index.js';
|
|
5
|
+
import { BATCH_SIZE } from '../providers/base.js';
|
|
6
|
+
import { getTokenCountStats } from '../chunking/token-counter.js';
|
|
7
|
+
import { readCodemap, writeCodemap } from '../codemap/io.js';
|
|
8
|
+
import { normalizeChunkMetadata } from '../types/codemap.js';
|
|
9
|
+
import { LANG_RULES } from '../languages/rules.js';
|
|
10
|
+
import { groupNodesForChunking } from '../chunking/file-grouper.js';
|
|
11
|
+
import { cloneMerkle, computeFastHash, loadMerkle, normalizeToProjectPath, removeMerkleEntry, saveMerkle } from '../indexer/merkle.js';
|
|
12
|
+
import { attachSymbolGraphToCodemap } from '../symbols/graph.js';
|
|
13
|
+
import { resolveEncryptionPreference, writeChunkToDisk, removeChunkArtifacts } from '../storage/encrypted-chunks.js';
|
|
14
|
+
import { Database, initDatabase } from '../database/db.js';
|
|
15
|
+
import { BatchEmbeddingProcessor } from './batch-indexer.js';
|
|
16
|
+
import { logger } from '../utils/logger.js';
|
|
17
|
+
import { FileScanner } from './indexing/file-scanner.js';
|
|
18
|
+
import { resolveProviderContext } from '../config/resolver.js';
|
|
19
|
+
import { ChunkPipeline } from './indexing/chunk-pipeline.js';
|
|
20
|
+
export class IndexerEngine {
|
|
21
|
+
options;
|
|
22
|
+
db = null;
|
|
23
|
+
codemap = {};
|
|
24
|
+
merkle = {};
|
|
25
|
+
updatedMerkle = {};
|
|
26
|
+
merkleDirty = false;
|
|
27
|
+
indexMutated = false;
|
|
28
|
+
processedChunks = 0;
|
|
29
|
+
errors = [];
|
|
30
|
+
chunkingStats = {
|
|
31
|
+
totalNodes: 0,
|
|
32
|
+
skippedSmall: 0,
|
|
33
|
+
subdivided: 0,
|
|
34
|
+
statementFallback: 0,
|
|
35
|
+
normalChunks: 0,
|
|
36
|
+
mergedSmall: 0
|
|
37
|
+
};
|
|
38
|
+
encryptionPreference;
|
|
39
|
+
embeddingProvider = null;
|
|
40
|
+
batchProcessor = null;
|
|
41
|
+
chunkDir = '';
|
|
42
|
+
constructor(options = {}) {
|
|
43
|
+
this.options = options;
|
|
44
|
+
}
|
|
45
|
+
async index() {
|
|
46
|
+
const { repoPath = '.', provider = 'auto', onProgress = null, changedFiles = null, deletedFiles = [], embeddingProviderOverride = null, encryptMode = undefined } = this.options;
|
|
47
|
+
const repo = path.resolve(repoPath);
|
|
48
|
+
try {
|
|
49
|
+
await fs.promises.access(repo);
|
|
50
|
+
}
|
|
51
|
+
catch {
|
|
52
|
+
throw new Error(`Directory ${repo} does not exist`);
|
|
53
|
+
}
|
|
54
|
+
try {
|
|
55
|
+
const normalizedChanged = Array.isArray(changedFiles)
|
|
56
|
+
? Array.from(new Set(changedFiles
|
|
57
|
+
.map(file => normalizeToProjectPath(repo, file))
|
|
58
|
+
.filter(Boolean)))
|
|
59
|
+
: null;
|
|
60
|
+
const providerContext = resolveProviderContext(repo);
|
|
61
|
+
const normalizedDeleted = Array.from(new Set((Array.isArray(deletedFiles) ? deletedFiles : [])
|
|
62
|
+
.map(file => normalizeToProjectPath(repo, file))
|
|
63
|
+
.filter(Boolean)));
|
|
64
|
+
const deletedSet = new Set(normalizedDeleted);
|
|
65
|
+
const scanner = new FileScanner();
|
|
66
|
+
const { files, toDelete } = await scanner.scan(repo, normalizedChanged);
|
|
67
|
+
for (const file of toDelete) {
|
|
68
|
+
deletedSet.add(file);
|
|
69
|
+
}
|
|
70
|
+
const isPartialUpdate = normalizedChanged !== null;
|
|
71
|
+
this.embeddingProvider = embeddingProviderOverride || createEmbeddingProvider(provider, providerContext.embedding);
|
|
72
|
+
if (!embeddingProviderOverride && this.embeddingProvider.init) {
|
|
73
|
+
await this.embeddingProvider.init();
|
|
74
|
+
}
|
|
75
|
+
const providerName = this.embeddingProvider.getName();
|
|
76
|
+
const modelName = this.embeddingProvider.getModelName ? this.embeddingProvider.getModelName() : null;
|
|
77
|
+
const modelProfile = await getModelProfile(providerName, modelName || providerName);
|
|
78
|
+
const limits = getSizeLimits(modelProfile);
|
|
79
|
+
if (!process.env.CODEVAULT_QUIET) {
|
|
80
|
+
logger.info(`Chunking Configuration`, {
|
|
81
|
+
provider: providerName,
|
|
82
|
+
model: modelName,
|
|
83
|
+
dimensions: this.embeddingProvider.getDimensions(),
|
|
84
|
+
mode: limits.unit
|
|
85
|
+
});
|
|
86
|
+
}
|
|
87
|
+
await initDatabase(this.embeddingProvider.getDimensions(), repo);
|
|
88
|
+
const codemapPath = path.join(repo, 'codevault.codemap.json');
|
|
89
|
+
this.chunkDir = path.join(repo, '.codevault/chunks');
|
|
90
|
+
const dbPath = path.join(repo, '.codevault/codevault.db');
|
|
91
|
+
await this.checkDimensionMismatch(dbPath, this.embeddingProvider);
|
|
92
|
+
this.encryptionPreference = resolveEncryptionPreference({ mode: encryptMode, logger: console }); // keeping console for logger shim in encrypted-chunks
|
|
93
|
+
this.codemap = readCodemap(codemapPath);
|
|
94
|
+
this.merkle = loadMerkle(repo);
|
|
95
|
+
this.updatedMerkle = cloneMerkle(this.merkle);
|
|
96
|
+
this.db = new Database(dbPath);
|
|
97
|
+
// Create batch processor for efficient embedding generation
|
|
98
|
+
this.batchProcessor = new BatchEmbeddingProcessor(this.embeddingProvider, this.db, BATCH_SIZE);
|
|
99
|
+
const chunkPipeline = new ChunkPipeline();
|
|
100
|
+
for (const rel of files) {
|
|
101
|
+
deletedSet.delete(rel);
|
|
102
|
+
const abs = path.join(repo, rel);
|
|
103
|
+
const ext = path.extname(rel).toLowerCase();
|
|
104
|
+
const rule = LANG_RULES[ext];
|
|
105
|
+
if (!rule)
|
|
106
|
+
continue;
|
|
107
|
+
const existingChunks = new Map(Object.entries(this.codemap)
|
|
108
|
+
.filter(([, metadata]) => metadata && metadata.file === rel));
|
|
109
|
+
const staleChunkIds = new Set(existingChunks.keys());
|
|
110
|
+
const chunkMerkleHashes = [];
|
|
111
|
+
let fileHash = null;
|
|
112
|
+
try {
|
|
113
|
+
const source = await fs.promises.readFile(abs, 'utf8');
|
|
114
|
+
fileHash = await computeFastHash(source);
|
|
115
|
+
const previousMerkle = this.merkle[rel];
|
|
116
|
+
if (previousMerkle && previousMerkle.shaFile === fileHash) {
|
|
117
|
+
continue;
|
|
118
|
+
}
|
|
119
|
+
const collectedNodes = await chunkPipeline.collectNodesForFile(source, rule);
|
|
120
|
+
const nodeGroups = await groupNodesForChunking(collectedNodes, source, modelProfile, rule);
|
|
121
|
+
const existingChunks = new Map(Object.entries(this.codemap)
|
|
122
|
+
.filter(([, metadata]) => metadata && metadata.file === rel));
|
|
123
|
+
const staleChunkIds = new Set(existingChunks.keys());
|
|
124
|
+
const chunkMerkleHashes = [];
|
|
125
|
+
await chunkPipeline.processGroups(nodeGroups, source, rule, limits, modelProfile, rel, { staleChunkIds, existingChunks }, chunkMerkleHashes, onProgress, (params) => this.embedAndStore(params), this.chunkingStats);
|
|
126
|
+
if (staleChunkIds.size > 0) {
|
|
127
|
+
await this.deleteChunks(Array.from(staleChunkIds), existingChunks);
|
|
128
|
+
this.indexMutated = true;
|
|
129
|
+
}
|
|
130
|
+
if (fileHash) {
|
|
131
|
+
this.updatedMerkle[rel] = {
|
|
132
|
+
shaFile: fileHash,
|
|
133
|
+
chunkShas: chunkMerkleHashes
|
|
134
|
+
};
|
|
135
|
+
this.merkleDirty = true;
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
catch (error) {
|
|
139
|
+
this.errors.push({ type: 'processing_error', file: rel, error: error.message });
|
|
140
|
+
// Fallback logic
|
|
141
|
+
try {
|
|
142
|
+
if (!fs.existsSync(abs)) {
|
|
143
|
+
continue;
|
|
144
|
+
}
|
|
145
|
+
const source = await fs.promises.readFile(abs, 'utf8');
|
|
146
|
+
const fallbackSymbol = path.basename(rel) || rel;
|
|
147
|
+
const sha = crypto.createHash('sha1').update(source).digest('hex');
|
|
148
|
+
const chunkId = `${rel}:fallback:${sha.substring(0, 8)}`;
|
|
149
|
+
const chunkMerkleHash = await computeFastHash(source);
|
|
150
|
+
const fallbackMetadata = { tags: [], intent: null, description: null };
|
|
151
|
+
const contextInfo = {
|
|
152
|
+
nodeType: 'file',
|
|
153
|
+
startLine: 1,
|
|
154
|
+
endLine: source.split('\n').length,
|
|
155
|
+
codeLength: source.length,
|
|
156
|
+
hasDocumentation: false,
|
|
157
|
+
variableCount: 0
|
|
158
|
+
};
|
|
159
|
+
await this.embedAndStore({
|
|
160
|
+
code: source,
|
|
161
|
+
enhancedEmbeddingText: source,
|
|
162
|
+
chunkId,
|
|
163
|
+
sha,
|
|
164
|
+
lang: rule.lang,
|
|
165
|
+
rel,
|
|
166
|
+
symbol: fallbackSymbol,
|
|
167
|
+
chunkType: 'file',
|
|
168
|
+
codevaultMetadata: fallbackMetadata,
|
|
169
|
+
importantVariables: [],
|
|
170
|
+
docComments: null,
|
|
171
|
+
contextInfo,
|
|
172
|
+
symbolData: {
|
|
173
|
+
signature: `${fallbackSymbol}()`,
|
|
174
|
+
parameters: [],
|
|
175
|
+
returnType: null,
|
|
176
|
+
calls: []
|
|
177
|
+
}
|
|
178
|
+
});
|
|
179
|
+
this.processedChunks++;
|
|
180
|
+
this.indexMutated = true;
|
|
181
|
+
if (onProgress) {
|
|
182
|
+
onProgress({ type: 'chunk_processed', file: rel, symbol: fallbackSymbol, chunkId });
|
|
183
|
+
}
|
|
184
|
+
staleChunkIds.delete(chunkId);
|
|
185
|
+
if (staleChunkIds.size > 0) {
|
|
186
|
+
await this.deleteChunks(Array.from(staleChunkIds), existingChunks);
|
|
187
|
+
this.indexMutated = true;
|
|
188
|
+
}
|
|
189
|
+
chunkMerkleHashes.length = 0;
|
|
190
|
+
chunkMerkleHashes.push(chunkMerkleHash);
|
|
191
|
+
fileHash = chunkMerkleHash;
|
|
192
|
+
this.updatedMerkle[rel] = {
|
|
193
|
+
shaFile: chunkMerkleHash,
|
|
194
|
+
chunkShas: [...chunkMerkleHashes]
|
|
195
|
+
};
|
|
196
|
+
this.merkleDirty = true;
|
|
197
|
+
}
|
|
198
|
+
catch (fallbackError) {
|
|
199
|
+
this.errors.push({ type: 'fallback_error', file: rel, error: fallbackError.message });
|
|
200
|
+
}
|
|
201
|
+
}
|
|
202
|
+
}
|
|
203
|
+
for (const fileRel of deletedSet) {
|
|
204
|
+
await this.removeFileArtifacts(fileRel);
|
|
205
|
+
}
|
|
206
|
+
if (!isPartialUpdate) {
|
|
207
|
+
const existingFilesSet = new Set(files);
|
|
208
|
+
for (const fileRel of Object.keys(this.merkle)) {
|
|
209
|
+
if (!existingFilesSet.has(fileRel)) {
|
|
210
|
+
await this.removeFileArtifacts(fileRel);
|
|
211
|
+
}
|
|
212
|
+
}
|
|
213
|
+
}
|
|
214
|
+
if (onProgress) {
|
|
215
|
+
onProgress({ type: 'finalizing' });
|
|
216
|
+
}
|
|
217
|
+
if (this.merkleDirty) {
|
|
218
|
+
saveMerkle(repo, this.updatedMerkle);
|
|
219
|
+
}
|
|
220
|
+
attachSymbolGraphToCodemap(this.codemap);
|
|
221
|
+
this.codemap = writeCodemap(codemapPath, this.codemap);
|
|
222
|
+
const tokenStats = getTokenCountStats();
|
|
223
|
+
if (!process.env.CODEVAULT_QUIET) {
|
|
224
|
+
logger.info('Chunking Statistics', {
|
|
225
|
+
stats: this.chunkingStats,
|
|
226
|
+
processedChunks: this.processedChunks,
|
|
227
|
+
totalChunks: Object.keys(this.codemap).length
|
|
228
|
+
});
|
|
229
|
+
}
|
|
230
|
+
return {
|
|
231
|
+
success: true,
|
|
232
|
+
processedChunks: this.processedChunks,
|
|
233
|
+
totalChunks: Object.keys(this.codemap).length,
|
|
234
|
+
provider: this.embeddingProvider.getName(),
|
|
235
|
+
errors: this.errors,
|
|
236
|
+
chunkingStats: this.chunkingStats,
|
|
237
|
+
tokenStats: modelProfile.useTokens ? tokenStats : undefined
|
|
238
|
+
};
|
|
239
|
+
}
|
|
240
|
+
finally {
|
|
241
|
+
// Ensure resources are cleaned up even on errors
|
|
242
|
+
try {
|
|
243
|
+
await this.batchProcessor?.flush();
|
|
244
|
+
}
|
|
245
|
+
catch (error) {
|
|
246
|
+
this.errors.push({ type: 'finalize_error', error: error.message });
|
|
247
|
+
}
|
|
248
|
+
try {
|
|
249
|
+
if (this.db) {
|
|
250
|
+
this.db.close();
|
|
251
|
+
this.db = null;
|
|
252
|
+
}
|
|
253
|
+
}
|
|
254
|
+
catch (error) {
|
|
255
|
+
this.errors.push({ type: 'db_close_error', error: error.message });
|
|
256
|
+
}
|
|
257
|
+
}
|
|
258
|
+
}
|
|
259
|
+
async embedAndStore(params) {
|
|
260
|
+
try {
|
|
261
|
+
if (!this.batchProcessor)
|
|
262
|
+
throw new Error('Batch processor not initialized');
|
|
263
|
+
await this.batchProcessor.addChunk({
|
|
264
|
+
chunkId: params.chunkId,
|
|
265
|
+
enhancedEmbeddingText: params.enhancedEmbeddingText,
|
|
266
|
+
params: {
|
|
267
|
+
code: params.code,
|
|
268
|
+
sha: params.sha,
|
|
269
|
+
lang: params.lang,
|
|
270
|
+
rel: params.rel,
|
|
271
|
+
symbol: params.symbol,
|
|
272
|
+
chunkType: params.chunkType,
|
|
273
|
+
codevaultMetadata: params.codevaultMetadata,
|
|
274
|
+
importantVariables: params.importantVariables,
|
|
275
|
+
docComments: params.docComments,
|
|
276
|
+
contextInfo: params.contextInfo
|
|
277
|
+
}
|
|
278
|
+
});
|
|
279
|
+
this.indexMutated = true;
|
|
280
|
+
await fs.promises.mkdir(this.chunkDir, { recursive: true });
|
|
281
|
+
const writeResult = writeChunkToDisk({
|
|
282
|
+
chunkDir: this.chunkDir,
|
|
283
|
+
sha: params.sha,
|
|
284
|
+
code: params.code,
|
|
285
|
+
encryption: this.encryptionPreference
|
|
286
|
+
});
|
|
287
|
+
const previousMetadata = this.codemap[params.chunkId];
|
|
288
|
+
this.codemap[params.chunkId] = normalizeChunkMetadata({
|
|
289
|
+
file: params.rel,
|
|
290
|
+
symbol: params.symbol,
|
|
291
|
+
sha: params.sha,
|
|
292
|
+
lang: params.lang,
|
|
293
|
+
chunkType: params.chunkType,
|
|
294
|
+
provider: this.embeddingProvider?.getName() || 'unknown',
|
|
295
|
+
dimensions: this.embeddingProvider?.getDimensions() || 0,
|
|
296
|
+
hasCodevaultTags: Array.isArray(params.codevaultMetadata.tags) && params.codevaultMetadata.tags.length > 0,
|
|
297
|
+
hasIntent: !!params.codevaultMetadata.intent,
|
|
298
|
+
hasDocumentation: !!params.docComments,
|
|
299
|
+
variableCount: Array.isArray(params.importantVariables) ? params.importantVariables.length : 0,
|
|
300
|
+
encrypted: !!(writeResult && writeResult.encrypted),
|
|
301
|
+
symbol_signature: params.symbolData && params.symbolData.signature ? params.symbolData.signature : undefined,
|
|
302
|
+
symbol_parameters: params.symbolData && Array.isArray(params.symbolData.parameters) ? params.symbolData.parameters : undefined,
|
|
303
|
+
symbol_return: params.symbolData && params.symbolData.returnType ? params.symbolData.returnType : undefined,
|
|
304
|
+
symbol_calls: params.symbolData && Array.isArray(params.symbolData.calls) ? params.symbolData.calls : undefined
|
|
305
|
+
}, previousMetadata);
|
|
306
|
+
}
|
|
307
|
+
catch (error) {
|
|
308
|
+
this.errors.push({ type: 'indexing_error', chunkId: params.chunkId, error: error.message });
|
|
309
|
+
throw error;
|
|
310
|
+
}
|
|
311
|
+
}
|
|
312
|
+
async deleteChunks(chunkIds, metadataLookup = new Map()) {
|
|
313
|
+
if (!Array.isArray(chunkIds) || chunkIds.length === 0) {
|
|
314
|
+
return;
|
|
315
|
+
}
|
|
316
|
+
if (this.db) {
|
|
317
|
+
await this.db.deleteChunks(chunkIds);
|
|
318
|
+
}
|
|
319
|
+
for (const chunkId of chunkIds) {
|
|
320
|
+
const metadata = metadataLookup.get(chunkId) || this.codemap[chunkId];
|
|
321
|
+
if (metadata && metadata.sha) {
|
|
322
|
+
removeChunkArtifacts(this.chunkDir, metadata.sha);
|
|
323
|
+
}
|
|
324
|
+
delete this.codemap[chunkId];
|
|
325
|
+
}
|
|
326
|
+
}
|
|
327
|
+
async removeFileArtifacts(fileRel) {
|
|
328
|
+
const entries = Object.entries(this.codemap)
|
|
329
|
+
.filter(([, metadata]) => metadata && metadata.file === fileRel);
|
|
330
|
+
if (entries.length > 0) {
|
|
331
|
+
const metadataLookup = new Map(entries);
|
|
332
|
+
await this.deleteChunks(entries.map(([chunkId]) => chunkId), metadataLookup);
|
|
333
|
+
this.indexMutated = true;
|
|
334
|
+
}
|
|
335
|
+
if (removeMerkleEntry(this.updatedMerkle, fileRel)) {
|
|
336
|
+
this.merkleDirty = true;
|
|
337
|
+
}
|
|
338
|
+
}
|
|
339
|
+
async checkDimensionMismatch(dbPath, embeddingProvider) {
|
|
340
|
+
try {
|
|
341
|
+
await fs.promises.access(dbPath);
|
|
342
|
+
}
|
|
343
|
+
catch {
|
|
344
|
+
return;
|
|
345
|
+
}
|
|
346
|
+
const db = new Database(dbPath);
|
|
347
|
+
try {
|
|
348
|
+
const existingDimensions = await db.getExistingDimensions();
|
|
349
|
+
if (existingDimensions.length > 0) {
|
|
350
|
+
const currentProvider = embeddingProvider.getName();
|
|
351
|
+
const currentDimensions = embeddingProvider.getDimensions();
|
|
352
|
+
const hasMismatch = existingDimensions.some(row => row.embedding_provider !== currentProvider ||
|
|
353
|
+
row.embedding_dimensions !== currentDimensions);
|
|
354
|
+
if (hasMismatch) {
|
|
355
|
+
logger.warn('Dimension/Provider Mismatch Detected!', {
|
|
356
|
+
existing: existingDimensions,
|
|
357
|
+
current: { provider: currentProvider, dimensions: currentDimensions },
|
|
358
|
+
recommendation: 'Full re-index recommended'
|
|
359
|
+
});
|
|
360
|
+
await new Promise(resolve => setTimeout(resolve, 2000));
|
|
361
|
+
}
|
|
362
|
+
}
|
|
363
|
+
}
|
|
364
|
+
catch (error) {
|
|
365
|
+
// Ignore migration check errors
|
|
366
|
+
}
|
|
367
|
+
finally {
|
|
368
|
+
db.close();
|
|
369
|
+
}
|
|
370
|
+
}
|
|
371
|
+
}
|
|
372
|
+
//# sourceMappingURL=IndexerEngine.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"IndexerEngine.js","sourceRoot":"","sources":["../../src/core/IndexerEngine.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,uBAAuB,EAAE,eAAe,EAAE,aAAa,EAA0B,MAAM,uBAAuB,CAAC;AACxH,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAC7D,OAAO,EAAE,sBAAsB,EAAgB,MAAM,qBAAqB,CAAC;AAC3E,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnD,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EACL,WAAW,EACX,eAAe,EACf,UAAU,EACV,sBAAsB,EACtB,iBAAiB,EACjB,UAAU,EAEX,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AACjE,OAAO,EACL,2BAA2B,EAC3B,gBAAgB,EAChB,oBAAoB,EACrB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAC3D,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAE7D,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AACzD,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAG7D,MAAM,OAAO,aAAa;IAsBJ;IArBZ,EAAE,GAAoB,IAAI,CAAC;IAC3B,OAAO,GAAY,EAAE,CAAC;IACtB,MAAM,GAAe,EAAE,CAAC;IACxB,aAAa,GAAe,EAAE,CAAC;IAC/B,WAAW,GAAG,KAAK,CAAC;IACpB,YAAY,GAAG,KAAK,CAAC;IACrB,eAAe,GAAG,CAAC,CAAC;IACpB,MAAM,GAAU,EAAE,CAAC;IACnB,aAAa,GAAkB;QACrC,UAAU,EAAE,CAAC;QACb,YAAY,EAAE,CAAC;QACf,UAAU,EAAE,CAAC;QACb,iBAAiB,EAAE,CAAC;QACpB,YAAY,EAAE,CAAC;QACf,WAAW,EAAE,CAAC;KACf,CAAC;IACM,oBAAoB,CAAM;IAC1B,iBAAiB,GAA6B,IAAI,CAAC;IACnD,cAAc,GAAmC,IAAI,CAAC;IACtD,QAAQ,GAAW,EAAE,CAAC;IAE9B,YAAoB,UAA+B,EAAE;QAAjC,YAAO,GAAP,OAAO,CAA0B;IAAG,CAAC;IAElD,KAAK,CAAC,KAAK;QAChB,MAAM,EACJ,QAAQ,GAAG,GAAG,EACd,QAAQ,GAAG,MAAM,EACjB,UAAU,GAAG,IAAI,EACjB,YAAY,GAAG,IAAI,EACnB,YAAY,GAAG,EAAE,EACjB,yBAAyB,GAAG,IAAI,EAChC,WAAW,GAAG,SAAS,EACxB,GAAG,IAAI,CAAC,OAAO,CAAC;QAEjB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAEpC,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,KAAK,CAAC,aAAa,IAAI,iBAAiB,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,CAAC;YACH,MAAM,iBAAiB,GAAG,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC;gBACnD,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAChB,YAAY;qBACT,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,sBAAsB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;qBAC/C,MAAM,CAAC,OAAO,CAAa,CAC/B,CAAC;gBACJ,CAAC,CAAC,IAAI,CAAC;YACT,MAAM,eAAe,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;YAErD,MAAM,iBAAiB,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAC1C,CAAC,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC9C,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,sBAAsB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;iBAC/C,MAAM,CAAC,OAAO,CAAa,CAC/B,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,iBAAiB,CAAC,CAAC;YAE9C,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;YAExE,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;gBAC5B,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACvB,CAAC;YACD,MAAM,eAAe,GAAG,iBAAiB,KAAK,IAAI,CAAC;YAEnD,IAAI,CAAC,iBAAiB,GAAG,yBAAyB,IAAI,uBAAuB,CAAC,QAAQ,EAAE,eAAe,CAAC,SAAS,CAAC,CAAC;YAEnH,IAAI,CAAC,yBAAyB,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC;gBAC9D,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC;YACtC,CAAC;YAED,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC;YACtD,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YACrG,MAAM,YAAY,GAAG,MAAM,eAAe,CAAC,YAAY,EAAE,SAAS,IAAI,YAAY,CAAC,CAAC;YACpF,MAAM,MAAM,GAAG,aAAa,CAAC,YAAY,CAAC,CAAC;YAE3C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;gBACjC,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE;oBACpC,QAAQ,EAAE,YAAY;oBACtB,KAAK,EAAE,SAAS;oBAChB,UAAU,EAAE,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE;oBAClD,IAAI,EAAE,MAAM,CAAC,IAAI;iBAClB,CAAC,CAAC;YACL,CAAC;YAED,MAAM,YAAY,CAAC,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,EAAE,IAAI,CAAC,CAAC;YAEjE,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,wBAAwB,CAAC,CAAC;YAC9D,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;YACrD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC;YAE1D,MAAM,IAAI,CAAC,sBAAsB,CAAC,MAAM,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAElE,IAAI,CAAC,oBAAoB,GAAG,2BAA2B,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,sDAAsD;YACvJ,IAAI,CAAC,OAAO,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;YAExC,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;YAC/B,IAAI,CAAC,aAAa,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAE9C,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;YAE/B,4DAA4D;YAC5D,IAAI,CAAC,cAAc,GAAG,IAAI,uBAAuB,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAC/F,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;YAE1C,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,CAAC;gBACxB,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAEvB,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;gBACjC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;gBAC5C,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;gBAE7B,IAAI,CAAC,IAAI;oBAAE,SAAS;gBAEpB,MAAM,cAAc,GAAG,IAAI,GAAG,CAC5B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;qBACzB,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,GAAG,CAAoB,CAClF,CAAC;gBACF,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;gBACrD,MAAM,iBAAiB,GAAa,EAAE,CAAC;gBACvC,IAAI,QAAQ,GAAkB,IAAI,CAAC;gBAEnC,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;oBACvD,QAAQ,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;oBAEzC,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;oBACxC,IAAI,cAAc,IAAI,cAAc,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;wBAC1D,SAAS;oBACX,CAAC;oBAED,MAAM,cAAc,GAAG,MAAM,aAAa,CAAC,mBAAmB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;oBAE7E,MAAM,UAAU,GAAG,MAAM,qBAAqB,CAC5C,cAAc,EACd,MAAM,EACN,YAAY,EACZ,IAAI,CACL,CAAC;oBAEF,MAAM,cAAc,GAAG,IAAI,GAAG,CAC5B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;yBACzB,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,GAAG,CAAoB,CAClF,CAAC;oBACF,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;oBACrD,MAAM,iBAAiB,GAAa,EAAE,CAAC;oBAEvC,MAAM,aAAa,CAAC,aAAa,CAC/B,UAAU,EACV,MAAM,EACN,IAAI,EACJ,MAAM,EACN,YAAY,EACZ,GAAG,EACH,EAAE,aAAa,EAAE,cAAc,EAAE,EACjC,iBAAiB,EACjB,UAAU,EACV,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,EACtC,IAAI,CAAC,aAAa,CACnB,CAAC;oBAEF,IAAI,aAAa,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;wBAC3B,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,cAAc,CAAC,CAAC;wBACnE,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;oBAC3B,CAAC;oBAED,IAAI,QAAQ,EAAE,CAAC;wBACb,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG;4BACxB,OAAO,EAAE,QAAQ;4BACjB,SAAS,EAAE,iBAAiB;yBAC7B,CAAC;wBACF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;oBAC1B,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;oBAE3F,iBAAiB;oBACjB,IAAI,CAAC;wBACF,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;4BACzB,SAAS;wBACX,CAAC;wBAED,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;wBACvD,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC;wBACjD,MAAM,GAAG,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;wBACnE,MAAM,OAAO,GAAG,GAAG,GAAG,aAAa,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;wBACzD,MAAM,eAAe,GAAG,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;wBACtD,MAAM,gBAAgB,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;wBACvE,MAAM,WAAW,GAAG;4BAClB,QAAQ,EAAE,MAAM;4BAChB,SAAS,EAAE,CAAC;4BACZ,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM;4BAClC,UAAU,EAAE,MAAM,CAAC,MAAM;4BACzB,gBAAgB,EAAE,KAAK;4BACvB,aAAa,EAAE,CAAC;yBACjB,CAAC;wBAEF,MAAM,IAAI,CAAC,aAAa,CAAC;4BACvB,IAAI,EAAE,MAAM;4BACZ,qBAAqB,EAAE,MAAM;4BAC7B,OAAO;4BACP,GAAG;4BACH,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,GAAG;4BACH,MAAM,EAAE,cAAc;4BACtB,SAAS,EAAE,MAAM;4BACjB,iBAAiB,EAAE,gBAAgB;4BACnC,kBAAkB,EAAE,EAAE;4BACtB,WAAW,EAAE,IAAI;4BACjB,WAAW;4BACX,UAAU,EAAE;gCACV,SAAS,EAAE,GAAG,cAAc,IAAI;gCAChC,UAAU,EAAE,EAAE;gCACd,UAAU,EAAE,IAAI;gCAChB,KAAK,EAAE,EAAE;6BACV;yBACF,CAAC,CAAC;wBAEH,IAAI,CAAC,eAAe,EAAE,CAAC;wBACvB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;wBAEzB,IAAI,UAAU,EAAE,CAAC;4BACf,UAAU,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE,CAAC,CAAC;wBACtF,CAAC;wBAED,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;wBAC9B,IAAI,aAAa,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;4BAC3B,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,cAAc,CAAC,CAAC;4BACnE,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;wBAC3B,CAAC;wBAED,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;wBAC7B,iBAAiB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;wBACxC,QAAQ,GAAG,eAAe,CAAC;wBAC3B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG;4BACxB,OAAO,EAAE,eAAe;4BACxB,SAAS,EAAE,CAAC,GAAG,iBAAiB,CAAC;yBAClC,CAAC;wBACF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;oBAE1B,CAAC;oBAAC,OAAO,aAAa,EAAE,CAAC;wBACvB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAG,aAAuB,CAAC,OAAO,EAAE,CAAC,CAAC;oBACnG,CAAC;gBACH,CAAC;YACH,CAAC;YAED,KAAK,MAAM,OAAO,IAAI,UAAU,EAAE,CAAC;gBACjC,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;YAC1C,CAAC;YAED,IAAI,CAAC,eAAe,EAAE,CAAC;gBACrB,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;gBACxC,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC/C,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;wBACnC,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;oBAC1C,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,UAAU,EAAE,CAAC;gBACf,UAAU,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;YACrC,CAAC;YAED,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;gBACrB,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;YACvC,CAAC;YAED,0BAA0B,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACzC,IAAI,CAAC,OAAO,GAAG,YAAY,CAAC,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YAEvD,MAAM,UAAU,GAAG,kBAAkB,EAAE,CAAC;YAExC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;gBAC/B,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE;oBAC/B,KAAK,EAAE,IAAI,CAAC,aAAa;oBACzB,eAAe,EAAE,IAAI,CAAC,eAAe;oBACrC,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM;iBAChD,CAAC,CAAC;YACP,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,eAAe,EAAE,IAAI,CAAC,eAAe;gBACrC,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM;gBAC7C,QAAQ,EAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE;gBAC1C,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,UAAU,EAAE,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;aAC5D,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,iDAAiD;YACjD,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,CAAC;YACrC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;YAChF,CAAC;YAED,IAAI,CAAC;gBACH,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;oBACZ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;oBAChB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;gBACjB,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;YAChF,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,MAc3B;QACC,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,cAAc;gBAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;YAE7E,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC;gBACjC,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,qBAAqB,EAAE,MAAM,CAAC,qBAAqB;gBACnD,MAAM,EAAE;oBACN,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,GAAG,EAAE,MAAM,CAAC,GAAG;oBACf,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,GAAG,EAAE,MAAM,CAAC,GAAG;oBACf,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,SAAS,EAAE,MAAM,CAAC,SAAS;oBAC3B,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;oBAC3C,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;oBAC7C,WAAW,EAAE,MAAM,CAAC,WAAW;oBAC/B,WAAW,EAAE,MAAM,CAAC,WAAW;iBAChC;aACF,CAAC,CAAC;YAEH,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YAEzB,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5D,MAAM,WAAW,GAAG,gBAAgB,CAAC;gBACnC,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,GAAG,EAAE,MAAM,CAAC,GAAG;gBACf,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,UAAU,EAAE,IAAI,CAAC,oBAAoB;aACtC,CAAC,CAAC;YAEH,MAAM,gBAAgB,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACtD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,sBAAsB,CAAC;gBACpD,IAAI,EAAE,MAAM,CAAC,GAAG;gBAChB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,GAAG,EAAE,MAAM,CAAC,GAAG;gBACf,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,QAAQ,EAAE,IAAI,CAAC,iBAAiB,EAAE,OAAO,EAAE,IAAI,SAAS;gBACxD,UAAU,EAAE,IAAI,CAAC,iBAAiB,EAAE,aAAa,EAAE,IAAI,CAAC;gBACxD,gBAAgB,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC;gBAC1G,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,iBAAiB,CAAC,MAAM;gBAC5C,gBAAgB,EAAE,CAAC,CAAC,MAAM,CAAC,WAAW;gBACtC,aAAa,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAC9F,SAAS,EAAE,CAAC,CAAC,CAAC,WAAW,IAAI,WAAW,CAAC,SAAS,CAAC;gBACnD,gBAAgB,EAAE,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBAC5G,iBAAiB,EAAE,MAAM,CAAC,UAAU,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;gBAC9H,aAAa,EAAE,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;gBAC3G,YAAY,EAAE,MAAM,CAAC,UAAU,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;aAChH,EAAE,gBAAgB,CAAC,CAAC;QAEvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;YACvG,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,QAAkB,EAAE,iBAAiB,IAAI,GAAG,EAAe;QACpF,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtD,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACV,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QACzC,CAAC;QAED,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACtE,IAAI,QAAQ,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC;gBAC7B,oBAAoB,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC;YACpD,CAAC;YACD,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,OAAe;QAC/C,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;aACzC,MAAM,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;QAEnE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,OAA0B,CAAC,CAAC;YAC3D,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,cAAc,CAAC,CAAC;YAC7E,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC3B,CAAC;QAED,IAAI,iBAAiB,CAAC,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,EAAE,CAAC;YACnD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAAC,MAAc,EAAE,iBAAsB;QACzE,IAAI,CAAC;YACD,MAAM,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC;QAAC,MAAM,CAAC;YACL,OAAO;QACX,CAAC;QAED,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAChC,IAAI,CAAC;YACH,MAAM,kBAAkB,GAAG,MAAM,EAAE,CAAC,qBAAqB,EAAE,CAAC;YAE5D,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClC,MAAM,eAAe,GAAG,iBAAiB,CAAC,OAAO,EAAE,CAAC;gBACpD,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,aAAa,EAAE,CAAC;gBAE5D,MAAM,WAAW,GAAG,kBAAkB,CAAC,IAAI,CACzC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,kBAAkB,KAAK,eAAe;oBAC1C,GAAG,CAAC,oBAAoB,KAAK,iBAAiB,CACtD,CAAC;gBAEF,IAAI,WAAW,EAAE,CAAC;oBAChB,MAAM,CAAC,IAAI,CAAC,uCAAuC,EAAE;wBACjD,QAAQ,EAAE,kBAAkB;wBAC5B,OAAO,EAAE,EAAE,QAAQ,EAAE,eAAe,EAAE,UAAU,EAAE,iBAAiB,EAAE;wBACrE,cAAc,EAAE,2BAA2B;qBAC9C,CAAC,CAAC;oBAEH,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC1D,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,gCAAgC;QAClC,CAAC;gBAAS,CAAC;YACT,EAAE,CAAC,KAAK,EAAE,CAAC;QACb,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import type { ScopeFilters } from '../types/search.js';
|
|
2
|
+
import type { SearchCodeResult, GetChunkResult } from './types.js';
|
|
3
|
+
export declare class SearchService {
|
|
4
|
+
private bm25Cache;
|
|
5
|
+
private chunkCache;
|
|
6
|
+
constructor();
|
|
7
|
+
clearCaches(): void;
|
|
8
|
+
search(query: string, limit?: number, provider?: string, workingPath?: string, scopeOptions?: ScopeFilters): Promise<SearchCodeResult>;
|
|
9
|
+
getOverview(limit?: number, workingPath?: string): Promise<SearchCodeResult>;
|
|
10
|
+
getChunk(sha: string, workingPath?: string): Promise<GetChunkResult>;
|
|
11
|
+
private createErrorResult;
|
|
12
|
+
private normalizeQuery;
|
|
13
|
+
private buildVectorPool;
|
|
14
|
+
private tryHybridFusion;
|
|
15
|
+
private sortWithSymbolBoost;
|
|
16
|
+
private applyReranker;
|
|
17
|
+
private mapResults;
|
|
18
|
+
private cosineSimilarity;
|
|
19
|
+
private getBm25CacheKey;
|
|
20
|
+
private getChunkCacheKey;
|
|
21
|
+
private readChunkTextCached;
|
|
22
|
+
private buildBm25Document;
|
|
23
|
+
private ensureBm25IndexForChunks;
|
|
24
|
+
}
|
|
25
|
+
//# sourceMappingURL=SearchService.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"SearchService.d.ts","sourceRoot":"","sources":["../../src/core/SearchService.ts"],"names":[],"mappings":"AAqBA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,gBAAgB,EAAgB,cAAc,EAAE,MAAM,YAAY,CAAC;AA0BjF,qBAAa,aAAa;IACxB,OAAO,CAAC,SAAS,CAA8D;IAC/E,OAAO,CAAC,UAAU,CAAmC;;IAO9C,WAAW,IAAI,IAAI;IAMb,MAAM,CACjB,KAAK,EAAE,MAAM,EACb,KAAK,GAAE,MAA8C,EACrD,QAAQ,GAAE,MAAe,EACzB,WAAW,GAAE,MAAY,EACzB,YAAY,GAAE,YAAiB,GAC9B,OAAO,CAAC,gBAAgB,CAAC;IA4If,WAAW,CAAC,KAAK,GAAE,MAAW,EAAE,WAAW,GAAE,MAAY,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA4BrF,QAAQ,CAAC,GAAG,EAAE,MAAM,EAAE,WAAW,SAAM,GAAG,OAAO,CAAC,cAAc,CAAC;IAoB9E,OAAO,CAAC,iBAAiB;IAczB,OAAO,CAAC,cAAc;YAIR,eAAe;IA4E7B,OAAO,CAAC,eAAe;IAyEvB,OAAO,CAAC,mBAAmB;YAwBb,aAAa;IA6B3B,OAAO,CAAC,UAAU;IAmClB,OAAO,CAAC,gBAAgB;IAWxB,OAAO,CAAC,eAAe;IAIvB,OAAO,CAAC,gBAAgB;IAIxB,OAAO,CAAC,mBAAmB;IAiB3B,OAAO,CAAC,iBAAiB;IAYzB,OAAO,CAAC,wBAAwB;CA+BjC"}
|