mirror of
https://github.com/rohitg00/agentmemory.git
synced 2026-09-14 20:16:33 +08:00
Rohit Ghumare
6761a99ba1
#1047: JSON.parse("null") returns null without throwing, so every hook's parse guard passed it through and the first data.xxx access threw a TypeError. Bare main() turned that into an unhandled rejection -> exit 1 -> host reported 'hook failed' on every affected tool call. All 13 hook entrypoints now guard non-object payloads before dereferencing and wrap main() in .catch() to fail closed (silent exit 0). #1057: mem::context and api::context filtered candidate sessions by project only, leaking cross-agent observations/summaries under AGENTMEMORY_AGENT_SCOPE=isolated. Now applies the same agent-scope filter as mem::search (#817); api::context, api::session::start, and event::session::started forward agentId. Also: bump 0.9.28 across manifests/deploy/export-import set; refresh stale README/AGENTS stats (files/LOC/functions/KV; AGENTS tests 950+ -> 1,428+) and regenerate the website meta snapshot to 0.9.28; CHANGELOG 0.9.28 section; remove the rate-limited star-history chart from README and all 11 translations.
85 lines
2.7 KiB
JavaScript
Executable File
85 lines
2.7 KiB
JavaScript
Executable File
#!/usr/bin/env node
|
|
//#region src/hooks/pre-tool-use.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 INJECT_CONTEXT = process.env["AGENTMEMORY_INJECT_CONTEXT"] === "true";
|
|
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() {
|
|
if (!INJECT_CONTEXT) return;
|
|
let input = "";
|
|
for await (const chunk of process.stdin) input += chunk;
|
|
let data;
|
|
try {
|
|
data = JSON.parse(input);
|
|
} catch {
|
|
return;
|
|
}
|
|
if (!data || typeof data !== "object") return;
|
|
if (isSdkChildContext(data)) return;
|
|
const toolName = typeof data.tool_name === "string" ? data.tool_name : typeof data.toolName === "string" ? data.toolName : void 0;
|
|
if (!toolName) return;
|
|
const normalizedToolName = toolName.toLowerCase();
|
|
if (![
|
|
"edit",
|
|
"write",
|
|
"create",
|
|
"read",
|
|
"view",
|
|
"glob",
|
|
"grep"
|
|
].includes(normalizedToolName)) return;
|
|
const rawToolInput = data.tool_input ?? data.toolArgs;
|
|
const toolInput = typeof rawToolInput === "object" && rawToolInput !== null && !Array.isArray(rawToolInput) ? rawToolInput : {};
|
|
const files = [];
|
|
const fileKeys = normalizedToolName === "grep" ? ["path", "file"] : [
|
|
"file_path",
|
|
"path",
|
|
"file",
|
|
"pattern"
|
|
];
|
|
for (const key of fileKeys) {
|
|
const val = toolInput[key];
|
|
if (typeof val === "string" && val.length > 0) files.push(val);
|
|
}
|
|
if (files.length === 0) return;
|
|
const terms = [];
|
|
if (normalizedToolName === "grep" || normalizedToolName === "glob") {
|
|
const pattern = toolInput["pattern"];
|
|
if (typeof pattern === "string" && pattern.length > 0) terms.push(pattern);
|
|
}
|
|
const rawSessionId = data.session_id || data.sessionId;
|
|
const sessionId = typeof rawSessionId === "string" && rawSessionId.length > 0 ? rawSessionId : "unknown";
|
|
const project = typeof data.project === "string" && data.project.trim().length > 0 ? data.project.trim() : void 0;
|
|
try {
|
|
const res = await fetch(`${REST_URL}/agentmemory/enrich`, {
|
|
method: "POST",
|
|
headers: authHeaders(),
|
|
body: JSON.stringify({
|
|
sessionId,
|
|
files,
|
|
terms,
|
|
toolName,
|
|
...project !== void 0 && { project }
|
|
}),
|
|
signal: AbortSignal.timeout(2e3)
|
|
});
|
|
if (res.ok) {
|
|
const result = await res.json();
|
|
if (result.context) process.stdout.write(result.context);
|
|
}
|
|
} catch {}
|
|
}
|
|
main().catch(() => process.exit(0));
|
|
//#endregion
|
|
export {};
|
|
|
|
//# sourceMappingURL=pre-tool-use.mjs.map
|