@jphutchins/code-review 0.1.0-alpha.33 → 0.1.0-alpha.35

package/dist/index.js

@@ -773,24 +773,35 @@ var DEFAULT_RESERVE = {
   frac: 0.15,
   growth: 0.25,
   flatUsd: 0.02,
-  flatMs: 12e4
+  flatMs: 12e4,
+  flatMem: 2 * 1024 * 1024 * 1024
 };
 var SOFT_MULTIPLE = 2;
-var costAxis = (i) => i.spentUsd !== null && i.budgetUsd !== null && i.budgetUsd > 0 ? { used: i.spentUsd, limit: i.budgetUsd, flat: i.reserve.flatUsd } : null;
-var timeAxis = (i) => i.elapsedMs !== null && i.wallMs !== null && i.wallMs > 0 ? { used: i.elapsedMs, limit: i.wallMs, flat: i.reserve.flatMs } : null;
-var axisSeverity = (a, reserve) => {
-  const usedFrac = Math.min(1, Math.max(0, a.used / a.limit));
-  const effFrac = reserve.frac + reserve.growth * usedFrac;
-  const hardReserve = Math.max(a.flat, effFrac * a.limit);
+var growingReserve = (used, limit, flat, r) => {
+  const usedFrac = Math.min(1, Math.max(0, used / limit));
+  return Math.max(flat, (r.frac + r.growth * usedFrac) * limit);
+};
+var costAxis = (i) => i.spentUsd !== null && i.budgetUsd !== null && i.budgetUsd > 0 ? {
+  used: i.spentUsd,
+  limit: i.budgetUsd,
+  hardReserve: growingReserve(i.spentUsd, i.budgetUsd, i.reserve.flatUsd, i.reserve)
+} : null;
+var timeAxis = (i) => i.elapsedMs !== null && i.wallMs !== null && i.wallMs > 0 ? {
+  used: i.elapsedMs,
+  limit: i.wallMs,
+  hardReserve: growingReserve(i.elapsedMs, i.wallMs, i.reserve.flatMs, i.reserve)
+} : null;
+var axisSeverity = (a) => {
   const remaining = a.limit - a.used;
-  if (remaining <= hardReserve) return 2;
-  if (remaining <= SOFT_MULTIPLE * hardReserve) return 1;
+  if (remaining <= a.hardReserve) return 2;
+  if (remaining <= SOFT_MULTIPLE * a.hardReserve) return 1;
   return 0;
 };
 var decideBudget = (i) => {
-  const worst = [costAxis(i), timeAxis(i)].filter((a) => a !== null).reduce((max, a) => Math.max(max, axisSeverity(a, i.reserve)), 0);
+  const worst = [costAxis(i), timeAxis(i)].filter((a) => a !== null).reduce((max, a) => Math.max(max, axisSeverity(a)), 0);
   return worst === 2 ? { kind: "hard" } : worst === 1 ? { kind: "soft" } : { kind: "ok" };
 };
+var memoryCritical = (availMemBytes, totalMemBytes, floorBytes) => availMemBytes !== null && totalMemBytes !== null && totalMemBytes > 0 && availMemBytes <= floorBytes;
 var pct = (n) => `${String(Math.round(n * 100))}%`;
 var money = (n) => `$${n.toFixed(2)}`;
 var mins = (ms) => `${(ms / 6e4).toFixed(1)}m`;
@@ -845,6 +856,7 @@ var lastValidPath = (draftPath) => {
 };
 var mainHasWrittenDraft = (draftMtimeMs, seedMarkerMtimeMs) => draftMtimeMs !== null && (seedMarkerMtimeMs === null || draftMtimeMs > seedMarkerMtimeMs);
 var spawnFloorMessage = (draftPath) => `Write your own first-pass findings to ${draftPath} before spawning subagents \u2014 a review must never depend on subagents alone, and a pre-seeded draft does not count until you have revised it yourself this run. Write ${draftPath} from what you have read so far (preliminary findings are fine), run \`code-review validate ${draftPath} --explain\` until it passes, then fan out; your subagents run in the background, so keep refining the draft as their reports arrive.`;
+var memoryPressureMessage = (draftPath) => `System memory is critically low right now, so another subagent can't be spawned \u2014 a fresh process is the fastest way to tip the runner into an out-of-memory kill that would lose the whole review. Don't wind down: keep reading code directly and keep folding the reports from subagents already running into ${draftPath}, then try spawning again in a moment \u2014 this clears as soon as running subagents finish and free their memory.`;
 var forceBackgroundSpawn = (toolInput) => ({
   hookSpecificOutput: {
     hookEventName: "PreToolUse",
@@ -890,6 +902,8 @@ var evaluateBudgetHook = (input, params) => {
       if (phase.kind === "hard" && blockedDuringConvergence(toolName, rec["tool_input"]))
         return denyPreTool(budgetMessage(inputs, phase, params.draftPath, isSubagent));
       if (SPAWN_TOOLS.has(toolName)) {
+        if (memoryCritical(params.availMemBytes, params.totalMemBytes, params.reserve.flatMem))
+          return denyPreTool(memoryPressureMessage(params.draftPath));
         if (!isSubagent && !params.mainDraftWritten)
           return denyPreTool(spawnFloorMessage(params.draftPath));
         return forceBackgroundSpawn(rec["tool_input"]);
@@ -936,6 +950,21 @@ var parseFraction = (raw, fallback) => {
   const n = Number.parseFloat(raw);
   return Number.isFinite(n) && n >= 0 && n <= 1 ? n : fallback;
 };
+var BYTE_UNIT = {
+  "": 1,
+  k: 1024,
+  m: 1024 * 1024,
+  g: 1024 * 1024 * 1024,
+  t: 1024 * 1024 * 1024 * 1024
+};
+var parseByteSize = (raw) => {
+  const m = /^(\d+(?:\.\d+)?)\s*([kmgt])?(?:i?b)?$/i.exec(raw.trim());
+  if (m === null) return null;
+  const [, num = "", unit = ""] = m;
+  const n = Number.parseFloat(num);
+  const mult = BYTE_UNIT[unit.toLowerCase()];
+  return Number.isFinite(n) && mult !== void 0 ? n * mult : null;
+};
 var budgetHookCommand = (draftPath, opts) => [
   "code-review budget-hook --draft",
   shellQuote(draftPath),
@@ -945,7 +974,8 @@ var budgetHookCommand = (draftPath, opts) => [
   ...opts.reserveFrac ? ["--reserve-frac", shellQuote(opts.reserveFrac)] : [],
   ...opts.reserveGrowth ? ["--reserve-growth", shellQuote(opts.reserveGrowth)] : [],
   ...opts.reserveUsd ? ["--reserve-usd", shellQuote(opts.reserveUsd)] : [],
-  ...opts.reserveWall ? ["--reserve-wall", shellQuote(opts.reserveWall)] : []
+  ...opts.reserveWall ? ["--reserve-wall", shellQuote(opts.reserveWall)] : [],
+  ...opts.reserveMem ? ["--reserve-mem", shellQuote(opts.reserveMem)] : []
 ].join(" ");
 
 // src/format.ts
@@ -3002,6 +3032,18 @@ var snapshotIfValid = (draftPath) => {
     );
   }
 };
+var readMemInfo = () => {
+  try {
+    const text = readFileSync("/proc/meminfo", "utf8");
+    const kb = (key2) => {
+      const m = new RegExp(`^${key2}:\\s+(\\d+)\\s+kB`, "m").exec(text);
+      return m?.[1] !== void 0 ? Number(m[1]) * 1024 : null;
+    };
+    return { availBytes: kb("MemAvailable"), totalBytes: kb("MemTotal") };
+  } catch {
+    return { availBytes: null, totalBytes: null };
+  }
+};
 var budgetHookCmd = defineCommand({
   meta: {
     name: "budget-hook",
@@ -3040,12 +3082,17 @@ var budgetHookCmd = defineCommand({
     "reserve-wall": {
       type: "string",
       description: "Flat wall-clock wind-down floor (e.g. 2m, 120s), whichever is larger with --reserve-frac (default: 2m)"
+    },
+    "reserve-mem": {
+      type: "string",
+      description: "Free-RAM floor (e.g. 2g, 1536m) below which new subagent spawns are denied until memory recovers; not a convergence axis (default: 2g)"
     }
   },
   run: async ({ args }) => {
     try {
       const draftPath = resolve$1(args.draft);
       const input = readStdinJSON();
+      const mem = readMemInfo();
       const transcriptPath = transcriptPathOf(input);
       const tree = transcriptPath ? readTranscriptTree(resolve$1(transcriptPath)) : void 0;
       const usage = tree ? sumTranscriptUsage(tree.entries) : void 0;
@@ -3062,11 +3109,14 @@ var budgetHookCmd = defineCommand({
           nowMs: Date.now()
         }),
         wallMs,
+        availMemBytes: mem.availBytes,
+        totalMemBytes: mem.totalBytes,
         reserve: {
           frac: parseFraction(args["reserve-frac"], DEFAULT_RESERVE.frac),
           growth: parseFraction(args["reserve-growth"], DEFAULT_RESERVE.growth),
           flatUsd: parseBudgetUsd(args["reserve-usd"]) ?? DEFAULT_RESERVE.flatUsd,
-          flatMs: args["reserve-wall"] ? parseWallMs(args["reserve-wall"]) ?? DEFAULT_RESERVE.flatMs : DEFAULT_RESERVE.flatMs
+          flatMs: args["reserve-wall"] ? parseWallMs(args["reserve-wall"]) ?? DEFAULT_RESERVE.flatMs : DEFAULT_RESERVE.flatMs,
+          flatMem: args["reserve-mem"] ? parseByteSize(args["reserve-mem"]) ?? DEFAULT_RESERVE.flatMem : DEFAULT_RESERVE.flatMem
         },
         draftPath,
         mainDraftWritten: mainHasWrittenDraft(
@@ -3143,6 +3193,10 @@ var printSettingsCmd = defineCommand({
     "reserve-wall": {
       type: "string",
       description: "Flat wall-clock wind-down floor (e.g. 2m), whichever is larger with --reserve-frac (default: 2m)"
+    },
+    "reserve-mem": {
+      type: "string",
+      description: "Free-RAM floor (e.g. 2g) below which new subagent spawns are denied until memory recovers (default: 2g)"
     }
   },
   run: async ({ args }) => {
@@ -3164,7 +3218,8 @@ var printSettingsCmd = defineCommand({
         reserveFrac: args["reserve-frac"],
         reserveGrowth: args["reserve-growth"],
         reserveUsd: args["reserve-usd"],
-        reserveWall: args["reserve-wall"]
+        reserveWall: args["reserve-wall"],
+        reserveMem: args["reserve-mem"]
       }
     });
     process.stdout.write(`${JSON.stringify(settings)}