Compare commits

...

1 Commits

Author SHA1 Message Date
Peter Steinberger
a535be7832 refactor: centralize cli manager cleanup
Co-authored-by: Nicholas Spisak <jsnsdirect@gmail.com>
2026-01-18 00:11:45 +00:00
4 changed files with 293 additions and 193 deletions

View File

@ -47,6 +47,7 @@ Docs: https://docs.clawd.bot
### Changes ### Changes
- CLI: stamp build commit into dist metadata so banners show the commit in npm installs. - CLI: stamp build commit into dist metadata so banners show the commit in npm installs.
- CLI: close memory manager after memory commands to avoid hanging processes. (#1127) — thanks @NicholasSpisak.
## 2026.1.16-1 ## 2026.1.16-1

31
src/cli/cli-utils.ts Normal file
View File

@ -0,0 +1,31 @@
export type ManagerLookupResult<T> = {
manager: T | null;
error?: string;
};
export function formatErrorMessage(err: unknown): string {
return err instanceof Error ? err.message : String(err);
}
export async function withManager<T>(params: {
getManager: () => Promise<ManagerLookupResult<T>>;
onMissing: (error?: string) => void;
run: (manager: T) => Promise<void>;
close: (manager: T) => Promise<void>;
onCloseError?: (err: unknown) => void;
}): Promise<void> {
const { manager, error } = await params.getManager();
if (!manager) {
params.onMissing(error);
return;
}
try {
await params.run(manager);
} finally {
try {
await params.close(manager);
} catch (err) {
params.onCloseError?.(err);
}
}
}

View File

@ -135,6 +135,42 @@ describe("memory cli", () => {
expect(close).toHaveBeenCalled(); expect(close).toHaveBeenCalled();
}); });
it("logs close failure after status", async () => {
const { registerMemoryCli } = await import("./memory-cli.js");
const { defaultRuntime } = await import("../runtime.js");
const close = vi.fn(async () => {
throw new Error("close boom");
});
getMemorySearchManager.mockResolvedValueOnce({
manager: {
probeVectorAvailability: vi.fn(async () => true),
status: () => ({
files: 1,
chunks: 1,
dirty: false,
workspaceDir: "/tmp/clawd",
dbPath: "/tmp/memory.sqlite",
provider: "openai",
model: "text-embedding-3-small",
requestedProvider: "openai",
}),
close,
},
});
const error = vi.spyOn(defaultRuntime, "error").mockImplementation(() => {});
const program = new Command();
program.name("test");
registerMemoryCli(program);
await program.parseAsync(["memory", "status"], { from: "user" });
expect(close).toHaveBeenCalled();
expect(error).toHaveBeenCalledWith(
expect.stringContaining("Memory manager close failed: close boom"),
);
expect(process.exitCode).toBeUndefined();
});
it("reindexes on status --index", async () => { it("reindexes on status --index", async () => {
const { registerMemoryCli } = await import("./memory-cli.js"); const { registerMemoryCli } = await import("./memory-cli.js");
const { defaultRuntime } = await import("../runtime.js"); const { defaultRuntime } = await import("../runtime.js");
@ -225,6 +261,42 @@ describe("memory cli", () => {
expect(process.exitCode).toBeUndefined(); expect(process.exitCode).toBeUndefined();
}); });
it("logs close failure after search", async () => {
const { registerMemoryCli } = await import("./memory-cli.js");
const { defaultRuntime } = await import("../runtime.js");
const close = vi.fn(async () => {
throw new Error("close boom");
});
const search = vi.fn(async () => [
{
path: "memory/2026-01-12.md",
startLine: 1,
endLine: 2,
score: 0.5,
snippet: "Hello",
},
]);
getMemorySearchManager.mockResolvedValueOnce({
manager: {
search,
close,
},
});
const error = vi.spyOn(defaultRuntime, "error").mockImplementation(() => {});
const program = new Command();
program.name("test");
registerMemoryCli(program);
await program.parseAsync(["memory", "search", "hello"], { from: "user" });
expect(search).toHaveBeenCalled();
expect(close).toHaveBeenCalled();
expect(error).toHaveBeenCalledWith(
expect.stringContaining("Memory manager close failed: close boom"),
);
expect(process.exitCode).toBeUndefined();
});
it("closes manager after search error", async () => { it("closes manager after search error", async () => {
const { registerMemoryCli } = await import("./memory-cli.js"); const { registerMemoryCli } = await import("./memory-cli.js");
const { defaultRuntime } = await import("../runtime.js"); const { defaultRuntime } = await import("../runtime.js");

View File

@ -3,6 +3,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, withProgressTotals } from "./progress.js"; import { withProgress, withProgressTotals } from "./progress.js";
import { formatErrorMessage, withManager } from "./cli-utils.js";
import { getMemorySearchManager, type MemorySearchManagerResult } from "../memory/index.js"; import { getMemorySearchManager, type MemorySearchManagerResult } 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";
@ -23,34 +24,6 @@ function resolveAgent(cfg: ReturnType<typeof loadConfig>, agent?: string) {
return resolveDefaultAgentId(cfg); return resolveDefaultAgentId(cfg);
} }
function formatErrorMessage(err: unknown): string {
return err instanceof Error ? err.message : String(err);
}
async function closeManager(manager: MemoryManager): Promise<void> {
try {
await manager.close();
} catch (err) {
defaultRuntime.error(`Memory manager close failed: ${formatErrorMessage(err)}`);
}
}
async function withMemoryManager(
params: { cfg: ReturnType<typeof loadConfig>; agentId: string },
run: (manager: MemoryManager) => Promise<void>,
): Promise<void> {
const { manager, error } = await getMemorySearchManager(params);
if (!manager) {
defaultRuntime.log(error ?? "Memory search disabled.");
return;
}
try {
await run(manager);
} finally {
await closeManager(manager);
}
}
export function registerMemoryCli(program: Command) { export function registerMemoryCli(program: Command) {
const memory = program const memory = program
.command("memory") .command("memory")
@ -71,9 +44,17 @@ export function registerMemoryCli(program: Command) {
.action(async (opts: MemoryCommandOptions) => { .action(async (opts: MemoryCommandOptions) => {
const cfg = loadConfig(); const cfg = loadConfig();
const agentId = resolveAgent(cfg, opts.agent); const agentId = resolveAgent(cfg, opts.agent);
await withMemoryManager({ cfg, agentId }, async (manager) => { await withManager<MemoryManager>({
getManager: () => getMemorySearchManager({ cfg, agentId }),
onMissing: (error) => defaultRuntime.log(error ?? "Memory search disabled."),
onCloseError: (err) =>
defaultRuntime.error(`Memory manager close failed: ${formatErrorMessage(err)}`),
close: (manager) => manager.close(),
run: async (manager) => {
const deep = Boolean(opts.deep || opts.index); const deep = Boolean(opts.deep || opts.index);
let embeddingProbe: Awaited<ReturnType<typeof manager.probeEmbeddingAvailability>> | undefined; let embeddingProbe:
| Awaited<ReturnType<typeof manager.probeEmbeddingAvailability>>
| undefined;
let indexError: string | undefined; let indexError: string | undefined;
if (deep) { if (deep) {
await withProgress({ label: "Checking memory…", total: 2 }, async (progress) => { await withProgress({ label: "Checking memory…", total: 2 }, async (progress) => {
@ -200,6 +181,7 @@ export function registerMemoryCli(program: Command) {
lines.push(`${label("Index error")} ${warn(indexError)}`); lines.push(`${label("Index error")} ${warn(indexError)}`);
} }
defaultRuntime.log(lines.join("\n")); defaultRuntime.log(lines.join("\n"));
},
}); });
}); });
@ -211,7 +193,13 @@ export function registerMemoryCli(program: Command) {
.action(async (opts: MemoryCommandOptions & { force?: boolean }) => { .action(async (opts: MemoryCommandOptions & { force?: boolean }) => {
const cfg = loadConfig(); const cfg = loadConfig();
const agentId = resolveAgent(cfg, opts.agent); const agentId = resolveAgent(cfg, opts.agent);
await withMemoryManager({ cfg, agentId }, async (manager) => { await withManager<MemoryManager>({
getManager: () => getMemorySearchManager({ cfg, agentId }),
onMissing: (error) => defaultRuntime.log(error ?? "Memory search disabled."),
onCloseError: (err) =>
defaultRuntime.error(`Memory manager close failed: ${formatErrorMessage(err)}`),
close: (manager) => manager.close(),
run: async (manager) => {
try { try {
await manager.sync({ reason: "cli", force: opts.force }); await manager.sync({ reason: "cli", force: opts.force });
defaultRuntime.log("Memory index updated."); defaultRuntime.log("Memory index updated.");
@ -220,6 +208,7 @@ export function registerMemoryCli(program: Command) {
defaultRuntime.error(`Memory index failed: ${message}`); defaultRuntime.error(`Memory index failed: ${message}`);
process.exitCode = 1; process.exitCode = 1;
} }
},
}); });
}); });
@ -241,7 +230,13 @@ export function registerMemoryCli(program: Command) {
) => { ) => {
const cfg = loadConfig(); const cfg = loadConfig();
const agentId = resolveAgent(cfg, opts.agent); const agentId = resolveAgent(cfg, opts.agent);
await withMemoryManager({ cfg, agentId }, async (manager) => { await withManager<MemoryManager>({
getManager: () => getMemorySearchManager({ cfg, agentId }),
onMissing: (error) => defaultRuntime.log(error ?? "Memory search disabled."),
onCloseError: (err) =>
defaultRuntime.error(`Memory manager close failed: ${formatErrorMessage(err)}`),
close: (manager) => manager.close(),
run: async (manager) => {
let results: Awaited<ReturnType<typeof manager.search>>; let results: Awaited<ReturnType<typeof manager.search>>;
try { try {
results = await manager.search(query, { results = await manager.search(query, {
@ -276,6 +271,7 @@ export function registerMemoryCli(program: Command) {
lines.push(""); lines.push("");
} }
defaultRuntime.log(lines.join("\n").trim()); defaultRuntime.log(lines.join("\n").trim());
},
}); });
}, },
); );