#!/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/post-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 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 || data.sessionId || "unknown"; const toolName = data.tool_name ?? data.toolName; const toolInput = data.tool_input ?? data.toolArgs; const { imageData, cleanOutput } = extractImageData(toolOutput(data)); fetch(`${REST_URL}/agentmemory/observe`, { method: "POST", headers: authHeaders(), body: JSON.stringify({ hookType: "post_tool_use", sessionId, project: resolveProject(data.cwd), cwd: data.cwd || process.cwd(), timestamp: (/* @__PURE__ */ new Date()).toISOString(), data: { tool_name: toolName, tool_input: toolInput, tool_output: truncate(cleanOutput, 8e3), ...imageData ? { image_data: imageData } : {} } }), signal: AbortSignal.timeout(3e3) }).catch(() => {}); setTimeout(() => process.exit(0), 500).unref(); } function toolOutput(data) { if (data.tool_response !== void 0) return data.tool_response; if (data.tool_output !== void 0) return data.tool_output; const result = data.tool_result ?? data.toolResult; if (typeof result === "object" && result !== null) { const obj = result; return obj.text_result_for_llm ?? obj.textResultForLlm ?? result; } return result; } function isBase64Image(val) { return typeof val === "string" && (val.startsWith("data:image/") || val.startsWith("iVBORw0KGgo") || val.startsWith("/9j/")); } function extractImageData(output) { if (isBase64Image(output)) return { imageData: output, cleanOutput: "[image data extracted]" }; if (typeof output === "object" && output !== null && !Array.isArray(output)) { const obj = output; let imageData; const clean = {}; for (const [key, val] of Object.entries(obj)) if (!imageData && isBase64Image(val)) { imageData = val; clean[key] = "[image data extracted]"; } else clean[key] = val; return { imageData, cleanOutput: clean }; } return { imageData: void 0, cleanOutput: output }; } function truncate(value, max) { if (typeof value === "string" && value.length > max) return value.slice(0, max) + "\n[...truncated]"; if (typeof value === "object" && value !== null) { const str = JSON.stringify(value); if (str.length > max) return str.slice(0, max) + "...[truncated]"; return value; } return value; } main(); //#endregion export { }; //# sourceMappingURL=post-tool-use.mjs.map