mirror of
https://github.com/rohitg00/agentmemory.git
synced 2026-09-14 20:16:33 +08:00
Rohit Ghumare
d626b4ea60
* perf(hooks): fire-and-forget telemetry hooks (closes #573) Telemetry hooks (notification, post-tool-failure, post-tool-use, prompt-submit, stop, session-end, subagent-start, subagent-stop, task-completed) previously `await fetch(..., AbortSignal.timeout(N))` inside a try/catch. The await kept the hook process alive until the response arrived — up to N ms per request — which blocks Claude Code's next-prompt boundary on every assistant turn. Switch to fire-and-forget: fetch(url, { signal: AbortSignal.timeout(N) }).catch(() => {}); setTimeout(() => process.exit(0), 500).unref(); The unawaited fetch dispatches the request; the unref'd setTimeout force-exits the process after the request has been flushed to the local daemon's socket buffer (~500ms is enough). Without the setTimeout Node keeps the event loop alive waiting for any in-flight fetch to settle, which means the hook still blocks Claude Code's next-prompt boundary for up to the AbortSignal duration. Context-injecting hooks (pre-tool-use, pre-compact, session-start) still use `await fetch` because Claude Code reads their stdout for context injection — left untouched. AGENTS.md updated with the two-pattern guidance. * chore(hooks): drop verbose comments on fire-and-forget hooks * fix(hooks): bump stop+session-end exit delay to 1500ms Multi-request hooks (stop fires 2, session-end up to 4) need more than 500ms to initiate all fetches when AGENTMEMORY_URL points to a remote daemon — DNS + TCP + TLS handshakes can eat the budget before the second/third fetch is even dispatched. Bump to 1500ms on those two hooks only; single-request hooks keep 500ms. AGENTS.md updated with the multi-request exception.
75 lines
2.1 KiB
JavaScript
Executable File
75 lines
2.1 KiB
JavaScript
Executable File
#!/usr/bin/env node
|
|
import { execSync } from "node:child_process";
|
|
import { basename } from "node:path";
|
|
|
|
//#region src/hooks/_project.ts
|
|
function resolveProject(cwd) {
|
|
const explicit = process.env["AGENTMEMORY_PROJECT_NAME"];
|
|
if (explicit && explicit.trim()) return explicit.trim();
|
|
const dir = cwd && cwd.trim() ? cwd : process.cwd();
|
|
try {
|
|
const top = execSync("git rev-parse --show-toplevel", {
|
|
cwd: dir,
|
|
stdio: [
|
|
"ignore",
|
|
"pipe",
|
|
"ignore"
|
|
],
|
|
timeout: 500
|
|
}).toString().trim();
|
|
if (top) return basename(top);
|
|
} catch {}
|
|
return basename(dir);
|
|
}
|
|
|
|
//#endregion
|
|
//#region src/hooks/task-completed.ts
|
|
function isSdkChildContext(payload) {
|
|
if (process.env["AGENTMEMORY_SDK_CHILD"] === "1") return true;
|
|
if (!payload || typeof payload !== "object") return false;
|
|
return payload.entrypoint === "sdk-ts";
|
|
}
|
|
const REST_URL = process.env["AGENTMEMORY_URL"] || "http://localhost:3111";
|
|
const SECRET = process.env["AGENTMEMORY_SECRET"] || "";
|
|
function authHeaders() {
|
|
const h = { "Content-Type": "application/json" };
|
|
if (SECRET) h["Authorization"] = `Bearer ${SECRET}`;
|
|
return h;
|
|
}
|
|
async function main() {
|
|
let input = "";
|
|
for await (const chunk of process.stdin) input += chunk;
|
|
let data;
|
|
try {
|
|
data = JSON.parse(input);
|
|
} catch {
|
|
return;
|
|
}
|
|
if (isSdkChildContext(data)) return;
|
|
const sessionId = data.session_id || "unknown";
|
|
fetch(`${REST_URL}/agentmemory/observe`, {
|
|
method: "POST",
|
|
headers: authHeaders(),
|
|
body: JSON.stringify({
|
|
hookType: "task_completed",
|
|
sessionId,
|
|
project: resolveProject(data.cwd),
|
|
cwd: data.cwd || process.cwd(),
|
|
timestamp: (/* @__PURE__ */ new Date()).toISOString(),
|
|
data: {
|
|
task_id: data.task_id,
|
|
task_subject: data.task_subject,
|
|
task_description: typeof data.task_description === "string" ? data.task_description.slice(0, 2e3) : "",
|
|
teammate_name: data.teammate_name,
|
|
team_name: data.team_name
|
|
}
|
|
}),
|
|
signal: AbortSignal.timeout(2e3)
|
|
}).catch(() => {});
|
|
setTimeout(() => process.exit(0), 500).unref();
|
|
}
|
|
main();
|
|
|
|
//#endregion
|
|
export { };
|
|
//# sourceMappingURL=task-completed.mjs.map
|