@brainervirus/workit-pi 8.36.0 → 8.37.0

package/dist/worker.js

@@ -8209,21 +8209,60 @@ var uniformImplementerModel = (cwd, host, dir) => {
 };
 var ROLE_SELECTORS = MODEL_ROLES.flatMap((role) => role === "implementer" ? MODEL_TIERS.map((tier) => ({ role, tier })) : [{ role }]);
 
-// packages/workit-core/src/hooks/standing.ts
+// packages/workit-core/src/hooks/budget-signal.ts
+import { createHash as createHash4 } from "node:crypto";
 import fs4 from "node:fs";
 import path7 from "node:path";
+var hash = (text) => createHash4("sha256").update(text).digest("hex").slice(0, 32);
+var budgetDir = (store) => path7.join(store, "budget");
+var agentBase = (store, session, agentId) => path7.join(budgetDir(store), hash(`${session}\x00${agentId ?? ""}`));
+var spawnBudget = (prompt) => {
+  const found = /^TIMEBOX:[^\n]*?\b(\d+)\s+tool calls\b/mu.exec(prompt);
+  const n = found ? Number(found[1]) : NaN;
+  return Number.isInteger(n) && n > 0 ? n : null;
+};
+var startBudget = (store, session, agentId, n) => {
+  const dir = budgetDir(store);
+  fs4.mkdirSync(dir, { recursive: true });
+  const cutoff = Date.now() - 604800000;
+  for (const name of fs4.readdirSync(dir))
+    try {
+      if (fs4.statSync(path7.join(dir, name)).mtimeMs < cutoff)
+        fs4.rmSync(path7.join(dir, name));
+    } catch {}
+  const base = agentBase(store, session, agentId);
+  fs4.writeFileSync(`${base}.json`, JSON.stringify({ limit: n, told: 0 }));
+  fs4.rmSync(`${base}.calls`, { force: true });
+};
+var bindBudget = (cwd, session, agentId, prompt) => {
+  try {
+    const n = spawnBudget(prompt);
+    const store = n && session ? gitStoreDir(cwd) : null;
+    if (!n || !store)
+      return null;
+    startBudget(store, session, agentId, n);
+    return n;
+  } catch {
+    return null;
+  }
+};
+var CODEX_GATED = new Set(["bash", "unifiedexec", "shell", "applypatch", "edit", "write"]);
+
+// packages/workit-core/src/hooks/standing.ts
+import fs5 from "node:fs";
+import path8 from "node:path";
 var readPlans = (store) => {
-  const dir = path7.join(store, "fanouts");
+  const dir = path8.join(store, "fanouts");
   let names;
   try {
-    names = fs4.readdirSync(dir).filter((name) => name.endsWith(".json"));
+    names = fs5.readdirSync(dir).filter((name) => name.endsWith(".json"));
   } catch {
     return [];
   }
   const plans = [];
   for (const name of names) {
     try {
-      const raw = JSON.parse(fs4.readFileSync(path7.join(dir, name), "utf8"));
+      const raw = JSON.parse(fs5.readFileSync(path8.join(dir, name), "utf8"));
       if (typeof raw.name !== "string" || !Array.isArray(raw.slices))
         continue;
       const texts = (value) => Array.isArray(value) ? value.filter((item) => typeof item === "string" && item !== "") : [];
@@ -8454,6 +8493,8 @@ function launchWorker(assignment, options) {
     transcriptPath: null,
     event: { kind: "session.start", source: "startup" }
   });
+  bindBudget(spec.cwd, env.WORKIT_SESSION_ID, null, `${options.pendingPrompt ?? ""}
+${assignment.objective}`);
   try {
     const spawn = options.spawn ?? ((command, args, spawnOptions) => nodeSpawn(command, args, spawnOptions));
     handle.spec = { ...spec, env };

package/dist/workit.js

@@ -24609,7 +24609,8 @@ function handleHook(input, deps) {
   const decision = decide(input, deps);
   if (deps.descriptor.transport !== "hook-process" || !PRE_TOOL.has(input.event.kind) || decision.kind === "deny" || decision.kind === "ask")
     return decision;
-  const delta = standingDelta(input);
+  const delta = [standingDelta(input), deps.budget?.line(input) ?? null].filter(Boolean).join(`
+`) || null;
   if (!delta)
     return decision;
   return decision.kind === "rewrite" || decision.kind === "update-input" ? { ...decision, text: [decision.text, delta].filter(Boolean).join(`
@@ -24719,7 +24720,11 @@ function decide(input, deps) {
     case "subagent.start":
       if (workitRole(descriptor, event.agentType) === "judge")
         trackJudge(input, event.agentId, true);
-      seedStanding({ ...input, session: { ...input.session, agentId: event.agentId } });
+      seedStanding({
+        ...input,
+        session: { ...input.session, agentId: event.agentId }
+      });
+      deps.budget?.start(input, event.agentId, event.agentType);
       return {
         kind: "context",
         text: subagentStartText(input, descriptor, event)
@@ -24732,13 +24737,19 @@ function decide(input, deps) {
     case "tool.pre":
       if (event.skill)
         recordSkillLoad(input, event.skill, "tool");
-      if (event.agentInput && input.host === "claude_code")
-        return spawnDecision(descriptor, input.cwd, event.agentInput);
+      if (event.agentInput && input.host === "claude_code") {
+        const spawn = spawnDecision(descriptor, input.cwd, event.agentInput);
+        if (spawn.kind !== "deny")
+          deps.budget?.spawn(input, event.agentInput);
+        return spawn;
+      }
       return NONE6;
     case "stop":
       return usable2(descriptor.events.stop.support) ? stopDecision(input, descriptor) : NONE6;
     case "subagent.stop":
       trackJudge(input, event.agentId, false);
+      if (event.agentId)
+        deps.budget?.stop(input, event.agentId);
       return NONE6;
     case "prompt.submit":
       return NONE6;
@@ -24856,6 +24867,88 @@ var modelKey = (model) => {
   return typeof source === "string" && source ? `${source}/${id}` : id;
 };
 
+// packages/workit-core/src/hooks/budget-signal.ts
+import { createHash as createHash18 } from "node:crypto";
+import fs26 from "node:fs";
+import path35 from "node:path";
+var hash2 = (text) => createHash18("sha256").update(text).digest("hex").slice(0, 32);
+var budgetDir = (store) => path35.join(store, "budget");
+var agentBase = (store, session, agentId) => path35.join(budgetDir(store), hash2(`${session}\x00${agentId ?? ""}`));
+var spawnBudget = (prompt) => {
+  const found = /^TIMEBOX:[^\n]*?\b(\d+)\s+tool calls\b/mu.exec(prompt);
+  const n = found ? Number(found[1]) : NaN;
+  return Number.isInteger(n) && n > 0 ? n : null;
+};
+var readBudget = (file) => {
+  try {
+    const { limit, told } = JSON.parse(fs26.readFileSync(file, "utf8"));
+    return typeof limit === "number" && limit > 0 ? { limit, told: Number(told) || 0 } : null;
+  } catch {
+    return null;
+  }
+};
+var startBudget = (store, session, agentId, n) => {
+  const dir = budgetDir(store);
+  fs26.mkdirSync(dir, { recursive: true });
+  const cutoff = Date.now() - 604800000;
+  for (const name of fs26.readdirSync(dir))
+    try {
+      if (fs26.statSync(path35.join(dir, name)).mtimeMs < cutoff)
+        fs26.rmSync(path35.join(dir, name));
+    } catch {}
+  const base = agentBase(store, session, agentId);
+  fs26.writeFileSync(`${base}.json`, JSON.stringify({ limit: n, told: 0 }));
+  fs26.rmSync(`${base}.calls`, { force: true });
+};
+var bindBudget = (cwd, session, agentId, prompt) => {
+  try {
+    const n = spawnBudget(prompt);
+    const store = n && session ? gitStoreDir(cwd) : null;
+    if (!n || !store)
+      return null;
+    startBudget(store, session, agentId, n);
+    return n;
+  } catch {
+    return null;
+  }
+};
+var tickBudget = (cwd, session, agentId) => {
+  try {
+    const store = session ? gitStoreDir(cwd) : null;
+    if (!store)
+      return false;
+    const base = agentBase(store, session, agentId);
+    const budget = readBudget(`${base}.json`);
+    if (!budget)
+      return false;
+    fs26.appendFileSync(`${base}.calls`, ".");
+    return fs26.statSync(`${base}.calls`).size >= budget.limit * (budget.told + 1);
+  } catch {
+    return false;
+  }
+};
+var lineFor = (calls) => `Tool-call budget reached (${calls}): commit your work, write your report, stop.`;
+var reachedOf = (cwd, session, agentId) => {
+  const store = session ? gitStoreDir(cwd) : null;
+  const base = store && agentBase(store, session, agentId);
+  const budget = base && readBudget(`${base}.json`);
+  if (!budget)
+    return null;
+  return { base, budget, reached: Math.floor(fs26.statSync(`${base}.calls`).size / budget.limit) };
+};
+var budgetLine = (cwd, session, agentId) => {
+  try {
+    const at = reachedOf(cwd, session, agentId);
+    if (!at || at.reached <= at.budget.told)
+      return null;
+    fs26.writeFileSync(`${at.base}.json`, JSON.stringify({ ...at.budget, told: at.reached }));
+    return lineFor(at.budget.limit * at.reached);
+  } catch {
+    return null;
+  }
+};
+var CODEX_GATED = new Set(["bash", "unifiedexec", "shell", "applypatch", "edit", "write"]);
+
 // packages/workit-pi/src/context.ts
 var piCapabilities = (ctx) => capabilitiesFor(PI_DESCRIPTOR, { ui: ctx?.hasUI === true });
 var piContext = (ctx) => ({
@@ -24910,6 +25003,12 @@ var standingNotice = (ctx) => {
   const input = turnInput(ctx);
   return standingDelta(worker ? { ...input, session: { ...input.session, id: worker } } : input);
 };
+var budgetNotice = (ctx) => {
+  if (!ctx.isProjectTrusted())
+    return null;
+  const session = process.env.WORKIT_PI_WORKER_SESSION && process.env.WORKIT_SESSION_ID?.trim() || ctx.sessionManager.getSessionId();
+  return tickBudget(ctx.cwd, session, null) ? budgetLine(ctx.cwd, session, null) : null;
+};
 var workitContext = (ctx, taskContext = taskTurnContext(ctx)) => {
   const session = ctx.sessionManager.getSessionId();
   if (!ctx.isProjectTrusted())
@@ -24934,11 +25033,11 @@ var unfinishedTaskOffer2 = (ctx) => {
 };
 
 // packages/workit-core/src/hooks/repeat-failure.ts
-import { createHash as createHash18 } from "node:crypto";
+import { createHash as createHash19 } from "node:crypto";
 var SAME_CALL_FAILED = "Same call failed twice:";
 var MAX_SESSIONS = 256;
 var stable = (value) => JSON.stringify(value, (_key, entry) => entry && typeof entry === "object" && !Array.isArray(entry) ? Object.fromEntries(Object.entries(entry).toSorted(([a], [b]) => a.localeCompare(b))) : entry) ?? "null";
-var hashOf = (call) => createHash18("sha256").update(call).digest("hex").slice(0, 32);
+var hashOf = (call) => createHash19("sha256").update(call).digest("hex").slice(0, 32);
 var record2 = (memo, sessionId, call, failed, next) => {
   const previous = memo.get(sessionId);
   memo.delete(sessionId);
@@ -25138,12 +25237,12 @@ var skillPromptNudge = (prompt, ctx) => process.env.WORKIT_PI_WORKER_SESSION ? n
 // packages/workit-pi/src/launches.ts
 import { mkdirSync as mkdirSync3, readFileSync as readFileSync7, readdirSync as readdirSync5, rmSync as rmSync2, writeFileSync as writeFileSync3 } from "node:fs";
 import os11 from "node:os";
-import path35 from "node:path";
+import path36 from "node:path";
 var launchDir = (root) => {
   const location = resolveStore(root);
-  return location instanceof Error ? null : path35.join(location.dir, "pi-launches");
+  return location instanceof Error ? null : path36.join(location.dir, "pi-launches");
 };
-var fileFor = (dir, workerId) => path35.join(dir, `${workerId.replace(/[^A-Za-z0-9-]/g, "_")}.json`);
+var fileFor = (dir, workerId) => path36.join(dir, `${workerId.replace(/[^A-Za-z0-9-]/g, "_")}.json`);
 var recordLaunch = (root, attempt) => {
   const dir = launchDir(root);
   if (!dir)
@@ -25189,7 +25288,7 @@ var abandonedLaunches = (root) => {
   const found = [];
   for (const name of names) {
     try {
-      const attempt = JSON.parse(readFileSync7(path35.join(dir, name), "utf8"));
+      const attempt = JSON.parse(readFileSync7(path36.join(dir, name), "utf8"));
       if (typeof attempt.workerId === "string" && typeof attempt.session === "string") {
         if (!alive(attempt))
           found.push(attempt);
@@ -25379,6 +25478,8 @@ function launchWorker(assignment, options) {
     transcriptPath: null,
     event: { kind: "session.start", source: "startup" }
   });
+  bindBudget(spec.cwd, env.WORKIT_SESSION_ID, null, `${options.pendingPrompt ?? ""}
+${assignment.objective}`);
   try {
     const spawn = options.spawn ?? ((command, args, spawnOptions) => nodeSpawn(command, args, spawnOptions));
     handle.spec = { ...spec, env };
@@ -26129,7 +26230,8 @@ ${nudge}` : content) };
     const observed = observeResult(event, ctx);
     let standing = null;
     try {
-      standing = standingNotice(ctx);
+      standing = [standingNotice(ctx), budgetNotice(ctx)].filter(Boolean).join(`
+`) || null;
     } catch {}
     if (!standing)
       return observed;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@brainervirus/workit-pi",
-  "version": "8.36.0",
+  "version": "8.37.0",
   "description": "Workit native extension for Pi — tools, method skills, and session continuity",
   "keywords": [
     "agentic-coding",