feat: add progress to memory status deep

This commit is contained in:
Peter Steinberger
2026-01-17 20:25:19 +00:00
parent e0158c5d5d
commit fb393c3c51

View File

@@ -2,6 +2,7 @@ import type { Command } from "commander";
import { resolveDefaultAgentId } from "../agents/agent-scope.js"; import { resolveDefaultAgentId } from "../agents/agent-scope.js";
import { loadConfig } from "../config/config.js"; import { loadConfig } from "../config/config.js";
import { withProgress } from "./progress.js";
import { getMemorySearchManager } from "../memory/index.js"; import { getMemorySearchManager } from "../memory/index.js";
import { defaultRuntime } from "../runtime.js"; import { defaultRuntime } from "../runtime.js";
import { formatDocsLink } from "../terminal/links.js"; import { formatDocsLink } from "../terminal/links.js";
@@ -46,18 +47,32 @@ export function registerMemoryCli(program: Command) {
return; return;
} }
try { try {
await manager.probeVectorAvailability();
const deep = Boolean(opts.deep || opts.index); const deep = Boolean(opts.deep || opts.index);
const embeddingProbe = deep ? await manager.probeEmbeddingAvailability() : undefined; let embeddingProbe: Awaited<ReturnType<typeof manager.probeEmbeddingAvailability>> | undefined;
let indexError: string | undefined; let indexError: string | undefined;
if (opts.index) { if (deep) {
try { const total = opts.index ? 3 : 2;
await manager.sync({ reason: "cli" }); await withProgress({ label: "Checking memory…", total }, async (progress) => {
} catch (err) { progress.setLabel("Probing vector…");
indexError = err instanceof Error ? err.message : String(err); await manager.probeVectorAvailability();
defaultRuntime.error(`Memory index failed: ${indexError}`); progress.tick();
process.exitCode = 1; progress.setLabel("Probing embeddings…");
} embeddingProbe = await manager.probeEmbeddingAvailability();
progress.tick();
if (opts.index) {
progress.setLabel("Indexing memory…");
try {
await manager.sync({ reason: "cli" });
} catch (err) {
indexError = err instanceof Error ? err.message : String(err);
defaultRuntime.error(`Memory index failed: ${indexError}`);
process.exitCode = 1;
}
progress.tick();
}
});
} else {
await manager.probeVectorAvailability();
} }
const status = manager.status(); const status = manager.status();
if (opts.json) { if (opts.json) {