@vivantel/virage-core 0.2.2 → 0.2.4
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/evaluate.d.ts.map +1 -1
- package/dist/cli/evaluate.js +4 -1
- package/dist/cli/evaluate.js.map +1 -1
- package/dist/cli/experiment.d.ts.map +1 -1
- package/dist/cli/experiment.js +4 -1
- package/dist/cli/experiment.js.map +1 -1
- package/dist/cli/file-detect.js +1 -1
- package/dist/cli/file-detect.js.map +1 -1
- package/dist/cli/init.d.ts.map +1 -1
- package/dist/cli/init.js +69 -10
- package/dist/cli/init.js.map +1 -1
- package/dist/cli/validate.d.ts.map +1 -1
- package/dist/cli/validate.js.map +1 -1
- package/dist/core/chunk-processor.d.ts +1 -1
- package/dist/core/chunk-processor.d.ts.map +1 -1
- package/dist/core/chunk-processor.js +3 -1
- package/dist/core/chunk-processor.js.map +1 -1
- package/dist/core/embedder.d.ts +11 -4
- package/dist/core/embedder.d.ts.map +1 -1
- package/dist/core/embedder.js +73 -97
- package/dist/core/embedder.js.map +1 -1
- package/dist/core/embeddings-db.d.ts +20 -0
- package/dist/core/embeddings-db.d.ts.map +1 -0
- package/dist/core/embeddings-db.js +181 -0
- package/dist/core/embeddings-db.js.map +1 -0
- package/dist/core/orchestrator.d.ts +1 -0
- package/dist/core/orchestrator.d.ts.map +1 -1
- package/dist/core/orchestrator.js +34 -14
- package/dist/core/orchestrator.js.map +1 -1
- package/dist/core/uploader.d.ts +8 -2
- package/dist/core/uploader.d.ts.map +1 -1
- package/dist/core/uploader.js +41 -30
- package/dist/core/uploader.js.map +1 -1
- package/dist/core/utils.d.ts +2 -0
- package/dist/core/utils.d.ts.map +1 -1
- package/dist/core/utils.js +24 -0
- package/dist/core/utils.js.map +1 -1
- package/dist/eval/runner.d.ts +1 -1
- package/dist/eval/runner.d.ts.map +1 -1
- package/dist/eval/runner.js +4 -2
- package/dist/eval/runner.js.map +1 -1
- package/dist/logger/consola-logger.d.ts.map +1 -1
- package/dist/logger/consola-logger.js +12 -2
- package/dist/logger/consola-logger.js.map +1 -1
- package/dist/plugin-registry.d.ts.map +1 -1
- package/dist/plugin-registry.js +0 -13
- package/dist/plugin-registry.js.map +1 -1
- package/dist/progress/progress-bar.d.ts +6 -0
- package/dist/progress/progress-bar.d.ts.map +1 -0
- package/dist/progress/progress-bar.js +23 -0
- package/dist/progress/progress-bar.js.map +1 -0
- package/package.json +5 -1
- package/schemas/virage.config.schema.json +5 -0
|
@@ -2,15 +2,15 @@ import { GitTracker } from "./git-tracker.js";
|
|
|
2
2
|
import { ChunkProcessor } from "./chunk-processor.js";
|
|
3
3
|
import { EmbedderProcessor } from "./embedder.js";
|
|
4
4
|
import { Uploader } from "./uploader.js";
|
|
5
|
+
import { EmbeddingsDb } from "./embeddings-db.js";
|
|
5
6
|
import { TelemetryCollector } from "./telemetry.js";
|
|
7
|
+
import { createProgressBar, } from "../progress/progress-bar.js";
|
|
6
8
|
import { NullLogger } from "../logger/null-logger.js";
|
|
7
9
|
import { readFile } from "fs/promises";
|
|
8
|
-
import { readEmbeddingsFile } from "./embeddings-io.js";
|
|
9
10
|
import { defaultChunksFile, defaultEmbeddingsFile } from "./virage-defaults.js";
|
|
10
|
-
|
|
11
|
-
const { chunks } = await readEmbeddingsFile(embeddingsFile);
|
|
11
|
+
function loadPreviousState(db) {
|
|
12
12
|
const state = new Map();
|
|
13
|
-
for (const chunk of
|
|
13
|
+
for (const chunk of db.getAll()) {
|
|
14
14
|
if (chunk.sourceFile && chunk.commitHash) {
|
|
15
15
|
state.set(chunk.sourceFile, chunk.commitHash);
|
|
16
16
|
}
|
|
@@ -54,6 +54,8 @@ export class Orchestrator {
|
|
|
54
54
|
const logger = (opts.logger ?? new NullLogger()).withTag("orchestrator");
|
|
55
55
|
const telemetry = opts.telemetry ? new TelemetryCollector() : null;
|
|
56
56
|
telemetry?.start();
|
|
57
|
+
const dbPath = this.embeddingsFile.replace(/\.json$/, ".db");
|
|
58
|
+
const db = new EmbeddingsDb(dbPath);
|
|
57
59
|
let uploadStats = { uploaded: 0, deleted: 0 };
|
|
58
60
|
logger.debug(`Config: chunksFile=${this.chunksFile} embeddingsFile=${this.embeddingsFile} force=${opts.force ?? false}`);
|
|
59
61
|
try {
|
|
@@ -65,7 +67,7 @@ export class Orchestrator {
|
|
|
65
67
|
const currentState = await gitTracker.getCurrentState();
|
|
66
68
|
const previousState = opts.force
|
|
67
69
|
? new Map()
|
|
68
|
-
:
|
|
70
|
+
: loadPreviousState(db);
|
|
69
71
|
const { toProcess, toDelete } = await gitTracker.getChangedFiles(previousState, currentState);
|
|
70
72
|
const gitDuration = Date.now() - t1;
|
|
71
73
|
logger.verbose(`Git tracking done in ${gitDuration}ms`);
|
|
@@ -77,6 +79,7 @@ export class Orchestrator {
|
|
|
77
79
|
});
|
|
78
80
|
if (toProcess.length === 0 && toDelete.length === 0 && !opts.force) {
|
|
79
81
|
logger.info("✨ No changes detected.");
|
|
82
|
+
db.close();
|
|
80
83
|
return;
|
|
81
84
|
}
|
|
82
85
|
logger.info(`📊 Changes: ${toProcess.length} to process, ${toDelete.length} to delete`);
|
|
@@ -93,7 +96,9 @@ export class Orchestrator {
|
|
|
93
96
|
const existingChunks = opts.force
|
|
94
97
|
? []
|
|
95
98
|
: await loadExistingChunks(this.chunksFile);
|
|
96
|
-
const
|
|
99
|
+
const chunkBar = createProgressBar("Chunking", toProcess.length);
|
|
100
|
+
const chunks = await chunkProcessor.processFiles(toProcess, fileState, existingChunks, (done, total) => chunkBar.update(done < total ? done : total));
|
|
101
|
+
chunkBar.stop();
|
|
97
102
|
await chunkProcessor.saveChunksLocal(chunks, this.chunksFile);
|
|
98
103
|
const chunkDuration = Date.now() - t2;
|
|
99
104
|
logger.verbose(`Chunking done in ${chunkDuration}ms`);
|
|
@@ -105,11 +110,17 @@ export class Orchestrator {
|
|
|
105
110
|
});
|
|
106
111
|
if (chunks.length === 0) {
|
|
107
112
|
logger.warn("⚠️ No chunks generated. Exiting.");
|
|
113
|
+
db.close();
|
|
108
114
|
return;
|
|
109
115
|
}
|
|
110
116
|
// Step 3: Generate embeddings
|
|
111
117
|
logger.info("🔢 Step 3: Generating embeddings...");
|
|
112
118
|
const t3 = Date.now();
|
|
119
|
+
const embedBars = [];
|
|
120
|
+
const uploader = new Uploader(this.config.vectorStore, {
|
|
121
|
+
retry: opts.retry,
|
|
122
|
+
logger: opts.logger,
|
|
123
|
+
});
|
|
113
124
|
const embedder = new EmbedderProcessor(this.config.embedder, {
|
|
114
125
|
rateLimitMs: opts.rateLimitMs,
|
|
115
126
|
batchSize: opts.batchSize,
|
|
@@ -117,9 +128,20 @@ export class Orchestrator {
|
|
|
117
128
|
retry: opts.retry,
|
|
118
129
|
concurrency: opts.concurrency,
|
|
119
130
|
vectorStoreName: this.config.vectorStore.name,
|
|
131
|
+
minIngestionBatchSize: opts.minIngestionBatchSize,
|
|
120
132
|
logger: opts.logger,
|
|
133
|
+
onProgress: (done, total) => {
|
|
134
|
+
if (embedBars.length === 0)
|
|
135
|
+
embedBars.push(createProgressBar("Embedding", total));
|
|
136
|
+
embedBars[0].update(done < total ? done : total);
|
|
137
|
+
},
|
|
121
138
|
});
|
|
122
|
-
const newEmbeddings = await embedder.run(this.chunksFile, opts.force || false
|
|
139
|
+
const newEmbeddings = await embedder.run(db, this.chunksFile, opts.force || false, opts.skipUpload
|
|
140
|
+
? undefined
|
|
141
|
+
: async () => {
|
|
142
|
+
await uploader.uploadPending(db);
|
|
143
|
+
});
|
|
144
|
+
embedBars[0]?.stop();
|
|
123
145
|
const embedDuration = Date.now() - t3;
|
|
124
146
|
logger.verbose(`Embedding done in ${embedDuration}ms`);
|
|
125
147
|
telemetry?.recordEmbedding({
|
|
@@ -129,8 +151,7 @@ export class Orchestrator {
|
|
|
129
151
|
});
|
|
130
152
|
// Step 4: Upload
|
|
131
153
|
if (opts.dryRun) {
|
|
132
|
-
const
|
|
133
|
-
const { toUpload: dryUpload, toDelete: dryDelete } = await uploader.getItemsToUpload(this.embeddingsFile, opts.force || false);
|
|
154
|
+
const { toUpload: dryUpload, toDelete: dryDelete } = await uploader.getItemsToUpload(db, opts.force || false);
|
|
134
155
|
logger.info("📤 Step 4: Upload (dry-run — no changes written)");
|
|
135
156
|
logger.info(` Would upload: ${dryUpload.length} document(s)`);
|
|
136
157
|
logger.info(` Would delete: ${dryDelete.length} source file(s)`);
|
|
@@ -138,11 +159,7 @@ export class Orchestrator {
|
|
|
138
159
|
else if (!opts.skipUpload) {
|
|
139
160
|
logger.info("📤 Step 4: Uploading to vector store...");
|
|
140
161
|
const t4 = Date.now();
|
|
141
|
-
|
|
142
|
-
retry: opts.retry,
|
|
143
|
-
logger: opts.logger,
|
|
144
|
-
});
|
|
145
|
-
uploadStats = await uploader.sync(this.embeddingsFile, opts.force || false);
|
|
162
|
+
uploadStats = await uploader.sync(db, opts.force || false);
|
|
146
163
|
const uploadDuration = Date.now() - t4;
|
|
147
164
|
logger.verbose(`Upload done in ${uploadDuration}ms`);
|
|
148
165
|
telemetry?.recordUpload({
|
|
@@ -177,6 +194,9 @@ export class Orchestrator {
|
|
|
177
194
|
}
|
|
178
195
|
throw err;
|
|
179
196
|
}
|
|
197
|
+
finally {
|
|
198
|
+
db.close();
|
|
199
|
+
}
|
|
180
200
|
}
|
|
181
201
|
}
|
|
182
202
|
//# sourceMappingURL=orchestrator.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"orchestrator.js","sourceRoot":"","sources":["../../src/core/orchestrator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"orchestrator.js","sourceRoot":"","sources":["../../src/core/orchestrator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EACL,iBAAiB,GAElB,MAAM,6BAA6B,CAAC;AAQrC,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEvC,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAwBhF,SAAS,iBAAiB,CAAC,EAAgB;IACzC,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAC;IACxC,KAAK,MAAM,KAAK,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC;QAChC,IAAI,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;YACzC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,KAAK,UAAU,kBAAkB,CAAC,UAAkB;IAClD,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACpD,MAAM,MAAM,GAAY,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC5C,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAE,MAAkB,CAAC,CAAC,CAAC,EAAE,CAAC;IAC1D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,aAAa,CAC1B,GAAW,EACX,MAA2B,EAC3B,OAAgC;IAEhC,IAAI,CAAC;QACH,MAAM,KAAK,CAAC,GAAG,EAAE;YACf,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,EAAE,CAAC;SAC7C,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,sCAAsC;IACxC,CAAC;AACH,CAAC;AAED,MAAM,OAAO,YAAY;IACf,MAAM,CAAoB;IAC1B,UAAU,CAAS;IACnB,cAAc,CAAS;IAE/B,YAAY,MAAyB;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,OAAO,EAAE,UAAU,IAAI,iBAAiB,EAAE,CAAC;QACpE,IAAI,CAAC,cAAc;YACjB,MAAM,CAAC,OAAO,EAAE,cAAc,IAAI,qBAAqB,EAAE,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,GAAG;QACP,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACzE,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,kBAAkB,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QACnE,SAAS,EAAE,KAAK,EAAE,CAAC;QAEnB,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC7D,MAAM,EAAE,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QAEpC,IAAI,WAAW,GAAG,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QAE9C,MAAM,CAAC,KAAK,CACV,sBAAsB,IAAI,CAAC,UAAU,mBAAmB,IAAI,CAAC,cAAc,UAAU,IAAI,CAAC,KAAK,IAAI,KAAK,EAAE,CAC3G,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAE3C,2BAA2B;YAC3B,MAAM,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;YAClD,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACrE,MAAM,YAAY,GAAG,MAAM,UAAU,CAAC,eAAe,EAAE,CAAC;YAExD,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK;gBAC9B,CAAC,CAAC,IAAI,GAAG,EAAkB;gBAC3B,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE1B,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,MAAM,UAAU,CAAC,eAAe,CAC9D,aAAa,EACb,YAAY,CACb,CAAC;YAEF,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YACpC,MAAM,CAAC,OAAO,CAAC,wBAAwB,WAAW,IAAI,CAAC,CAAC;YAExD,SAAS,EAAE,iBAAiB,CAAC;gBAC3B,UAAU,EAAE,WAAW;gBACvB,YAAY,EAAE,YAAY,CAAC,IAAI;gBAC/B,SAAS,EAAE,SAAS,CAAC,MAAM;gBAC3B,QAAQ,EAAE,QAAQ,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;gBACnE,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;gBACtC,EAAE,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YAED,MAAM,CAAC,IAAI,CACT,eAAe,SAAS,CAAC,MAAM,gBAAgB,QAAQ,CAAC,MAAM,YAAY,CAC3E,CAAC;YAEF,wCAAwC;YACxC,MAAM,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YAC/C,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,MAAM,cAAc,GAAG,IAAI,cAAc,CACvC,IAAI,CAAC,MAAM,CAAC,QAAQ,EACpB,IAAI,CAAC,MAAM,CACZ,CAAC;YAEF,MAAM,SAAS,GAAG,IAAI,GAAG,EAGtB,CAAC;YACJ,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;gBAC7B,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBACpC,IAAI,IAAI;oBAAE,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACtC,CAAC;YAED,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK;gBAC/B,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC,MAAM,kBAAkB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAE9C,MAAM,QAAQ,GAAG,iBAAiB,CAAC,UAAU,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;YACjE,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,YAAY,CAC9C,SAAS,EACT,SAAS,EACT,cAAc,EACd,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAC9D,CAAC;YACF,QAAQ,CAAC,IAAI,EAAE,CAAC;YAChB,MAAM,cAAc,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;YAE9D,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,oBAAoB,aAAa,IAAI,CAAC,CAAC;YAEtD,SAAS,EAAE,cAAc,CAAC;gBACxB,UAAU,EAAE,aAAa;gBACzB,cAAc,EAAE,SAAS,CAAC,MAAM;gBAChC,eAAe,EAAE,MAAM,CAAC,MAAM;gBAC9B,MAAM,EAAE,CAAC;aACV,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;gBAChD,EAAE,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YAED,8BAA8B;YAC9B,MAAM,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;YACnD,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,MAAM,SAAS,GAAkB,EAAE,CAAC;YAEpC,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE;gBACrD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;aACpB,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;gBAC3D,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI;gBAC7C,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;gBACjD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,UAAU,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;oBAC1B,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;wBACxB,SAAS,CAAC,IAAI,CAAC,iBAAiB,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC;oBACxD,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACnD,CAAC;aACF,CAAC,CAAC;YAEH,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC,GAAG,CACtC,EAAE,EACF,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,KAAK,IAAI,KAAK,EACnB,IAAI,CAAC,UAAU;gBACb,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,KAAK,IAAI,EAAE;oBACT,MAAM,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;gBACnC,CAAC,CACN,CAAC;YACF,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;YAErB,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,qBAAqB,aAAa,IAAI,CAAC,CAAC;YAEvD,SAAS,EAAE,eAAe,CAAC;gBACzB,UAAU,EAAE,aAAa;gBACzB,cAAc,EAAE,aAAa,CAAC,MAAM;gBACpC,aAAa,EAAE,MAAM,CAAC,MAAM,GAAG,aAAa,CAAC,MAAM;aACpD,CAAC,CAAC;YAEH,iBAAiB;YACjB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,GAChD,MAAM,QAAQ,CAAC,gBAAgB,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,CAAC;gBAC3D,MAAM,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;gBAChE,MAAM,CAAC,IAAI,CAAC,oBAAoB,SAAS,CAAC,MAAM,cAAc,CAAC,CAAC;gBAChE,MAAM,CAAC,IAAI,CAAC,oBAAoB,SAAS,CAAC,MAAM,iBAAiB,CAAC,CAAC;YACrE,CAAC;iBAAM,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC5B,MAAM,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;gBACvD,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBACtB,WAAW,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,CAAC;gBAE3D,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;gBACvC,MAAM,CAAC,OAAO,CAAC,kBAAkB,cAAc,IAAI,CAAC,CAAC;gBAErD,SAAS,EAAE,YAAY,CAAC;oBACtB,UAAU,EAAE,cAAc;oBAC1B,QAAQ,EAAE,WAAW,CAAC,QAAQ;oBAC9B,OAAO,EAAE,WAAW,CAAC,OAAO;iBAC7B,CAAC,CAAC;YACL,CAAC;YAED,MAAM,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAC;YAE3C,IAAI,SAAS,EAAE,CAAC;gBACd,SAAS,CAAC,MAAM,EAAE,CAAC;gBACnB,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACpC,MAAM,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAC3C,aAAa,EACb,gBAAgB,CACjB,CAAC;gBACF,MAAM,SAAS,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,IAAI,CAAC,aAAa,EAAE,UAAU,EAAE,CAAC;gBACnC,MAAM,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,SAAS,EAAE;oBAC5D,UAAU,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,UAAU;oBAC3C,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,MAAM;oBACnC,QAAQ,EAAE,WAAW,CAAC,QAAQ;oBAC9B,OAAO,EAAE,WAAW,CAAC,OAAO;iBAC7B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,EAAE,MAAM,EAAE,CAAC;YAEpB,IAAI,IAAI,CAAC,aAAa,EAAE,UAAU,EAAE,CAAC;gBACnC,MAAM,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,OAAO,EAAE;oBAC1D,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;oBACvD,UAAU,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,UAAU;iBAC5C,CAAC,CAAC;YACL,CAAC;YAED,MAAM,GAAG,CAAC;QACZ,CAAC;gBAAS,CAAC;YACT,EAAE,CAAC,KAAK,EAAE,CAAC;QACb,CAAC;IACH,CAAC;CACF"}
|
package/dist/core/uploader.d.ts
CHANGED
|
@@ -1,6 +1,8 @@
|
|
|
1
1
|
import { VectorStore, EmbeddedChunk } from "../interfaces/index.js";
|
|
2
2
|
import type { Logger } from "../interfaces/logger.js";
|
|
3
3
|
import { RetryOptions } from "./utils.js";
|
|
4
|
+
import { EmbeddingsDb } from "./embeddings-db.js";
|
|
5
|
+
export declare function isFatalVectorStoreError(err: unknown): boolean;
|
|
4
6
|
export declare class Uploader {
|
|
5
7
|
private vectorStore;
|
|
6
8
|
private retryOptions;
|
|
@@ -10,11 +12,15 @@ export declare class Uploader {
|
|
|
10
12
|
logger?: Logger;
|
|
11
13
|
});
|
|
12
14
|
private chunkToDocument;
|
|
13
|
-
getItemsToUpload(
|
|
15
|
+
getItemsToUpload(db: EmbeddingsDb, force?: boolean): Promise<{
|
|
14
16
|
toUpload: EmbeddedChunk[];
|
|
15
17
|
toDelete: string[];
|
|
16
18
|
}>;
|
|
17
|
-
|
|
19
|
+
uploadPending(db: EmbeddingsDb): Promise<{
|
|
20
|
+
uploaded: number;
|
|
21
|
+
deleted: number;
|
|
22
|
+
}>;
|
|
23
|
+
sync(db: EmbeddingsDb, force?: boolean): Promise<{
|
|
18
24
|
uploaded: number;
|
|
19
25
|
deleted: number;
|
|
20
26
|
}>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"uploader.d.ts","sourceRoot":"","sources":["../../src/core/uploader.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,WAAW,EAEX,aAAa,EACd,MAAM,wBAAwB,CAAC;AAChC,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;
|
|
1
|
+
{"version":3,"file":"uploader.d.ts","sourceRoot":"","sources":["../../src/core/uploader.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,WAAW,EAEX,aAAa,EACd,MAAM,wBAAwB,CAAC;AAChC,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAGtD,OAAO,EAAa,YAAY,EAAE,MAAM,YAAY,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AASlD,wBAAgB,uBAAuB,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAK7D;AAED,qBAAa,QAAQ;IACnB,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,MAAM,CAAS;gBAGrB,WAAW,EAAE,WAAW,EACxB,OAAO,GAAE;QAAE,KAAK,CAAC,EAAE,YAAY,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAO;IAOzD,OAAO,CAAC,eAAe;IAejB,gBAAgB,CACpB,EAAE,EAAE,YAAY,EAChB,KAAK,GAAE,OAAe,GACrB,OAAO,CAAC;QACT,QAAQ,EAAE,aAAa,EAAE,CAAC;QAC1B,QAAQ,EAAE,MAAM,EAAE,CAAC;KACpB,CAAC;IA4DI,aAAa,CACjB,EAAE,EAAE,YAAY,GACf,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;IAwB3C,IAAI,CACR,EAAE,EAAE,YAAY,EAChB,KAAK,GAAE,OAAe,GACrB,OAAO,CAAC;QACT,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;CA4DH"}
|
package/dist/core/uploader.js
CHANGED
|
@@ -1,12 +1,14 @@
|
|
|
1
1
|
import { NullLogger } from "../logger/null-logger.js";
|
|
2
2
|
import { createHash } from "crypto";
|
|
3
|
-
import { UploadError } from "./errors.js";
|
|
4
3
|
import { withRetry } from "./utils.js";
|
|
5
|
-
import { readEmbeddingsFile } from "./embeddings-io.js";
|
|
6
4
|
function contentHash(chunk) {
|
|
7
5
|
return (chunk.contentHash ??
|
|
8
6
|
createHash("sha256").update(chunk.content).digest("hex").slice(0, 16));
|
|
9
7
|
}
|
|
8
|
+
export function isFatalVectorStoreError(err) {
|
|
9
|
+
const msg = String(err instanceof Error ? err.message : err).toLowerCase();
|
|
10
|
+
return /schema error|schema mismatch|unauthorized|authentication failed/.test(msg);
|
|
11
|
+
}
|
|
10
12
|
export class Uploader {
|
|
11
13
|
vectorStore;
|
|
12
14
|
retryOptions;
|
|
@@ -27,26 +29,11 @@ export class Uploader {
|
|
|
27
29
|
collection,
|
|
28
30
|
};
|
|
29
31
|
}
|
|
30
|
-
async getItemsToUpload(
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
catch (err) {
|
|
36
|
-
throw new UploadError(`Failed to load embeddings from ${embeddingsFile}: ${err instanceof Error ? err.message : String(err)}`, {
|
|
37
|
-
suggestion: "Run the pipeline without --skip-upload to regenerate embeddings first.",
|
|
38
|
-
cause: err,
|
|
39
|
-
});
|
|
40
|
-
}
|
|
41
|
-
const { meta, chunks: embeddings } = result;
|
|
42
|
-
if (embeddings.length === 0 && !result) {
|
|
43
|
-
throw new UploadError(`No embeddings found at ${embeddingsFile}`, {
|
|
44
|
-
suggestion: "Run the pipeline without --skip-upload to regenerate embeddings first.",
|
|
45
|
-
});
|
|
46
|
-
}
|
|
47
|
-
this.logger.info(`📖 Loaded ${embeddings.length} embeddings from ${embeddingsFile}`);
|
|
48
|
-
// If the vector store changed since the last run, force a full re-upload to ensure
|
|
49
|
-
// the new store starts with consistent state.
|
|
32
|
+
async getItemsToUpload(db, force = false) {
|
|
33
|
+
const meta = db.getMeta();
|
|
34
|
+
const embeddings = db.getAll();
|
|
35
|
+
this.logger.info(`📖 Loaded ${embeddings.length} embeddings from db`);
|
|
36
|
+
// If the vector store changed since the last run, force a full re-upload.
|
|
50
37
|
let effectiveForce = force;
|
|
51
38
|
if (!force &&
|
|
52
39
|
meta?.vectorStoreName &&
|
|
@@ -85,10 +72,27 @@ export class Uploader {
|
|
|
85
72
|
toDelete: [...toDeleteSet],
|
|
86
73
|
};
|
|
87
74
|
}
|
|
88
|
-
async
|
|
75
|
+
async uploadPending(db) {
|
|
76
|
+
const pending = db.getPending();
|
|
77
|
+
if (pending.length === 0)
|
|
78
|
+
return { uploaded: 0, deleted: 0 };
|
|
79
|
+
this.logger.verbose(`📤 Uploading ${pending.length} pending embeddings...`);
|
|
80
|
+
const batchSize = 50;
|
|
81
|
+
for (let i = 0; i < pending.length; i += batchSize) {
|
|
82
|
+
const batch = pending.slice(i, i + batchSize);
|
|
83
|
+
const documents = batch.map((e) => this.chunkToDocument(e));
|
|
84
|
+
await withRetry(() => this.vectorStore.upsert(documents), {
|
|
85
|
+
...this.retryOptions,
|
|
86
|
+
isRetryable: (err) => !isFatalVectorStoreError(err),
|
|
87
|
+
}, this.logger);
|
|
88
|
+
db.markUploaded(batch.map((e) => contentHash(e)));
|
|
89
|
+
}
|
|
90
|
+
return { uploaded: pending.length, deleted: 0 };
|
|
91
|
+
}
|
|
92
|
+
async sync(db, force = false) {
|
|
89
93
|
this.logger.info("📤 Starting incremental upload...");
|
|
90
94
|
await this.vectorStore.initialize();
|
|
91
|
-
const { toUpload, toDelete } = await this.getItemsToUpload(
|
|
95
|
+
const { toUpload, toDelete } = await this.getItemsToUpload(db, force);
|
|
92
96
|
this.logger.info(`📊 Need to upload: ${toUpload.length} documents`);
|
|
93
97
|
this.logger.info(` Need to delete: ${toDelete.length} files`);
|
|
94
98
|
if (toUpload.length === 0 && toDelete.length === 0) {
|
|
@@ -96,16 +100,23 @@ export class Uploader {
|
|
|
96
100
|
return { uploaded: 0, deleted: 0 };
|
|
97
101
|
}
|
|
98
102
|
if (toDelete.length > 0) {
|
|
99
|
-
await withRetry(() => this.vectorStore.deleteBySourceFile(toDelete),
|
|
103
|
+
await withRetry(() => this.vectorStore.deleteBySourceFile(toDelete), {
|
|
104
|
+
...this.retryOptions,
|
|
105
|
+
isRetryable: (err) => !isFatalVectorStoreError(err),
|
|
106
|
+
}, this.logger);
|
|
100
107
|
this.logger.verbose(`🗑️ Deleted ${toDelete.length} obsolete docs`);
|
|
101
108
|
}
|
|
102
109
|
if (toUpload.length > 0) {
|
|
103
|
-
const documents = toUpload.map((e) => this.chunkToDocument(e));
|
|
104
110
|
const batchSize = 50;
|
|
105
|
-
const totalBatches = Math.ceil(
|
|
106
|
-
for (let i = 0; i <
|
|
107
|
-
const batch =
|
|
108
|
-
|
|
111
|
+
const totalBatches = Math.ceil(toUpload.length / batchSize);
|
|
112
|
+
for (let i = 0; i < toUpload.length; i += batchSize) {
|
|
113
|
+
const batch = toUpload.slice(i, i + batchSize);
|
|
114
|
+
const documents = batch.map((e) => this.chunkToDocument(e));
|
|
115
|
+
await withRetry(() => this.vectorStore.upsert(documents), {
|
|
116
|
+
...this.retryOptions,
|
|
117
|
+
isRetryable: (err) => !isFatalVectorStoreError(err),
|
|
118
|
+
}, this.logger);
|
|
119
|
+
db.markUploaded(batch.map((e) => contentHash(e)));
|
|
109
120
|
const batchNum = Math.floor(i / batchSize) + 1;
|
|
110
121
|
this.logger.verbose(` Batch ${batchNum}/${totalBatches}: ${batch.length} docs`);
|
|
111
122
|
this.logger.silly(` IDs: ${batch
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"uploader.js","sourceRoot":"","sources":["../../src/core/uploader.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"uploader.js","sourceRoot":"","sources":["../../src/core/uploader.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,SAAS,EAAgB,MAAM,YAAY,CAAC;AAGrD,SAAS,WAAW,CAAC,KAAoB;IACvC,OAAO,CACL,KAAK,CAAC,WAAW;QACjB,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CACtE,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,GAAY;IAClD,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;IAC3E,OAAO,iEAAiE,CAAC,IAAI,CAC3E,GAAG,CACJ,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,QAAQ;IACX,WAAW,CAAc;IACzB,YAAY,CAAe;IAC3B,MAAM,CAAS;IAEvB,YACE,WAAwB,EACxB,UAAqD,EAAE;QAEvD,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;QACxC,IAAI,CAAC,MAAM,GAAG,CAAC,OAAO,CAAC,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvE,CAAC;IAEO,eAAe,CACrB,KAAoB,EACpB,UAAmB;QAEnB,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,WAAW,EAAE,WAAW,CAAC,KAAK,CAAC;YAC/B,UAAU;SACX,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,EAAgB,EAChB,QAAiB,KAAK;QAKtB,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;QAC1B,MAAM,UAAU,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC;QAE/B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,UAAU,CAAC,MAAM,qBAAqB,CAAC,CAAC;QAEtE,0EAA0E;QAC1E,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,IACE,CAAC,KAAK;YACN,IAAI,EAAE,eAAe;YACrB,IAAI,CAAC,eAAe,KAAK,IAAI,CAAC,WAAW,CAAC,IAAI,EAC9C,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;YACzD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC;YAC1D,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;YAChE,cAAc,GAAG,IAAI,CAAC;QACxB,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,cAAc,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YACzE,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC;QAC5D,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,eAAe,EAAE,CAAC;QAC/D,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,oBAAoB,aAAa,CAAC,IAAI,oBAAoB,CAC3D,CAAC;QAEF,MAAM,YAAY,GAAoB,EAAE,CAAC;QACzC,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;QAEtC,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;YAC7B,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAEvD,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,KAAK,GAAG,CAAC,UAAU,yBAAyB,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,WAAW,CAClF,CAAC;gBACF,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;iBAAM,IAAI,YAAY,KAAK,GAAG,CAAC,UAAU,EAAE,CAAC;gBAC3C,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,KAAK,GAAG,CAAC,UAAU,WAAW,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,WAAW,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,WAAW,CACvG,CAAC;gBACF,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBAChC,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,KAAK,GAAG,CAAC,UAAU,WAAW,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,cAAc,CACrE,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,YAAY;YACtB,QAAQ,EAAE,CAAC,GAAG,WAAW,CAAC;SAC3B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,EAAgB;QAEhB,MAAM,OAAO,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC;QAChC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QAE7D,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,gBAAgB,OAAO,CAAC,MAAM,wBAAwB,CAAC,CAAC;QAE5E,MAAM,SAAS,GAAG,EAAE,CAAC;QACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YACnD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;YAC9C,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5D,MAAM,SAAS,CACb,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,EACxC;gBACE,GAAG,IAAI,CAAC,YAAY;gBACpB,WAAW,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,uBAAuB,CAAC,GAAG,CAAC;aACpD,EACD,IAAI,CAAC,MAAM,CACZ,CAAC;YACF,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC;QAED,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,IAAI,CACR,EAAgB,EAChB,QAAiB,KAAK;QAKtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;QAEtD,MAAM,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;QAEpC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAEtE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,QAAQ,CAAC,MAAM,YAAY,CAAC,CAAC;QACpE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,QAAQ,CAAC,MAAM,QAAQ,CAAC,CAAC;QAEhE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YAC3C,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QACrC,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,SAAS,CACb,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EACnD;gBACE,GAAG,IAAI,CAAC,YAAY;gBACpB,WAAW,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,uBAAuB,CAAC,GAAG,CAAC;aACpD,EACD,IAAI,CAAC,MAAM,CACZ,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,QAAQ,CAAC,MAAM,gBAAgB,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,SAAS,GAAG,EAAE,CAAC;YACrB,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;YAC5D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;gBACpD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;gBAC/C,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC5D,MAAM,SAAS,CACb,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,EACxC;oBACE,GAAG,IAAI,CAAC,YAAY;oBACpB,WAAW,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,uBAAuB,CAAC,GAAG,CAAC;iBACpD,EACD,IAAI,CAAC,MAAM,CACZ,CAAC;gBACF,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;gBAC/C,IAAI,CAAC,MAAM,CAAC,OAAO,CACjB,WAAW,QAAQ,IAAI,YAAY,KAAK,KAAK,CAAC,MAAM,OAAO,CAC5D,CAAC;gBACF,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,UAAU,KAAK;qBACZ,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,GAAG,CAAC;qBAC7C,IAAI,CAAC,IAAI,CAAC,EAAE,CAChB,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,gBAAgB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACvD,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAEtD,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC;IACjE,CAAC;CACF"}
|
package/dist/core/utils.d.ts
CHANGED
|
@@ -5,7 +5,9 @@ export interface RetryOptions {
|
|
|
5
5
|
maxRetries?: number;
|
|
6
6
|
retryDelayMs?: number;
|
|
7
7
|
retryBackoffFactor?: number;
|
|
8
|
+
isRetryable?: (error: unknown) => boolean;
|
|
8
9
|
}
|
|
10
|
+
export declare function defaultIsRetryable(error: unknown): boolean;
|
|
9
11
|
export declare function withRetry<T>(fn: () => Promise<T>, opts?: RetryOptions, logger?: Logger): Promise<T>;
|
|
10
12
|
export declare function withConcurrency<T>(tasks: Array<() => Promise<T>>, limit: number): Promise<T[]>;
|
|
11
13
|
export declare function batchArray<T>(array: T[], batchSize: number): T[][];
|
package/dist/core/utils.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/core/utils.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAEtD,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAE1D;AAED,wBAAgB,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE/C;AAED,MAAM,WAAW,YAAY;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,kBAAkB,CAAC,EAAE,MAAM,CAAC;
|
|
1
|
+
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/core/utils.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAEtD,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAE1D;AAED,wBAAgB,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE/C;AAED,MAAM,WAAW,YAAY;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,OAAO,CAAC;CAC3C;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAyB1D;AAED,wBAAsB,SAAS,CAAC,CAAC,EAC/B,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EACpB,IAAI,GAAE,YAAiB,EACvB,MAAM,CAAC,EAAE,MAAM,GACd,OAAO,CAAC,CAAC,CAAC,CA2BZ;AAED,wBAAsB,eAAe,CAAC,CAAC,EACrC,KAAK,EAAE,KAAK,CAAC,MAAM,OAAO,CAAC,CAAC,CAAC,CAAC,EAC9B,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC,CAAC,EAAE,CAAC,CAed;AAED,wBAAgB,UAAU,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,SAAS,EAAE,MAAM,GAAG,CAAC,EAAE,EAAE,CAMlE;AAED,wBAAgB,WAAW,CAAC,CAAC,EAC3B,KAAK,EAAE,CAAC,EAAE,EACV,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,MAAM,EAC5B,OAAO,EAAE,MAAM,GACd,CAAC,EAAE,EAAE,CAqBP;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAExD;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAIzD"}
|
package/dist/core/utils.js
CHANGED
|
@@ -5,10 +5,31 @@ export function computeContentHash(content) {
|
|
|
5
5
|
export function sleep(ms) {
|
|
6
6
|
return new Promise((resolve) => setTimeout(resolve, ms));
|
|
7
7
|
}
|
|
8
|
+
export function defaultIsRetryable(error) {
|
|
9
|
+
const msg = error instanceof Error
|
|
10
|
+
? error.message
|
|
11
|
+
: typeof error === "string"
|
|
12
|
+
? error
|
|
13
|
+
: String(error);
|
|
14
|
+
const lower = msg.toLowerCase();
|
|
15
|
+
if (lower.includes("access denied") ||
|
|
16
|
+
lower.includes("forbidden") ||
|
|
17
|
+
lower.includes("unauthorized") ||
|
|
18
|
+
lower.includes("not found")) {
|
|
19
|
+
return false;
|
|
20
|
+
}
|
|
21
|
+
// Reject any 4xx HTTP status code except 429 (Too Many Requests)
|
|
22
|
+
const m = msg.match(/\b4(\d{2})\b/);
|
|
23
|
+
if (m) {
|
|
24
|
+
return parseInt(m[0], 10) === 429;
|
|
25
|
+
}
|
|
26
|
+
return true;
|
|
27
|
+
}
|
|
8
28
|
export async function withRetry(fn, opts = {}, logger) {
|
|
9
29
|
const maxRetries = opts.maxRetries ?? 3;
|
|
10
30
|
const delayMs = opts.retryDelayMs ?? 1000;
|
|
11
31
|
const factor = opts.retryBackoffFactor ?? 2;
|
|
32
|
+
const isRetryable = opts.isRetryable;
|
|
12
33
|
let lastErr;
|
|
13
34
|
for (let attempt = 0; attempt <= maxRetries; attempt++) {
|
|
14
35
|
try {
|
|
@@ -16,6 +37,9 @@ export async function withRetry(fn, opts = {}, logger) {
|
|
|
16
37
|
}
|
|
17
38
|
catch (err) {
|
|
18
39
|
lastErr = err;
|
|
40
|
+
if (isRetryable && !isRetryable(err)) {
|
|
41
|
+
throw err;
|
|
42
|
+
}
|
|
19
43
|
if (attempt < maxRetries) {
|
|
20
44
|
const wait = Math.round(delayMs * Math.pow(factor, attempt));
|
|
21
45
|
logger?.verbose(` ⚠️ Attempt ${attempt + 1}/${maxRetries + 1} failed` +
|
package/dist/core/utils.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/core/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAGpC,MAAM,UAAU,kBAAkB,CAAC,OAAe;IAChD,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AACzE,CAAC;AAED,MAAM,UAAU,KAAK,CAAC,EAAU;IAC9B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC;
|
|
1
|
+
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/core/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAGpC,MAAM,UAAU,kBAAkB,CAAC,OAAe;IAChD,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AACzE,CAAC;AAED,MAAM,UAAU,KAAK,CAAC,EAAU;IAC9B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC;AASD,MAAM,UAAU,kBAAkB,CAAC,KAAc;IAC/C,MAAM,GAAG,GACP,KAAK,YAAY,KAAK;QACpB,CAAC,CAAC,KAAK,CAAC,OAAO;QACf,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ;YACzB,CAAC,CAAC,KAAK;YACP,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAEtB,MAAM,KAAK,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;IAChC,IACE,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC;QAC/B,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC;QAC3B,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC9B,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,EAC3B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,iEAAiE;IACjE,MAAM,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IACpC,IAAI,CAAC,EAAE,CAAC;QACN,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC;IACpC,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,EAAoB,EACpB,OAAqB,EAAE,EACvB,MAAe;IAEf,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;IACxC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,kBAAkB,IAAI,CAAC,CAAC;IAC5C,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;IAErC,IAAI,OAAgB,CAAC;IACrB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;QACvD,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,GAAG,GAAG,CAAC;YACd,IAAI,WAAW,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrC,MAAM,GAAG,CAAC;YACZ,CAAC;YACD,IAAI,OAAO,GAAG,UAAU,EAAE,CAAC;gBACzB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;gBAC7D,MAAM,EAAE,OAAO,CACb,gBAAgB,OAAO,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,SAAS;oBACpD,KAAK,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG;oBACxD,iBAAiB,IAAI,OAAO,CAC/B,CAAC;gBACF,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC;YACpB,CAAC;QACH,CAAC;IACH,CAAC;IACD,MAAM,OAAO,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,KAA8B,EAC9B,KAAa;IAEb,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAClC,MAAM,OAAO,GAAG,IAAI,KAAK,CAAI,KAAK,CAAC,MAAM,CAAC,CAAC;IAC3C,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,KAAK,UAAU,MAAM;QACnB,OAAO,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;YAC5B,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;YAClB,OAAO,CAAC,CAAC,CAAC,GAAG,MAAM,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QAChC,CAAC;IACH,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAClD,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;IAC/D,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,UAAU,CAAI,KAAU,EAAE,SAAiB;IACzD,MAAM,OAAO,GAAU,EAAE,CAAC;IAC1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;QACjD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,WAAW,CACzB,KAAU,EACV,QAAgB,EAChB,OAA4B,EAC5B,OAAe;IAEf,MAAM,OAAO,GAAU,EAAE,CAAC;IAC1B,IAAI,OAAO,GAAQ,EAAE,CAAC;IACtB,IAAI,WAAW,GAAG,CAAC,CAAC;IAEpB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAC3B,IACE,OAAO,CAAC,MAAM,GAAG,CAAC;YAClB,CAAC,OAAO,CAAC,MAAM,IAAI,QAAQ,IAAI,WAAW,GAAG,IAAI,GAAG,OAAO,CAAC,EAC5D,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACtB,OAAO,GAAG,EAAE,CAAC;YACb,WAAW,GAAG,CAAC,CAAC;QAClB,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnB,WAAW,IAAI,IAAI,CAAC;IACtB,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9C,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,QAAgB;IAC9C,OAAO,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,QAAQ,CAAC;AAC/C,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAAgB;IAC/C,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAClC,KAAK,CAAC,GAAG,EAAE,CAAC;IACZ,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACzB,CAAC"}
|
package/dist/eval/runner.d.ts
CHANGED
|
@@ -11,6 +11,6 @@ export declare class EvalRunner {
|
|
|
11
11
|
private readonly dataset;
|
|
12
12
|
private readonly topK;
|
|
13
13
|
constructor(store: VectorStore, embedder: EmbeddingProvider, dataset: EvalDataset, topK?: number);
|
|
14
|
-
run(): Promise<EvalRunResult>;
|
|
14
|
+
run(onProgress?: (completed: number, total: number) => void): Promise<EvalRunResult>;
|
|
15
15
|
}
|
|
16
16
|
//# sourceMappingURL=runner.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"runner.d.ts","sourceRoot":"","sources":["../../src/eval/runner.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC7E,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAGxE,MAAM,WAAW,aAAa;IAC5B,UAAU,EAAE,UAAU,CAAC;IACvB,2EAA2E;IAC3E,gBAAgB,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED,qBAAa,UAAU;IAEnB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,IAAI;gBAHJ,KAAK,EAAE,WAAW,EAClB,QAAQ,EAAE,iBAAiB,EAC3B,OAAO,EAAE,WAAW,EACpB,IAAI,SAAK;IAGtB,GAAG,IAAI,OAAO,CAAC,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"runner.d.ts","sourceRoot":"","sources":["../../src/eval/runner.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC7E,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAGxE,MAAM,WAAW,aAAa;IAC5B,UAAU,EAAE,UAAU,CAAC;IACvB,2EAA2E;IAC3E,gBAAgB,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED,qBAAa,UAAU;IAEnB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,IAAI;gBAHJ,KAAK,EAAE,WAAW,EAClB,QAAQ,EAAE,iBAAiB,EAC3B,OAAO,EAAE,WAAW,EACpB,IAAI,SAAK;IAGtB,GAAG,CACP,UAAU,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,GACtD,OAAO,CAAC,aAAa,CAAC;CAiD1B"}
|
package/dist/eval/runner.js
CHANGED
|
@@ -10,9 +10,10 @@ export class EvalRunner {
|
|
|
10
10
|
this.dataset = dataset;
|
|
11
11
|
this.topK = topK;
|
|
12
12
|
}
|
|
13
|
-
async run() {
|
|
13
|
+
async run(onProgress) {
|
|
14
14
|
const queryResults = [];
|
|
15
|
-
for (
|
|
15
|
+
for (let qi = 0; qi < this.dataset.queries.length; qi++) {
|
|
16
|
+
const evalQuery = this.dataset.queries[qi];
|
|
16
17
|
const embedding = await this.embedder.embed(evalQuery.query);
|
|
17
18
|
const searchResults = await this.store.search(embedding, this.topK);
|
|
18
19
|
const relevantIds = new Set();
|
|
@@ -35,6 +36,7 @@ export class EvalRunner {
|
|
|
35
36
|
}
|
|
36
37
|
}
|
|
37
38
|
queryResults.push({ retrievedIds, relevantIds });
|
|
39
|
+
onProgress?.(qi + 1, this.dataset.queries.length);
|
|
38
40
|
}
|
|
39
41
|
const evalResult = computeEvalResult(queryResults);
|
|
40
42
|
const perQueryRrScores = queryResults.map(({ retrievedIds, relevantIds }) => reciprocalRank(retrievedIds, relevantIds));
|
package/dist/eval/runner.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/eval/runner.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAQjE,MAAM,OAAO,UAAU;IAEF;IACA;IACA;IACA;IAJnB,YACmB,KAAkB,EAClB,QAA2B,EAC3B,OAAoB,EACpB,OAAO,EAAE;QAHT,UAAK,GAAL,KAAK,CAAa;QAClB,aAAQ,GAAR,QAAQ,CAAmB;QAC3B,YAAO,GAAP,OAAO,CAAa;QACpB,SAAI,GAAJ,IAAI,CAAK;IACzB,CAAC;IAEJ,KAAK,CAAC,GAAG;
|
|
1
|
+
{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/eval/runner.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAQjE,MAAM,OAAO,UAAU;IAEF;IACA;IACA;IACA;IAJnB,YACmB,KAAkB,EAClB,QAA2B,EAC3B,OAAoB,EACpB,OAAO,EAAE;QAHT,UAAK,GAAL,KAAK,CAAa;QAClB,aAAQ,GAAR,QAAQ,CAAmB;QAC3B,YAAO,GAAP,OAAO,CAAa;QACpB,SAAI,GAAJ,IAAI,CAAK;IACzB,CAAC;IAEJ,KAAK,CAAC,GAAG,CACP,UAAuD;QAEvD,MAAM,YAAY,GAGb,EAAE,CAAC;QAER,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC;YACxD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAC7D,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAEpE,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;YAEtC,IAAI,SAAS,CAAC,gBAAgB,EAAE,CAAC;gBAC/B,KAAK,MAAM,EAAE,IAAI,SAAS,CAAC,gBAAgB,EAAE,CAAC;oBAC5C,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;YAED,2EAA2E;YAC3E,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CACpC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAC,CAAC,QAAQ,CAAC,WAAkC,IAAI,CAAC,CAAC,EAAE,CAC9D,CAAC;YAEF,2BAA2B;YAC3B,IAAI,SAAS,CAAC,eAAe,IAAI,SAAS,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtE,KAAK,MAAM,QAAQ,IAAI,SAAS,CAAC,eAAe,EAAE,CAAC;oBACjD,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;wBACnC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;4BACtC,WAAW,CAAC,GAAG,CACZ,MAAM,CAAC,QAAQ,CAAC,WAAkC;gCACjD,MAAM,CAAC,EAAE,CACZ,CAAC;wBACJ,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,YAAY,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC;YACjD,UAAU,EAAE,CAAC,EAAE,GAAG,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,UAAU,GAAG,iBAAiB,CAAC,YAAY,CAAC,CAAC;QACnD,MAAM,gBAAgB,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE,CAC1E,cAAc,CAAC,YAAY,EAAE,WAAW,CAAC,CAC1C,CAAC;QAEF,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC;IAC1C,CAAC;CACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"consola-logger.d.ts","sourceRoot":"","sources":["../../src/logger/consola-logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,KAAK,eAAe,EAAE,MAAM,SAAS,CAAC;AAC9D,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAatD,qBAAa,aAAc,YAAW,MAAM;IAC1C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAiB;gBAE9B,QAAQ,EAAE,eAAe;IAIrC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,GAAG,aAAa;IAuC/C,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAGhD,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAGhD,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAG/C,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAG/C,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAIlD,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAIlD,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAIhD,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;
|
|
1
|
+
{"version":3,"file":"consola-logger.d.ts","sourceRoot":"","sources":["../../src/logger/consola-logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,KAAK,eAAe,EAAE,MAAM,SAAS,CAAC;AAC9D,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAatD,qBAAa,aAAc,YAAW,MAAM;IAC1C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAiB;gBAE9B,QAAQ,EAAE,eAAe;IAIrC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,GAAG,aAAa;IAuC/C,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAGhD,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAGhD,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAG/C,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAG/C,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAIlD,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAIlD,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAIhD,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAahD,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAUhD,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM;CAG7B"}
|
|
@@ -52,12 +52,22 @@ export class ConsolaLogger {
|
|
|
52
52
|
// in args so FancyReporter renders it (formatArgs uses args, not message).
|
|
53
53
|
if (this._consola.level < 6)
|
|
54
54
|
return;
|
|
55
|
-
this._consola._log({
|
|
55
|
+
this._consola._log({
|
|
56
|
+
level: 6,
|
|
57
|
+
message: "",
|
|
58
|
+
args: [message, ...args],
|
|
59
|
+
date: new Date(),
|
|
60
|
+
});
|
|
56
61
|
}
|
|
57
62
|
silly(message, ...args) {
|
|
58
63
|
if (this._consola.level < 999)
|
|
59
64
|
return;
|
|
60
|
-
this._consola._log({
|
|
65
|
+
this._consola._log({
|
|
66
|
+
level: 999,
|
|
67
|
+
message: "",
|
|
68
|
+
args: [message, ...args],
|
|
69
|
+
date: new Date(),
|
|
70
|
+
});
|
|
61
71
|
}
|
|
62
72
|
withTag(tag) {
|
|
63
73
|
return new ConsolaLogger(this._consola.withTag(tag));
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"consola-logger.js","sourceRoot":"","sources":["../../src/logger/consola-logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAwB,MAAM,SAAS,CAAC;AAG9D,MAAM,gBAAgB,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,CAAU,CAAC;AAWvD,MAAM,OAAO,aAAa;IACP,QAAQ,CAAiB;IAE1C,YAAY,QAAyB;QACnC,IAAI,CAAC,QAAQ,GAAG,QAA0B,CAAC;IAC7C,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,SAAiB;QAC7B,MAAM,KAAK,GACT,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACrE,MAAM,QAAQ,GAAG,aAAa,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QAE1C,yEAAyE;QACzE,4EAA4E;QAC5E,gEAAgE;QAChE,MAAM,QAAQ,GACZ,QAWD,CAAC,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1B,IAAI,OAAO,QAAQ,EAAE,WAAW,KAAK,UAAU,EAAE,CAAC;YAChD,MAAM,IAAI,GAAG,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACjD,QAAQ,CAAC,WAAW,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;gBAC9C,MAAM,QAAQ,GAAG,KAAK;qBACnB,KAAK,CAAC,IAAI,CAAC;qBACX,MAAM,CACL,CAAC,IAAI,EAAE,EAAE,CACP,CAAC,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CAAC;oBACtC,CAAC,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CACnC;qBACA,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,OAAO,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YACvC,CAAC,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,aAAa,CAAC,QAA0B,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACxC,CAAC;IACD,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACxC,CAAC;IACD,IAAI,CAAC,OAAe,EAAE,GAAG,IAAe;QACtC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACvC,CAAC;IACD,IAAI,CAAC,OAAe,EAAE,GAAG,IAAe;QACtC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,CAAC,OAAe,EAAE,GAAG,IAAe;QACzC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED,OAAO,CAAC,OAAe,EAAE,GAAG,IAAe;QACzC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,UAAU;IACnD,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,UAAU;IACnD,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,mEAAmE;QACnE,sEAAsE;QACtE,2EAA2E;QAC3E,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,GAAG,CAAC;YAAE,OAAO;QACpC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,
|
|
1
|
+
{"version":3,"file":"consola-logger.js","sourceRoot":"","sources":["../../src/logger/consola-logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAwB,MAAM,SAAS,CAAC;AAG9D,MAAM,gBAAgB,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,CAAU,CAAC;AAWvD,MAAM,OAAO,aAAa;IACP,QAAQ,CAAiB;IAE1C,YAAY,QAAyB;QACnC,IAAI,CAAC,QAAQ,GAAG,QAA0B,CAAC;IAC7C,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,SAAiB;QAC7B,MAAM,KAAK,GACT,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACrE,MAAM,QAAQ,GAAG,aAAa,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QAE1C,yEAAyE;QACzE,4EAA4E;QAC5E,gEAAgE;QAChE,MAAM,QAAQ,GACZ,QAWD,CAAC,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1B,IAAI,OAAO,QAAQ,EAAE,WAAW,KAAK,UAAU,EAAE,CAAC;YAChD,MAAM,IAAI,GAAG,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACjD,QAAQ,CAAC,WAAW,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;gBAC9C,MAAM,QAAQ,GAAG,KAAK;qBACnB,KAAK,CAAC,IAAI,CAAC;qBACX,MAAM,CACL,CAAC,IAAI,EAAE,EAAE,CACP,CAAC,IAAI,CAAC,QAAQ,CAAC,sBAAsB,CAAC;oBACtC,CAAC,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CACnC;qBACA,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,OAAO,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YACvC,CAAC,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,aAAa,CAAC,QAA0B,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACxC,CAAC;IACD,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACxC,CAAC;IACD,IAAI,CAAC,OAAe,EAAE,GAAG,IAAe;QACtC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACvC,CAAC;IACD,IAAI,CAAC,OAAe,EAAE,GAAG,IAAe;QACtC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,CAAC,OAAe,EAAE,GAAG,IAAe;QACzC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED,OAAO,CAAC,OAAe,EAAE,GAAG,IAAe;QACzC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,UAAU;IACnD,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,UAAU;IACnD,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,mEAAmE;QACnE,sEAAsE;QACtE,2EAA2E;QAC3E,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,GAAG,CAAC;YAAE,OAAO;QACpC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,EAAE;YACX,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC;YACxB,IAAI,EAAE,IAAI,IAAI,EAAE;SACjB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,OAAe,EAAE,GAAG,IAAe;QACvC,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,GAAG,GAAG;YAAE,OAAO;QACtC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACjB,KAAK,EAAE,GAAG;YACV,OAAO,EAAE,EAAE;YACX,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC;YACxB,IAAI,EAAE,IAAI,IAAI,EAAE;SACjB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,CAAC,GAAW;QACjB,OAAO,IAAI,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;IACvD,CAAC;CACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"plugin-registry.d.ts","sourceRoot":"","sources":["../src/plugin-registry.ts"],"names":[],"mappings":"AAGA,MAAM,MAAM,UAAU,GAAG,UAAU,GAAG,aAAa,CAAC;AAEpD,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,UAAU,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACxC;AAED,eAAO,MAAM,gBAAgB,EAAE,WAAW,
|
|
1
|
+
{"version":3,"file":"plugin-registry.d.ts","sourceRoot":"","sources":["../src/plugin-registry.ts"],"names":[],"mappings":"AAGA,MAAM,MAAM,UAAU,GAAG,UAAU,GAAG,aAAa,CAAC;AAEpD,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,UAAU,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACxC;AAED,eAAO,MAAM,gBAAgB,EAAE,WAAW,EA4FzC,CAAC;AAEF,wBAAsB,uBAAuB,CAC3C,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC,WAAW,EAAE,CAAC,CAkExB;AAED,MAAM,WAAW,cAAc;IAC7B,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,MAAM,EAAE,WAAW,EAAE,CAAC;CACvB;AAED,wBAAsB,YAAY,CAChC,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC,cAAc,CAAC,CAgBzB"}
|
package/dist/plugin-registry.js
CHANGED
|
@@ -14,19 +14,6 @@ export const BUILT_IN_PLUGINS = [
|
|
|
14
14
|
dimensions: 1536,
|
|
15
15
|
},
|
|
16
16
|
},
|
|
17
|
-
{
|
|
18
|
-
type: "embedder",
|
|
19
|
-
label: "GitHub Models (Azure-compatible endpoint)",
|
|
20
|
-
key: "github-models",
|
|
21
|
-
package: "@vivantel/virage-embedder-openai",
|
|
22
|
-
envVars: ["GITHUB_TOKEN"],
|
|
23
|
-
defaultConfig: {
|
|
24
|
-
apiKey: "${GITHUB_TOKEN}",
|
|
25
|
-
baseURL: "https://models.github.ai/inference",
|
|
26
|
-
model: "openai/text-embedding-3-small",
|
|
27
|
-
dimensions: 1536,
|
|
28
|
-
},
|
|
29
|
-
},
|
|
30
17
|
{
|
|
31
18
|
type: "embedder",
|
|
32
19
|
label: "FastEmbed (local, no API key required)",
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"plugin-registry.js","sourceRoot":"","sources":["../src/plugin-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAa5B,MAAM,CAAC,MAAM,gBAAgB,GAAkB;IAC7C,kBAAkB;IAClB;QACE,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,iCAAiC;QACxC,GAAG,EAAE,QAAQ;QACb,OAAO,EAAE,kCAAkC;QAC3C,OAAO,EAAE,CAAC,gBAAgB,CAAC;QAC3B,aAAa,EAAE;YACb,MAAM,EAAE,mBAAmB;YAC3B,KAAK,EAAE,wBAAwB;YAC/B,UAAU,EAAE,IAAI;SACjB;KACF;IACD;QACE,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,
|
|
1
|
+
{"version":3,"file":"plugin-registry.js","sourceRoot":"","sources":["../src/plugin-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAa5B,MAAM,CAAC,MAAM,gBAAgB,GAAkB;IAC7C,kBAAkB;IAClB;QACE,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,iCAAiC;QACxC,GAAG,EAAE,QAAQ;QACb,OAAO,EAAE,kCAAkC;QAC3C,OAAO,EAAE,CAAC,gBAAgB,CAAC;QAC3B,aAAa,EAAE;YACb,MAAM,EAAE,mBAAmB;YAC3B,KAAK,EAAE,wBAAwB;YAC/B,UAAU,EAAE,IAAI;SACjB;KACF;IACD;QACE,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,wCAAwC;QAC/C,GAAG,EAAE,WAAW;QAChB,OAAO,EAAE,qCAAqC;QAC9C,OAAO,EAAE,EAAE;QACX,aAAa,EAAE;YACb,KAAK,EAAE,wBAAwB;YAC/B,UAAU,EAAE,GAAG;SAChB;KACF;IACD;QACE,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,uDAAuD;QAC9D,GAAG,EAAE,cAAc;QACnB,OAAO,EAAE,wCAAwC;QACjD,OAAO,EAAE,EAAE;QACX,aAAa,EAAE;YACb,KAAK,EAAE,yBAAyB;YAChC,UAAU,EAAE,GAAG;SAChB;KACF;IACD;QACE,IAAI,EAAE,UAAU;QAChB,KAAK,EAAE,6DAA6D;QACpE,GAAG,EAAE,QAAQ;QACb,OAAO,EAAE,+BAA+B;QACxC,OAAO,EAAE,EAAE;QACX,aAAa,EAAE;YACb,MAAM,EAAE,iBAAiB;YACzB,KAAK,EAAE,iBAAiB;YACxB,UAAU,EAAE,IAAI;SACjB;KACF;IACD,sBAAsB;IACtB;QACE,IAAI,EAAE,aAAa;QACnB,KAAK,EAAE,yDAAyD;QAChE,GAAG,EAAE,UAAU;QACf,OAAO,EAAE,iCAAiC;QAC1C,OAAO,EAAE,CAAC,cAAc,CAAC;QACzB,aAAa,EAAE,EAAE,gBAAgB,EAAE,iBAAiB,EAAE;KACvD;IACD;QACE,IAAI,EAAE,aAAa;QACnB,KAAK,EAAE,gCAAgC;QACvC,GAAG,EAAE,QAAQ;QACb,OAAO,EAAE,+BAA+B;QACxC,OAAO,EAAE,EAAE;QACX,aAAa,EAAE;YACb,GAAG,EAAE,eAAe;YACpB,MAAM,EAAE,mBAAmB;SAC5B;KACF;IACD;QACE,IAAI,EAAE,aAAa;QACnB,KAAK,EAAE,mDAAmD;QAC1D,GAAG,EAAE,SAAS;QACd,OAAO,EAAE,gCAAgC;QACzC,OAAO,EAAE,EAAE;QACX,aAAa,EAAE,EAAE,GAAG,EAAE,iBAAiB,EAAE;KAC1C;IACD;QACE,IAAI,EAAE,aAAa;QACnB,KAAK,EAAE,4BAA4B;QACnC,GAAG,EAAE,UAAU;QACf,OAAO,EAAE,iCAAiC;QAC1C,OAAO,EAAE,EAAE;QACX,aAAa,EAAE,EAAE,IAAI,EAAE,uBAAuB,EAAE;KACjD;IACD;QACE,IAAI,EAAE,aAAa;QACnB,KAAK,EAAE,uDAAuD;QAC9D,GAAG,EAAE,QAAQ;QACb,OAAO,EAAE,4BAA4B;QACrC,OAAO,EAAE,EAAE;QACX,aAAa,EAAE,EAAE;KAClB;CACF,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,WAAmB;IAEnB,IAAI,OAAgC,CAAC;IACrC,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,EAAE,OAAO,CAAC,CAAC;QAC3E,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAA4B,CAAC;IAC3D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,IAAI,GAA2B;QACnC,GAAG,CAAE,OAAO,CAAC,YAAmD,IAAI,EAAE,CAAC;QACvE,GAAG,CAAE,OAAO,CAAC,eAAsD,IAAI,EAAE,CAAC;KAC3E,CAAC;IAEF,MAAM,OAAO,GAAkB,EAAE,CAAC;IAElC,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACxC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CACrB,WAAW,EACX,cAAc,EACd,OAAO,EACP,cAAc,CACf,CAAC;YACF,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YACvD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAA4B,CAAC;YAEjE,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;YACvC,IACE,CAAC,SAAS;gBACV,OAAO,SAAS,KAAK,QAAQ;gBAC7B,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC;gBAExB,SAAS;YAEX,MAAM,MAAM,GAAG,SAAoC,CAAC;YACpD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,MAAM,CAAC;YAEpC,IACE,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,aAAa,CAAC;gBAC/C,OAAO,KAAK,KAAK,QAAQ;gBACzB,OAAO,GAAG,KAAK,QAAQ;gBAEvB,SAAS;YAEX,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,IAAkB;gBACxB,KAAK;gBACL,GAAG;gBACH,OAAO,EAAE,OAAO;gBAChB,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;oBACpC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC;oBAClE,CAAC,CAAC,EAAE;gBACN,aAAa,EACX,OAAO,MAAM,CAAC,aAAa,KAAK,QAAQ;oBACxC,MAAM,CAAC,aAAa,KAAK,IAAI;oBAC7B,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC;oBAClC,CAAC,CAAE,MAAM,CAAC,aAAyC;oBACnD,CAAC,CAAC,EAAE;aACT,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,+CAA+C;QACjD,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAOD,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,WAAmB;IAEnB,MAAM,QAAQ,GAAG,MAAM,uBAAuB,CAAC,WAAW,CAAC,CAAC;IAE5D,MAAM,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC9C,KAAK,MAAM,KAAK,IAAI,gBAAgB,EAAE,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,GAAG,EAAE,EAAE,KAAK,CAAC,CAAC;IAClD,CAAC;IACD,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,GAAG,EAAE,EAAE,KAAK,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IACxC,OAAO;QACL,SAAS,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC;QACnD,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC;KACpD,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"progress-bar.d.ts","sourceRoot":"","sources":["../../src/progress/progress-bar.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACnC;AAED,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,WAAW,CAwB3E"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import cliProgress from "cli-progress";
|
|
2
|
+
export function createProgressBar(label, total) {
|
|
3
|
+
if (total === 0) {
|
|
4
|
+
return { update: () => { }, stop: () => { } };
|
|
5
|
+
}
|
|
6
|
+
const bar = new cliProgress.SingleBar({
|
|
7
|
+
format: `${label} [{bar}] {percentage}% | {value}/{total} | {duration_formatted} elapsed | ETA: {eta_formatted}`,
|
|
8
|
+
clearOnComplete: false,
|
|
9
|
+
hideCursor: true,
|
|
10
|
+
}, cliProgress.Presets.shades_classic);
|
|
11
|
+
bar.start(total, 0);
|
|
12
|
+
return {
|
|
13
|
+
update(current) {
|
|
14
|
+
bar.update(current);
|
|
15
|
+
},
|
|
16
|
+
stop(finalMessage) {
|
|
17
|
+
bar.stop();
|
|
18
|
+
if (finalMessage)
|
|
19
|
+
console.log(finalMessage);
|
|
20
|
+
},
|
|
21
|
+
};
|
|
22
|
+
}
|
|
23
|
+
//# sourceMappingURL=progress-bar.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"progress-bar.js","sourceRoot":"","sources":["../../src/progress/progress-bar.ts"],"names":[],"mappings":"AAAA,OAAO,WAAW,MAAM,cAAc,CAAC;AAOvC,MAAM,UAAU,iBAAiB,CAAC,KAAa,EAAE,KAAa;IAC5D,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;QAChB,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC;IAC9C,CAAC;IAED,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,SAAS,CACnC;QACE,MAAM,EAAE,GAAG,KAAK,gGAAgG;QAChH,eAAe,EAAE,KAAK;QACtB,UAAU,EAAE,IAAI;KACjB,EACD,WAAW,CAAC,OAAO,CAAC,cAAc,CACnC,CAAC;IACF,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAEpB,OAAO;QACL,MAAM,CAAC,OAAe;YACpB,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;QACD,IAAI,CAAC,YAAqB;YACxB,GAAG,CAAC,IAAI,EAAE,CAAC;YACX,IAAI,YAAY;gBAAE,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAC9C,CAAC;KACF,CAAC;AACJ,CAAC"}
|