feat: add progress to memory status deep
This commit is contained in:
@@ -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) {
|
||||||
|
|||||||
Reference in New Issue
Block a user