@phmotad/fire-code 0.2.8 → 0.3.0
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/cli/commands/index-cmd.d.ts +3 -0
- package/dist/cli/commands/index-cmd.d.ts.map +1 -1
- package/dist/cli/commands/index-cmd.js +94 -19
- package/dist/cli/commands/index-cmd.js.map +1 -1
- package/dist/cli/commands/install.d.ts.map +1 -1
- package/dist/cli/commands/install.js +12 -0
- package/dist/cli/commands/install.js.map +1 -1
- package/dist/cli/index.js +10 -3
- package/dist/cli/index.js.map +1 -1
- package/dist/core/ExecutionEngine.d.ts +3 -1
- package/dist/core/ExecutionEngine.d.ts.map +1 -1
- package/dist/core/ExecutionEngine.js +36 -17
- package/dist/core/ExecutionEngine.js.map +1 -1
- package/dist/db/DatabaseManager.d.ts +7 -2
- package/dist/db/DatabaseManager.d.ts.map +1 -1
- package/dist/db/DatabaseManager.js +51 -26
- package/dist/db/DatabaseManager.js.map +1 -1
- package/dist/db/SqlJsAdapter.d.ts +32 -0
- package/dist/db/SqlJsAdapter.d.ts.map +1 -0
- package/dist/db/SqlJsAdapter.js +130 -0
- package/dist/db/SqlJsAdapter.js.map +1 -0
- package/dist/db/schema.d.ts +1 -1
- package/dist/db/schema.d.ts.map +1 -1
- package/dist/db/schema.js +30 -44
- package/dist/db/schema.js.map +1 -1
- package/dist/git/GitManager.d.ts +10 -0
- package/dist/git/GitManager.d.ts.map +1 -1
- package/dist/git/GitManager.js +46 -0
- package/dist/git/GitManager.js.map +1 -1
- package/dist/graph/GraphStore.d.ts +6 -1
- package/dist/graph/GraphStore.d.ts.map +1 -1
- package/dist/graph/InMemoryGraphStore.d.ts +1 -0
- package/dist/graph/InMemoryGraphStore.d.ts.map +1 -1
- package/dist/graph/InMemoryGraphStore.js +13 -2
- package/dist/graph/InMemoryGraphStore.js.map +1 -1
- package/dist/graph/SQLiteGraphStore.d.ts +4 -2
- package/dist/graph/SQLiteGraphStore.d.ts.map +1 -1
- package/dist/graph/SQLiteGraphStore.js +24 -2
- package/dist/graph/SQLiteGraphStore.js.map +1 -1
- package/dist/indexing/ASTParser.d.ts +1 -1
- package/dist/indexing/ASTParser.d.ts.map +1 -1
- package/dist/indexing/ASTParser.js +7 -1
- package/dist/indexing/ASTParser.js.map +1 -1
- package/dist/indexing/EmbeddingGenerator.d.ts.map +1 -1
- package/dist/indexing/EmbeddingGenerator.js +2 -25
- package/dist/indexing/EmbeddingGenerator.js.map +1 -1
- package/dist/indexing/FileScanner.d.ts.map +1 -1
- package/dist/indexing/FileScanner.js +2 -1
- package/dist/indexing/FileScanner.js.map +1 -1
- package/dist/indexing/GitIndexer.d.ts +4 -0
- package/dist/indexing/GitIndexer.d.ts.map +1 -0
- package/dist/indexing/GitIndexer.js +59 -0
- package/dist/indexing/GitIndexer.js.map +1 -0
- package/dist/indexing/GraphBuilder.d.ts.map +1 -1
- package/dist/indexing/GraphBuilder.js +21 -1
- package/dist/indexing/GraphBuilder.js.map +1 -1
- package/dist/indexing/Indexer.d.ts +3 -2
- package/dist/indexing/Indexer.d.ts.map +1 -1
- package/dist/indexing/Indexer.js +31 -9
- package/dist/indexing/Indexer.js.map +1 -1
- package/dist/mcp/server.d.ts.map +1 -1
- package/dist/mcp/server.js +22 -7
- package/dist/mcp/server.js.map +1 -1
- package/dist/mcp/tools/execute.d.ts +3 -0
- package/dist/mcp/tools/execute.d.ts.map +1 -1
- package/dist/mcp/tools/execute.js +2 -0
- package/dist/mcp/tools/execute.js.map +1 -1
- package/dist/mcp/tools/find_similar.d.ts +17 -0
- package/dist/mcp/tools/find_similar.d.ts.map +1 -0
- package/dist/mcp/tools/find_similar.js +95 -0
- package/dist/mcp/tools/find_similar.js.map +1 -0
- package/dist/mcp/tools/get_context.d.ts +3 -0
- package/dist/mcp/tools/get_context.d.ts.map +1 -1
- package/dist/mcp/tools/get_context.js +68 -7
- package/dist/mcp/tools/get_context.js.map +1 -1
- package/dist/mcp/tools/search_code.d.ts.map +1 -1
- package/dist/mcp/tools/search_code.js +22 -4
- package/dist/mcp/tools/search_code.js.map +1 -1
- package/dist/mcp/tools/smart_search.d.ts.map +1 -1
- package/dist/mcp/tools/smart_search.js +122 -29
- package/dist/mcp/tools/smart_search.js.map +1 -1
- package/dist/memory/HybridMemory.d.ts +4 -0
- package/dist/memory/HybridMemory.d.ts.map +1 -1
- package/dist/memory/HybridMemory.js +120 -29
- package/dist/memory/HybridMemory.js.map +1 -1
- package/dist/utils/gitHooks.d.ts +7 -0
- package/dist/utils/gitHooks.d.ts.map +1 -0
- package/dist/utils/gitHooks.js +77 -0
- package/dist/utils/gitHooks.js.map +1 -0
- package/dist/utils/modelManager.d.ts +37 -0
- package/dist/utils/modelManager.d.ts.map +1 -0
- package/dist/utils/modelManager.js +90 -0
- package/dist/utils/modelManager.js.map +1 -0
- package/dist/utils/workerPool.d.ts +8 -0
- package/dist/utils/workerPool.d.ts.map +1 -0
- package/dist/utils/workerPool.js +75 -0
- package/dist/utils/workerPool.js.map +1 -0
- package/dist/vector/MemoryVectorStore.d.ts.map +1 -1
- package/dist/vector/MemoryVectorStore.js +19 -13
- package/dist/vector/MemoryVectorStore.js.map +1 -1
- package/dist/vector/SQLiteVectorStore.d.ts +13 -0
- package/dist/vector/SQLiteVectorStore.d.ts.map +1 -0
- package/dist/vector/SQLiteVectorStore.js +111 -0
- package/dist/vector/SQLiteVectorStore.js.map +1 -0
- package/dist/vector/VectorStore.d.ts +0 -1
- package/dist/vector/VectorStore.d.ts.map +1 -1
- package/dist/workers/embeddingWorker.d.ts +2 -0
- package/dist/workers/embeddingWorker.d.ts.map +1 -0
- package/dist/workers/embeddingWorker.js +42 -0
- package/dist/workers/embeddingWorker.js.map +1 -0
- package/package.json +5 -3
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"modelManager.d.ts","sourceRoot":"","sources":["../../src/utils/modelManager.ts"],"names":[],"mappings":"AAKA,UAAU,cAAc;IACtB,IAAI,EAAE,YAAY,CAAC;IACnB,IAAI,EAAE,MAAM,EAAE,CAAC;CAChB;AAED,KAAK,aAAa,GAAG,CACnB,KAAK,EAAE,MAAM,EAAE,EACf,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,cAAc,CAAC,CAAC;AAE7B,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,UAAU,GAAG,UAAU,GAAG,UAAU,GAAG,MAAM,GAAG,OAAO,GAAG,SAAS,CAAC;CAC7E;AAED,MAAM,MAAM,gBAAgB,GAAG,CAAC,IAAI,EAAE,gBAAgB,KAAK,IAAI,CAAC;AAIhE,qEAAqE;AACrE,wBAAgB,eAAe,IAAI,OAAO,CAEzC;AAED;;;;GAIG;AACH,wBAAsB,WAAW,CAAC,UAAU,CAAC,EAAE,gBAAgB,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC,CAkC9F;AAED,wDAAwD;AACxD,wBAAgB,iBAAiB,IAAI,aAAa,GAAG,IAAI,CAExD;AAED,mCAAmC;AACnC,wBAAgB,aAAa,IAAI,IAAI,CAEpC;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,cAAc,GAAG,MAAM,EAAE,EAAE,CAOpE;AAED;;;GAGG;AACH,wBAAsB,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAUrE"}
|
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.isPipelineReady = isPipelineReady;
|
|
4
|
+
exports.ensureModel = ensureModel;
|
|
5
|
+
exports.getCachedPipeline = getCachedPipeline;
|
|
6
|
+
exports.resetPipeline = resetPipeline;
|
|
7
|
+
exports.extractEmbeddings = extractEmbeddings;
|
|
8
|
+
exports.embedTexts = embedTexts;
|
|
9
|
+
const logger_js_1 = require("./logger.js");
|
|
10
|
+
const workerPool_js_1 = require("./workerPool.js");
|
|
11
|
+
const MODEL_ID = 'Xenova/all-MiniLM-L6-v2';
|
|
12
|
+
let cachedPipeline = null;
|
|
13
|
+
/** True if the pipeline has been loaded into memory this process. */
|
|
14
|
+
function isPipelineReady() {
|
|
15
|
+
return cachedPipeline !== null;
|
|
16
|
+
}
|
|
17
|
+
/**
|
|
18
|
+
* Load (and optionally download) the embedding model.
|
|
19
|
+
* Returns the pipeline or null if loading fails.
|
|
20
|
+
* @param onProgress Called for each download/load progress event.
|
|
21
|
+
*/
|
|
22
|
+
async function ensureModel(onProgress) {
|
|
23
|
+
if (cachedPipeline)
|
|
24
|
+
return cachedPipeline;
|
|
25
|
+
try {
|
|
26
|
+
const { pipeline, env } = await import('@xenova/transformers');
|
|
27
|
+
// Allow overriding cache dir via env var
|
|
28
|
+
if (process.env.FIRECODE_MODEL_CACHE) {
|
|
29
|
+
env.cacheDir = process.env.FIRECODE_MODEL_CACHE;
|
|
30
|
+
}
|
|
31
|
+
const progressCb = onProgress
|
|
32
|
+
? (info) => {
|
|
33
|
+
onProgress({
|
|
34
|
+
file: String(info.file ?? info.name ?? ''),
|
|
35
|
+
progress: typeof info.progress === 'number' ? info.progress : 0,
|
|
36
|
+
loaded: typeof info.loaded === 'number' ? info.loaded : 0,
|
|
37
|
+
total: typeof info.total === 'number' ? info.total : 0,
|
|
38
|
+
status: info.status ?? 'progress',
|
|
39
|
+
});
|
|
40
|
+
}
|
|
41
|
+
: undefined;
|
|
42
|
+
const pipe = await pipeline('feature-extraction', MODEL_ID, {
|
|
43
|
+
progress_callback: progressCb,
|
|
44
|
+
});
|
|
45
|
+
cachedPipeline = pipe;
|
|
46
|
+
logger_js_1.logger.info({ model: MODEL_ID }, 'Embedding model loaded');
|
|
47
|
+
return cachedPipeline;
|
|
48
|
+
}
|
|
49
|
+
catch (err) {
|
|
50
|
+
logger_js_1.logger.warn({ err: String(err) }, 'Could not load embedding model — falling back to text similarity');
|
|
51
|
+
return null;
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
/** Get the cached pipeline (null if not yet loaded). */
|
|
55
|
+
function getCachedPipeline() {
|
|
56
|
+
return cachedPipeline;
|
|
57
|
+
}
|
|
58
|
+
/** Reset the cache (for tests). */
|
|
59
|
+
function resetPipeline() {
|
|
60
|
+
cachedPipeline = null;
|
|
61
|
+
}
|
|
62
|
+
/**
|
|
63
|
+
* Extract individual embedding vectors from a batched pipeline tensor.
|
|
64
|
+
* The tensor's `data` is a flat Float32Array of shape [batch, dim].
|
|
65
|
+
*/
|
|
66
|
+
function extractEmbeddings(tensor) {
|
|
67
|
+
const [batchSize, dim] = tensor.dims;
|
|
68
|
+
const result = [];
|
|
69
|
+
for (let i = 0; i < batchSize; i++) {
|
|
70
|
+
result.push(Array.from(tensor.data.slice(i * dim, (i + 1) * dim)));
|
|
71
|
+
}
|
|
72
|
+
return result;
|
|
73
|
+
}
|
|
74
|
+
/**
|
|
75
|
+
* Embed texts using a worker thread when available (production), falling back
|
|
76
|
+
* to the main-thread pipeline in dev/test environments.
|
|
77
|
+
*/
|
|
78
|
+
async function embedTexts(texts) {
|
|
79
|
+
// Try worker thread first — returns null in dev/test mode or after worker crash
|
|
80
|
+
const workerResult = await (0, workerPool_js_1.workerEmbed)(texts).catch(() => null);
|
|
81
|
+
if (workerResult !== null)
|
|
82
|
+
return workerResult;
|
|
83
|
+
// Main-thread fallback
|
|
84
|
+
const pipe = await ensureModel();
|
|
85
|
+
if (!pipe)
|
|
86
|
+
return texts.map(() => []);
|
|
87
|
+
const tensor = await pipe(texts, { pooling: 'mean', normalize: true });
|
|
88
|
+
return extractEmbeddings(tensor);
|
|
89
|
+
}
|
|
90
|
+
//# sourceMappingURL=modelManager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"modelManager.js","sourceRoot":"","sources":["../../src/utils/modelManager.ts"],"names":[],"mappings":";;AA4BA,0CAEC;AAOD,kCAkCC;AAGD,8CAEC;AAGD,sCAEC;AAMD,8CAOC;AAMD,gCAUC;AA9GD,2CAAqC;AACrC,mDAA8C;AAE9C,MAAM,QAAQ,GAAG,yBAAyB,CAAC;AAsB3C,IAAI,cAAc,GAAyB,IAAI,CAAC;AAEhD,qEAAqE;AACrE,SAAgB,eAAe;IAC7B,OAAO,cAAc,KAAK,IAAI,CAAC;AACjC,CAAC;AAED;;;;GAIG;AACI,KAAK,UAAU,WAAW,CAAC,UAA6B;IAC7D,IAAI,cAAc;QAAE,OAAO,cAAc,CAAC;IAE1C,IAAI,CAAC;QACH,MAAM,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;QAE/D,yCAAyC;QACzC,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,CAAC;YACrC,GAAG,CAAC,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;QAClD,CAAC;QAED,MAAM,UAAU,GAAG,UAAU;YAC3B,CAAC,CAAC,CAAC,IAA6B,EAAE,EAAE;gBAChC,UAAU,CAAC;oBACT,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;oBAC1C,QAAQ,EAAE,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;oBAC/D,MAAM,EAAE,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;oBACzD,KAAK,EAAE,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBACtD,MAAM,EAAG,IAAI,CAAC,MAAqC,IAAI,UAAU;iBAClE,CAAC,CAAC;YACL,CAAC;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,oBAAoB,EAAE,QAAQ,EAAE;YAC1D,iBAAiB,EAAE,UAAU;SAC9B,CAAC,CAAC;QAEH,cAAc,GAAG,IAAgC,CAAC;QAClD,kBAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,wBAAwB,CAAC,CAAC;QAC3D,OAAO,cAAc,CAAC;IACxB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,kBAAM,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,kEAAkE,CAAC,CAAC;QACtG,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,wDAAwD;AACxD,SAAgB,iBAAiB;IAC/B,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,mCAAmC;AACnC,SAAgB,aAAa;IAC3B,cAAc,GAAG,IAAI,CAAC;AACxB,CAAC;AAED;;;GAGG;AACH,SAAgB,iBAAiB,CAAC,MAAsB;IACtD,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC;IACrC,MAAM,MAAM,GAAe,EAAE,CAAC;IAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACI,KAAK,UAAU,UAAU,CAAC,KAAe;IAC9C,gFAAgF;IAChF,MAAM,YAAY,GAAG,MAAM,IAAA,2BAAW,EAAC,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IAChE,IAAI,YAAY,KAAK,IAAI;QAAE,OAAO,YAAY,CAAC;IAE/C,uBAAuB;IACvB,MAAM,IAAI,GAAG,MAAM,WAAW,EAAE,CAAC;IACjC,IAAI,CAAC,IAAI;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;IACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACvE,OAAO,iBAAiB,CAAC,MAAM,CAAC,CAAC;AACnC,CAAC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Send texts to the embedding worker thread.
|
|
3
|
+
* Returns null when the compiled worker file isn't present (dev/test mode or after a crash),
|
|
4
|
+
* signalling the caller to fall back to the main-thread pipeline.
|
|
5
|
+
*/
|
|
6
|
+
export declare function workerEmbed(texts: string[]): Promise<number[][] | null>;
|
|
7
|
+
export declare function terminateEmbeddingWorker(): void;
|
|
8
|
+
//# sourceMappingURL=workerPool.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"workerPool.d.ts","sourceRoot":"","sources":["../../src/utils/workerPool.ts"],"names":[],"mappings":"AAkEA;;;;GAIG;AACH,wBAAsB,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,GAAG,IAAI,CAAC,CAU7E;AAED,wBAAgB,wBAAwB,IAAI,IAAI,CAG/C"}
|
|
@@ -0,0 +1,75 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.workerEmbed = workerEmbed;
|
|
4
|
+
exports.terminateEmbeddingWorker = terminateEmbeddingWorker;
|
|
5
|
+
const worker_threads_1 = require("worker_threads");
|
|
6
|
+
const crypto_1 = require("crypto");
|
|
7
|
+
const path_1 = require("path");
|
|
8
|
+
const fs_1 = require("fs");
|
|
9
|
+
const logger_js_1 = require("./logger.js");
|
|
10
|
+
// __dirname is available in CommonJS (this project builds to CJS output).
|
|
11
|
+
// At runtime: dist/utils/workerPool.js → dist/workers/embeddingWorker.js
|
|
12
|
+
// In dev/test: src/utils/workerPool.ts → src/workers/embeddingWorker.js (won't exist → fallback)
|
|
13
|
+
const WORKER_PATH = (0, path_1.join)(__dirname, '../workers/embeddingWorker.js');
|
|
14
|
+
let worker = null;
|
|
15
|
+
let workerFailed = false;
|
|
16
|
+
const pending = new Map();
|
|
17
|
+
function spawnWorker() {
|
|
18
|
+
const w = new worker_threads_1.Worker(WORKER_PATH);
|
|
19
|
+
w.on('message', (msg) => {
|
|
20
|
+
if (msg.type === 'ready') {
|
|
21
|
+
logger_js_1.logger.debug('Embedding worker ready');
|
|
22
|
+
return;
|
|
23
|
+
}
|
|
24
|
+
if (msg.type === 'error') {
|
|
25
|
+
workerFailed = true;
|
|
26
|
+
worker?.terminate();
|
|
27
|
+
worker = null;
|
|
28
|
+
logger_js_1.logger.warn({ err: msg.error }, 'Embedding worker failed to load — falling back to main thread');
|
|
29
|
+
for (const { reject } of pending.values())
|
|
30
|
+
reject(new Error(msg.error ?? 'Worker error'));
|
|
31
|
+
pending.clear();
|
|
32
|
+
return;
|
|
33
|
+
}
|
|
34
|
+
if (msg.id) {
|
|
35
|
+
const entry = pending.get(msg.id);
|
|
36
|
+
if (entry) {
|
|
37
|
+
pending.delete(msg.id);
|
|
38
|
+
if (msg.error)
|
|
39
|
+
entry.reject(new Error(msg.error));
|
|
40
|
+
else
|
|
41
|
+
entry.resolve(msg.embeddings ?? []);
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
});
|
|
45
|
+
w.on('error', (err) => {
|
|
46
|
+
workerFailed = true;
|
|
47
|
+
logger_js_1.logger.warn({ err: String(err) }, 'Embedding worker crashed');
|
|
48
|
+
for (const { reject } of pending.values())
|
|
49
|
+
reject(err);
|
|
50
|
+
pending.clear();
|
|
51
|
+
worker = null;
|
|
52
|
+
});
|
|
53
|
+
return w;
|
|
54
|
+
}
|
|
55
|
+
/**
|
|
56
|
+
* Send texts to the embedding worker thread.
|
|
57
|
+
* Returns null when the compiled worker file isn't present (dev/test mode or after a crash),
|
|
58
|
+
* signalling the caller to fall back to the main-thread pipeline.
|
|
59
|
+
*/
|
|
60
|
+
async function workerEmbed(texts) {
|
|
61
|
+
if (workerFailed || !(0, fs_1.existsSync)(WORKER_PATH))
|
|
62
|
+
return null;
|
|
63
|
+
if (!worker)
|
|
64
|
+
worker = spawnWorker();
|
|
65
|
+
return new Promise((resolve, reject) => {
|
|
66
|
+
const id = (0, crypto_1.randomUUID)();
|
|
67
|
+
pending.set(id, { resolve, reject });
|
|
68
|
+
worker.postMessage({ id, texts });
|
|
69
|
+
});
|
|
70
|
+
}
|
|
71
|
+
function terminateEmbeddingWorker() {
|
|
72
|
+
worker?.terminate();
|
|
73
|
+
worker = null;
|
|
74
|
+
}
|
|
75
|
+
//# sourceMappingURL=workerPool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"workerPool.js","sourceRoot":"","sources":["../../src/utils/workerPool.ts"],"names":[],"mappings":";;AAuEA,kCAUC;AAED,4DAGC;AAtFD,mDAAwC;AACxC,mCAAoC;AACpC,+BAA4B;AAC5B,2BAAgC;AAChC,2CAAqC;AAErC,0EAA0E;AAC1E,yEAAyE;AACzE,iGAAiG;AACjG,MAAM,WAAW,GAAG,IAAA,WAAI,EAAC,SAAS,EAAE,+BAA+B,CAAC,CAAC;AAerE,IAAI,MAAM,GAAkB,IAAI,CAAC;AACjC,IAAI,YAAY,GAAG,KAAK,CAAC;AACzB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAwB,CAAC;AAEhD,SAAS,WAAW;IAClB,MAAM,CAAC,GAAG,IAAI,uBAAM,CAAC,WAAW,CAAC,CAAC;IAElC,CAAC,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAoB,EAAE,EAAE;QACvC,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YACzB,kBAAM,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;YACvC,OAAO;QACT,CAAC;QACD,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YACzB,YAAY,GAAG,IAAI,CAAC;YACpB,MAAM,EAAE,SAAS,EAAE,CAAC;YACpB,MAAM,GAAG,IAAI,CAAC;YACd,kBAAM,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,KAAK,EAAE,EAAE,+DAA+D,CAAC,CAAC;YACjG,KAAK,MAAM,EAAE,MAAM,EAAE,IAAI,OAAO,CAAC,MAAM,EAAE;gBAAE,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,cAAc,CAAC,CAAC,CAAC;YAC1F,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QACD,IAAI,GAAG,CAAC,EAAE,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAClC,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACvB,IAAI,GAAG,CAAC,KAAK;oBAAE,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;;oBAC7C,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;QACpB,YAAY,GAAG,IAAI,CAAC;QACpB,kBAAM,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,0BAA0B,CAAC,CAAC;QAC9D,KAAK,MAAM,EAAE,MAAM,EAAE,IAAI,OAAO,CAAC,MAAM,EAAE;YAAE,MAAM,CAAC,GAAG,CAAC,CAAC;QACvD,OAAO,CAAC,KAAK,EAAE,CAAC;QAChB,MAAM,GAAG,IAAI,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,CAAC;AACX,CAAC;AAED;;;;GAIG;AACI,KAAK,UAAU,WAAW,CAAC,KAAe;IAC/C,IAAI,YAAY,IAAI,CAAC,IAAA,eAAU,EAAC,WAAW,CAAC;QAAE,OAAO,IAAI,CAAC;IAE1D,IAAI,CAAC,MAAM;QAAE,MAAM,GAAG,WAAW,EAAE,CAAC;IAEpC,OAAO,IAAI,OAAO,CAAa,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACjD,MAAM,EAAE,GAAG,IAAA,mBAAU,GAAE,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QACrC,MAAO,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,KAAK,EAAkB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAgB,wBAAwB;IACtC,MAAM,EAAE,SAAS,EAAE,CAAC;IACpB,MAAM,GAAG,IAAI,CAAC;AAChB,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"MemoryVectorStore.d.ts","sourceRoot":"","sources":["../../src/vector/MemoryVectorStore.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;
|
|
1
|
+
{"version":3,"file":"MemoryVectorStore.d.ts","sourceRoot":"","sources":["../../src/vector/MemoryVectorStore.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAoC9E,qBAAa,iBAAkB,YAAW,WAAW;IACnD,OAAO,CAAC,SAAS,CAAoC;IACrD,OAAO,CAAC,aAAa,CAAQ;gBAEjB,IAAI,CAAC,EAAE;QAAE,aAAa,CAAC,EAAE,OAAO,CAAA;KAAE;IAIxC,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAOzC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,SAAI,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IA+BvD,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5B,IAAI,IAAI,MAAM;IAId,SAAS,IAAI,MAAM;IAInB,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,iBAAiB;CAMpD"}
|
|
@@ -2,19 +2,17 @@
|
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.MemoryVectorStore = void 0;
|
|
4
4
|
const logger_js_1 = require("../utils/logger.js");
|
|
5
|
-
|
|
5
|
+
const modelManager_js_1 = require("../utils/modelManager.js");
|
|
6
6
|
async function getEmbedding(text) {
|
|
7
7
|
try {
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
}
|
|
12
|
-
|
|
13
|
-
return Array.from(output.data[0]);
|
|
8
|
+
const pipe = await (0, modelManager_js_1.ensureModel)();
|
|
9
|
+
if (!pipe)
|
|
10
|
+
return [];
|
|
11
|
+
const tensor = await pipe([text], { pooling: 'mean', normalize: true });
|
|
12
|
+
return (0, modelManager_js_1.extractEmbeddings)(tensor)[0] ?? [];
|
|
14
13
|
}
|
|
15
14
|
catch {
|
|
16
|
-
|
|
17
|
-
return new Array(384).fill(0);
|
|
15
|
+
return [];
|
|
18
16
|
}
|
|
19
17
|
}
|
|
20
18
|
function cosineSimilarity(a, b) {
|
|
@@ -59,10 +57,18 @@ class MemoryVectorStore {
|
|
|
59
57
|
let scored;
|
|
60
58
|
if (this.useEmbeddings && docsWithEmbeddings.length > 0) {
|
|
61
59
|
const queryEmbedding = await getEmbedding(query);
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
60
|
+
if (queryEmbedding.length > 0) {
|
|
61
|
+
scored = docsWithEmbeddings.map((doc) => ({
|
|
62
|
+
document: doc,
|
|
63
|
+
score: cosineSimilarity(queryEmbedding, doc.embedding),
|
|
64
|
+
}));
|
|
65
|
+
}
|
|
66
|
+
else {
|
|
67
|
+
scored = docs.map((doc) => ({
|
|
68
|
+
document: doc,
|
|
69
|
+
score: textSimilarity(query, doc.text),
|
|
70
|
+
}));
|
|
71
|
+
}
|
|
66
72
|
}
|
|
67
73
|
else {
|
|
68
74
|
scored = docs.map((doc) => ({
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"MemoryVectorStore.js","sourceRoot":"","sources":["../../src/vector/MemoryVectorStore.ts"],"names":[],"mappings":";;;AACA,kDAA4C;
|
|
1
|
+
{"version":3,"file":"MemoryVectorStore.js","sourceRoot":"","sources":["../../src/vector/MemoryVectorStore.ts"],"names":[],"mappings":";;;AACA,kDAA4C;AAC5C,8DAA0E;AAE1E,KAAK,UAAU,YAAY,CAAC,IAAY;IACtC,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,IAAA,6BAAW,GAAE,CAAC;QACjC,IAAI,CAAC,IAAI;YAAE,OAAO,EAAE,CAAC;QACrB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxE,OAAO,IAAA,mCAAiB,EAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAC5C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,CAAW,EAAE,CAAW;IAChD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM;QAAE,OAAO,CAAC,CAAC;IACpC,IAAI,GAAG,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,KAAK,CAAC;AACvC,CAAC;AAED,oDAAoD;AACpD,SAAS,cAAc,CAAC,KAAa,EAAE,IAAY;IACjD,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IAC7E,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClE,IAAI,UAAU,CAAC,IAAI,KAAK,CAAC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAC9D,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAClE,OAAO,OAAO,GAAG,CAAC,UAAU,CAAC,IAAI,GAAG,SAAS,CAAC,MAAM,GAAG,OAAO,CAAC,CAAC;AAClE,CAAC;AAED,MAAa,iBAAiB;IACpB,SAAS,GAA0B,IAAI,GAAG,EAAE,CAAC;IAC7C,aAAa,GAAG,IAAI,CAAC;IAE7B,YAAY,IAAkC;QAC5C,IAAI,IAAI,EAAE,aAAa,KAAK,KAAK;YAAE,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,SAAqB;QAC7B,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;YAC5B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QAClC,CAAC;QACD,kBAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,CAAC,MAAM,EAAE,EAAE,iCAAiC,CAAC,CAAC;IAC/E,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,CAAC,GAAG,CAAC;QAC/B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;QACjD,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEjC,MAAM,kBAAkB,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAErF,IAAI,MAAwB,CAAC;QAE7B,IAAI,IAAI,CAAC,aAAa,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,MAAM,cAAc,GAAG,MAAM,YAAY,CAAC,KAAK,CAAC,CAAC;YACjD,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9B,MAAM,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBACxC,QAAQ,EAAE,GAAG;oBACb,KAAK,EAAE,gBAAgB,CAAC,cAAc,EAAE,GAAG,CAAC,SAAU,CAAC;iBACxD,CAAC,CAAC,CAAC;YACN,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC1B,QAAQ,EAAE,GAAG;oBACb,KAAK,EAAE,cAAc,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC;iBACvC,CAAC,CAAC,CAAC;YACN,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC1B,QAAQ,EAAE,GAAG;gBACb,KAAK,EAAE,cAAc,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC;aACvC,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAa;QACxB,KAAK,MAAM,EAAE,IAAI,GAAG;YAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,IAAI;QACF,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;IAC7B,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC7D,CAAC;IAED,MAAM,CAAC,WAAW,CAAC,IAAY;QAC7B,MAAM,KAAK,GAAG,IAAI,iBAAiB,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAe,CAAC;QAC5C,KAAK,MAAM,GAAG,IAAI,IAAI;YAAE,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QACzD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AApED,8CAoEC"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import { SqlJsDatabase } from '../db/SqlJsAdapter.js';
|
|
2
|
+
import type { Document, ScoredDocument, VectorStore } from './VectorStore.js';
|
|
3
|
+
export declare class SQLiteVectorStore implements VectorStore {
|
|
4
|
+
private db;
|
|
5
|
+
private project;
|
|
6
|
+
constructor(db: SqlJsDatabase, project: string);
|
|
7
|
+
add(documents: Document[]): Promise<void>;
|
|
8
|
+
search(query: string, k?: number): Promise<ScoredDocument[]>;
|
|
9
|
+
delete(ids: string[]): Promise<void>;
|
|
10
|
+
clear(): Promise<void>;
|
|
11
|
+
size(): number;
|
|
12
|
+
}
|
|
13
|
+
//# sourceMappingURL=SQLiteVectorStore.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"SQLiteVectorStore.d.ts","sourceRoot":"","sources":["../../src/vector/SQLiteVectorStore.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAwC9E,qBAAa,iBAAkB,YAAW,WAAW;IACvC,OAAO,CAAC,EAAE;IAAiB,OAAO,CAAC,OAAO;gBAAlC,EAAE,EAAE,aAAa,EAAU,OAAO,EAAE,MAAM;IAExD,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAoBzC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,SAAI,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IA+CvD,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAMpC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5B,IAAI,IAAI,MAAM;CAKf"}
|
|
@@ -0,0 +1,111 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.SQLiteVectorStore = void 0;
|
|
4
|
+
const modelManager_js_1 = require("../utils/modelManager.js");
|
|
5
|
+
const logger_js_1 = require("../utils/logger.js");
|
|
6
|
+
async function getEmbedding(text) {
|
|
7
|
+
try {
|
|
8
|
+
const embeddings = await (0, modelManager_js_1.embedTexts)([text]);
|
|
9
|
+
return embeddings[0] ?? [];
|
|
10
|
+
}
|
|
11
|
+
catch {
|
|
12
|
+
return [];
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
function cosineSimilarity(a, b) {
|
|
16
|
+
if (a.length !== b.length || a.length === 0)
|
|
17
|
+
return 0;
|
|
18
|
+
let dot = 0, normA = 0, normB = 0;
|
|
19
|
+
for (let i = 0; i < a.length; i++) {
|
|
20
|
+
dot += a[i] * b[i];
|
|
21
|
+
normA += a[i] * a[i];
|
|
22
|
+
normB += b[i] * b[i];
|
|
23
|
+
}
|
|
24
|
+
const denom = Math.sqrt(normA) * Math.sqrt(normB);
|
|
25
|
+
return denom === 0 ? 0 : dot / denom;
|
|
26
|
+
}
|
|
27
|
+
function textSimilarity(query, text) {
|
|
28
|
+
const queryTerms = new Set(query.toLowerCase().split(/\W+/).filter(Boolean));
|
|
29
|
+
const textTerms = text.toLowerCase().split(/\W+/).filter(Boolean);
|
|
30
|
+
if (queryTerms.size === 0 || textTerms.length === 0)
|
|
31
|
+
return 0;
|
|
32
|
+
const matches = textTerms.filter(t => queryTerms.has(t)).length;
|
|
33
|
+
return matches / (queryTerms.size + textTerms.length - matches);
|
|
34
|
+
}
|
|
35
|
+
class SQLiteVectorStore {
|
|
36
|
+
db;
|
|
37
|
+
project;
|
|
38
|
+
constructor(db, project) {
|
|
39
|
+
this.db = db;
|
|
40
|
+
this.project = project;
|
|
41
|
+
}
|
|
42
|
+
async add(documents) {
|
|
43
|
+
const insert = this.db.prepare(`
|
|
44
|
+
INSERT OR REPLACE INTO vector_chunks (id, project, text, metadata, embedding)
|
|
45
|
+
VALUES (?, ?, ?, ?, ?)
|
|
46
|
+
`);
|
|
47
|
+
const insertMany = this.db.transaction((docs) => {
|
|
48
|
+
for (const doc of docs) {
|
|
49
|
+
insert.run(doc.id, this.project, doc.text, JSON.stringify(doc.metadata), doc.embedding && doc.embedding.length > 0 ? JSON.stringify(doc.embedding) : null);
|
|
50
|
+
}
|
|
51
|
+
});
|
|
52
|
+
insertMany(documents);
|
|
53
|
+
logger_js_1.logger.debug({ count: documents.length, project: this.project }, 'Vector chunks upserted to SQLite');
|
|
54
|
+
}
|
|
55
|
+
async search(query, k = 5) {
|
|
56
|
+
const rows = this.db.prepare(`SELECT id, text, metadata, embedding FROM vector_chunks WHERE project = ?`).all(this.project);
|
|
57
|
+
if (rows.length === 0)
|
|
58
|
+
return [];
|
|
59
|
+
const withEmbeddings = rows.filter(r => r.embedding !== null);
|
|
60
|
+
let scored;
|
|
61
|
+
if (withEmbeddings.length > 0) {
|
|
62
|
+
const queryEmbedding = await getEmbedding(query);
|
|
63
|
+
if (queryEmbedding.length > 0) {
|
|
64
|
+
scored = withEmbeddings.map(row => ({
|
|
65
|
+
document: {
|
|
66
|
+
id: row.id,
|
|
67
|
+
text: row.text,
|
|
68
|
+
metadata: JSON.parse(row.metadata),
|
|
69
|
+
embedding: JSON.parse(row.embedding),
|
|
70
|
+
},
|
|
71
|
+
score: cosineSimilarity(queryEmbedding, JSON.parse(row.embedding)),
|
|
72
|
+
}));
|
|
73
|
+
}
|
|
74
|
+
else {
|
|
75
|
+
scored = rows.map(row => ({
|
|
76
|
+
document: {
|
|
77
|
+
id: row.id,
|
|
78
|
+
text: row.text,
|
|
79
|
+
metadata: JSON.parse(row.metadata),
|
|
80
|
+
},
|
|
81
|
+
score: textSimilarity(query, row.text),
|
|
82
|
+
}));
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
else {
|
|
86
|
+
scored = rows.map(row => ({
|
|
87
|
+
document: {
|
|
88
|
+
id: row.id,
|
|
89
|
+
text: row.text,
|
|
90
|
+
metadata: JSON.parse(row.metadata),
|
|
91
|
+
},
|
|
92
|
+
score: textSimilarity(query, row.text),
|
|
93
|
+
}));
|
|
94
|
+
}
|
|
95
|
+
return scored.sort((a, b) => b.score - a.score).slice(0, k);
|
|
96
|
+
}
|
|
97
|
+
async delete(ids) {
|
|
98
|
+
if (ids.length === 0)
|
|
99
|
+
return;
|
|
100
|
+
const ph = ids.map(() => '?').join(',');
|
|
101
|
+
this.db.prepare(`DELETE FROM vector_chunks WHERE project = ? AND id IN (${ph})`).run(this.project, ...ids);
|
|
102
|
+
}
|
|
103
|
+
async clear() {
|
|
104
|
+
this.db.prepare(`DELETE FROM vector_chunks WHERE project = ?`).run(this.project);
|
|
105
|
+
}
|
|
106
|
+
size() {
|
|
107
|
+
return this.db.prepare(`SELECT COUNT(*) as c FROM vector_chunks WHERE project = ?`).get(this.project).c;
|
|
108
|
+
}
|
|
109
|
+
}
|
|
110
|
+
exports.SQLiteVectorStore = SQLiteVectorStore;
|
|
111
|
+
//# sourceMappingURL=SQLiteVectorStore.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"SQLiteVectorStore.js","sourceRoot":"","sources":["../../src/vector/SQLiteVectorStore.ts"],"names":[],"mappings":";;;AAEA,8DAAsD;AACtD,kDAA4C;AAE5C,KAAK,UAAU,YAAY,CAAC,IAAY;IACtC,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,IAAA,4BAAU,EAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5C,OAAO,UAAU,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,CAAW,EAAE,CAAW;IAChD,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IACtD,IAAI,GAAG,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,KAAK,CAAC;AACvC,CAAC;AAED,SAAS,cAAc,CAAC,KAAa,EAAE,IAAY;IACjD,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IAC7E,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClE,IAAI,UAAU,CAAC,IAAI,KAAK,CAAC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAC9D,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAChE,OAAO,OAAO,GAAG,CAAC,UAAU,CAAC,IAAI,GAAG,SAAS,CAAC,MAAM,GAAG,OAAO,CAAC,CAAC;AAClE,CAAC;AASD,MAAa,iBAAiB;IACR;IAA2B;IAA/C,YAAoB,EAAiB,EAAU,OAAe;QAA1C,OAAE,GAAF,EAAE,CAAe;QAAU,YAAO,GAAP,OAAO,CAAQ;IAAG,CAAC;IAElE,KAAK,CAAC,GAAG,CAAC,SAAqB;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;KAG9B,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,IAAgB,EAAE,EAAE;YAC1D,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,MAAM,CAAC,GAAG,CACR,GAAG,CAAC,EAAE,EACN,IAAI,CAAC,OAAO,EACZ,GAAG,CAAC,IAAI,EACR,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,EAC5B,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CACjF,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,SAAS,CAAC,CAAC;QACtB,kBAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,kCAAkC,CAAC,CAAC;IACvG,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,CAAC,GAAG,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAC1B,2EAA2E,CAC5E,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAgB,CAAC;QAEnC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEjC,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC;QAE9D,IAAI,MAAwB,CAAC;QAE7B,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,cAAc,GAAG,MAAM,YAAY,CAAC,KAAK,CAAC,CAAC;YACjD,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9B,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBAClC,QAAQ,EAAE;wBACR,EAAE,EAAE,GAAG,CAAC,EAAE;wBACV,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAA4B;wBAC7D,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAU,CAAa;qBAClD;oBACD,KAAK,EAAE,gBAAgB,CAAC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAU,CAAa,CAAC;iBAChF,CAAC,CAAC,CAAC;YACN,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBACxB,QAAQ,EAAE;wBACR,EAAE,EAAE,GAAG,CAAC,EAAE;wBACV,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAA4B;qBAC9D;oBACD,KAAK,EAAE,cAAc,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC;iBACvC,CAAC,CAAC,CAAC;YACN,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACxB,QAAQ,EAAE;oBACR,EAAE,EAAE,GAAG,CAAC,EAAE;oBACV,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAA4B;iBAC9D;gBACD,KAAK,EAAE,cAAc,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC;aACvC,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,GAAa;QACxB,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAC7B,MAAM,EAAE,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,0DAA0D,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,GAAG,CAAC,CAAC;IAC7G,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACnF,CAAC;IAED,IAAI;QACF,OAAQ,IAAI,CAAC,EAAE,CAAC,OAAO,CACrB,2DAA2D,CAC5D,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAmB,CAAC,CAAC,CAAC;IAC1C,CAAC;CACF;AArFD,8CAqFC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"VectorStore.d.ts","sourceRoot":"","sources":["../../src/vector/VectorStore.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACtB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;IAC7D,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACrC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,MAAM,CAAC;
|
|
1
|
+
{"version":3,"file":"VectorStore.d.ts","sourceRoot":"","sources":["../../src/vector/VectorStore.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;CACtB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;IAC7D,MAAM,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACrC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,MAAM,CAAC;CAChB"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embeddingWorker.d.ts","sourceRoot":"","sources":["../../src/workers/embeddingWorker.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const worker_threads_1 = require("worker_threads");
|
|
4
|
+
const MODEL_ID = 'Xenova/all-MiniLM-L6-v2';
|
|
5
|
+
let pipe = null;
|
|
6
|
+
async function init() {
|
|
7
|
+
try {
|
|
8
|
+
const { pipeline, env } = await import('@xenova/transformers');
|
|
9
|
+
if (process.env.FIRECODE_MODEL_CACHE) {
|
|
10
|
+
env.cacheDir = process.env.FIRECODE_MODEL_CACHE;
|
|
11
|
+
}
|
|
12
|
+
pipe = (await pipeline('feature-extraction', MODEL_ID));
|
|
13
|
+
worker_threads_1.parentPort.postMessage({ type: 'ready' });
|
|
14
|
+
}
|
|
15
|
+
catch (err) {
|
|
16
|
+
worker_threads_1.parentPort.postMessage({ type: 'error', error: String(err) });
|
|
17
|
+
}
|
|
18
|
+
}
|
|
19
|
+
worker_threads_1.parentPort.on('message', async (req) => {
|
|
20
|
+
if (!pipe) {
|
|
21
|
+
worker_threads_1.parentPort.postMessage({
|
|
22
|
+
id: req.id,
|
|
23
|
+
embeddings: [],
|
|
24
|
+
error: 'Model not ready',
|
|
25
|
+
});
|
|
26
|
+
return;
|
|
27
|
+
}
|
|
28
|
+
try {
|
|
29
|
+
const tensor = await pipe(req.texts, { pooling: 'mean', normalize: true });
|
|
30
|
+
const [batch, dim] = tensor.dims;
|
|
31
|
+
const embeddings = [];
|
|
32
|
+
for (let i = 0; i < batch; i++) {
|
|
33
|
+
embeddings.push(Array.from(tensor.data.slice(i * dim, (i + 1) * dim)));
|
|
34
|
+
}
|
|
35
|
+
worker_threads_1.parentPort.postMessage({ id: req.id, embeddings });
|
|
36
|
+
}
|
|
37
|
+
catch (err) {
|
|
38
|
+
worker_threads_1.parentPort.postMessage({ id: req.id, embeddings: [], error: String(err) });
|
|
39
|
+
}
|
|
40
|
+
});
|
|
41
|
+
init();
|
|
42
|
+
//# sourceMappingURL=embeddingWorker.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"embeddingWorker.js","sourceRoot":"","sources":["../../src/workers/embeddingWorker.ts"],"names":[],"mappings":";;AAAA,mDAA4C;AAgB5C,MAAM,QAAQ,GAAG,yBAAyB,CAAC;AAO3C,IAAI,IAAI,GAAoB,IAAI,CAAC;AAEjC,KAAK,UAAU,IAAI;IACjB,IAAI,CAAC;QACH,MAAM,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;QAC/D,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,CAAC;YACpC,GAA6B,CAAC,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;QAC7E,CAAC;QACD,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,oBAAoB,EAAE,QAAQ,CAAC,CAAwB,CAAC;QAC/E,2BAAW,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC7C,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,2BAAW,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;AACH,CAAC;AAED,2BAAW,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,GAAiB,EAAE,EAAE;IACpD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,2BAAW,CAAC,WAAW,CAAC;YACtB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,UAAU,EAAE,EAAE;YACd,KAAK,EAAE,iBAAiB;SACR,CAAC,CAAC;QACpB,OAAO;IACT,CAAC;IACD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3E,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC;QACjC,MAAM,UAAU,GAAe,EAAE,CAAC;QAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;QACzE,CAAC;QACD,2BAAW,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,UAAU,EAAmB,CAAC,CAAC;IACvE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,2BAAW,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAmB,CAAC,CAAC;IAC/F,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,IAAI,EAAE,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@phmotad/fire-code",
|
|
3
|
-
"version": "0.
|
|
3
|
+
"version": "0.3.0",
|
|
4
4
|
"description": "Intelligent MCP execution engine for AI coding agents — persistent memory, context-aware, Git-traceable",
|
|
5
5
|
"keywords": [
|
|
6
6
|
"mcp",
|
|
@@ -55,6 +55,8 @@
|
|
|
55
55
|
"test": "jest",
|
|
56
56
|
"test:unit": "jest --testPathPattern=__tests__/unit",
|
|
57
57
|
"test:integration": "jest --testPathPattern=__tests__/integration",
|
|
58
|
+
"test:bench": "jest --testPathPattern=__tests__/benchmark --testMatch='**/__tests__/**/*.bench.ts' --verbose --testTimeout=30000 --runInBand",
|
|
59
|
+
"bench:emb": "tsx scripts/benchmark-embeddings.ts",
|
|
58
60
|
"test:coverage": "jest --coverage",
|
|
59
61
|
"lint": "tsc --noEmit",
|
|
60
62
|
"prepublishOnly": "npm run build && npm test",
|
|
@@ -64,11 +66,9 @@
|
|
|
64
66
|
"@anthropic-ai/claude-agent-sdk": "^0.2.114",
|
|
65
67
|
"@anthropic-ai/sdk": "^0.30.0",
|
|
66
68
|
"@modelcontextprotocol/sdk": "^1.0.0",
|
|
67
|
-
"@types/better-sqlite3": "^7.6.13",
|
|
68
69
|
"@types/express": "^5.0.6",
|
|
69
70
|
"@xenova/transformers": "^2.17.2",
|
|
70
71
|
"axios": "^1.7.0",
|
|
71
|
-
"better-sqlite3": "^12.9.0",
|
|
72
72
|
"chalk": "^5.3.0",
|
|
73
73
|
"commander": "^12.1.0",
|
|
74
74
|
"express": "^5.2.1",
|
|
@@ -84,6 +84,7 @@
|
|
|
84
84
|
"pino": "^9.5.0",
|
|
85
85
|
"pino-pretty": "^11.3.0",
|
|
86
86
|
"simple-git": "^3.27.0",
|
|
87
|
+
"sql.js": "^1.14.1",
|
|
87
88
|
"tree-sitter": "^0.25.0",
|
|
88
89
|
"tree-sitter-go": "^0.25.0",
|
|
89
90
|
"tree-sitter-javascript": "^0.25.0",
|
|
@@ -98,6 +99,7 @@
|
|
|
98
99
|
"@types/inquirer": "^9.0.0",
|
|
99
100
|
"@types/jest": "^29.5.0",
|
|
100
101
|
"@types/node": "^20.0.0",
|
|
102
|
+
"@types/sql.js": "^1.4.11",
|
|
101
103
|
"cross-env": "^10.1.0",
|
|
102
104
|
"jest": "^29.7.0",
|
|
103
105
|
"ts-jest": "^29.2.0",
|