fix: show memory index counts in progress

This commit is contained in:
Peter Steinberger
2026-01-17 21:09:22 +00:00
parent e932772230
commit 40345642fa
2 changed files with 20 additions and 1 deletions

View File

@@ -145,6 +145,7 @@ describe("memory embedding batches", () => {
});
expect(updates.length).toBeGreaterThan(0);
expect(updates.some((update) => update.label?.includes("/"))).toBe(true);
const last = updates[updates.length - 1];
expect(last?.total).toBeGreaterThan(0);
expect(last?.completed).toBe(last?.total);

View File

@@ -69,6 +69,7 @@ type MemorySyncProgressUpdate = {
type MemorySyncProgressState = {
completed: number;
total: number;
label?: string;
report: (update: MemorySyncProgressUpdate) => void;
};
@@ -799,7 +800,24 @@ export class MemoryIndexManager {
progress?: (update: MemorySyncProgressUpdate) => void;
}) {
const progress: MemorySyncProgressState | null = params?.progress
? { completed: 0, total: 0, report: params.progress }
? {
completed: 0,
total: 0,
label: undefined,
report: (update) => {
if (!params.progress) return;
if (update.label) progress.label = update.label;
const label =
update.total > 0 && progress.label
? `${progress.label} ${update.completed}/${update.total}`
: progress.label;
params.progress({
completed: update.completed,
total: update.total,
label,
});
},
}
: null;
const vectorReady = await this.ensureVectorReady();
const meta = this.readMeta();