leopold-driver 0.6.0 → 0.7.1

package/assets/VERSION

@@ -1 +1 @@
-0.6.0
+0.7.1

package/assets/extensions/gstack/extension.json

@@ -4,5 +4,6 @@
   "summary": "Garry Tan's planning + QA skill suite that Leopold conducts (/spec, /autoplan, /plan-*-review).",
   "homepage": "https://github.com/garrytan/gstack",
   "license": "MIT",
-  "order": 20
+  "order": 20,
+  "capabilities": ["filesystem.home", "network"]
 }

package/assets/extensions/leopold/extension.json

@@ -4,5 +4,6 @@
   "summary": "The autonomous orchestration harness itself (skills + Stop/PreToolUse hooks).",
   "homepage": "https://github.com/Jonhvmp/leopold",
   "license": "MIT",
-  "order": 10
+  "order": 10,
+  "capabilities": ["settings.write", "filesystem.home"]
 }

package/assets/extensions/ovmem/extension.json

@@ -5,6 +5,7 @@
   "homepage": "https://github.com/Jonhvmp/leopold",
   "license": "MIT",
   "order": 30,
+  "capabilities": ["network", "settings.write", "filesystem.home", "package.install", "process.spawn"],
   "dashboard": {
     "label": "Memory",
     "module": "~/.claude/ovmem/dashboard.py",

package/assets/extensions/serena/extension.json

@@ -4,5 +4,6 @@
   "summary": "LSP-backed code intelligence (MCP). Symbol-level retrieval + editing instead of grep/whole-file reads — sharper edits, far fewer tokens. Mandatory for quality.",
   "homepage": "https://github.com/oraios/serena",
   "license": "MIT",
-  "order": 15
+  "order": 15,
+  "capabilities": ["mcp.register", "settings.write", "package.install", "network"]
 }

package/assets/hooks/guard-irreversible.sh

@@ -112,6 +112,10 @@ case "$tool" in
     # normalize: newlines/tabs -> space, collapse runs (defeats whitespace/tab evasion).
     norm="$(printf '%s' "$cmd" | tr '\n\t' '  ' | tr -s ' ')"
 
+    # secret vault / master key are off-limits to the worker (secrets arrive as env vars)
+    matches "$norm" 'secrets\.(key|env)' && \
+      deny "Leopold guard: touching the secret vault/key via shell is forbidden. Secrets are pre-loaded as environment variables."
+
     # Opt-in deny-by-default (LEOPOLD_PARANOID=1): only a small allowlist of
     # read/build/test/lint commands passes; everything else is denied. Best-effort
     # (it keys off the first command word), kept off by default in favor of the
@@ -184,6 +188,7 @@ case "$tool" in
       */GUARDRAILS.md)             deny "Leopold guard: GUARDRAILS.md is immutable during an autonomous run." ;;
       */settings.json|*/settings.local.json) deny "Leopold guard: editing Claude Code settings is forbidden in autonomous mode." ;;
       */leopold/hooks/*|*/.leopold/state.json) deny "Leopold guard: the guardrail hooks and run state are immutable during an autonomous run." ;;
+      */secrets.key|*/.leopold/secrets.env) deny "Leopold guard: the secret vault and master key are off-limits. Secrets are pre-loaded as env vars." ;;
     esac
     ;;
 esac

package/assets/scripts/leopold-menu.sh

@@ -47,6 +47,25 @@ ext_installed() { bash "$1/manage.sh" detect >/dev/null 2>&1; }
 ext_status()    { bash "$1/manage.sh" status 2>/dev/null || true; }
 ext_run()       { bash "$1/manage.sh" "$2"; }
 
+ext_caps() { # extension.json -> space-separated capabilities (empty if none)
+  if command -v jq >/dev/null 2>&1; then
+    jq -r '(.capabilities // []) | join(" ")' "$1" 2>/dev/null
+  elif command -v python3 >/dev/null 2>&1; then
+    python3 -c "import json,sys;print(' '.join(json.load(open(sys.argv[1])).get('capabilities',[])))" "$1" 2>/dev/null
+  fi
+}
+
+# Show an extension's declared capabilities and require explicit consent before
+# install/update grants them. No declaration -> nothing to gate, proceed.
+ext_consent() { # dir -> 0 if the user consents
+  local caps; caps="$(ext_caps "$1/extension.json")"
+  [ -n "$caps" ] || return 0
+  printf "\n  %sThis extension requests:%s %s%s%s\n" "$C_BOLD" "$C_RESET" "$C_YELLOW" "$caps" "$C_RESET"
+  printf "  Install/update grants these. Proceed? [y/N] "
+  local a; read -r a || a=""
+  case "$a" in [yY]*) return 0 ;; *) echo "  cancelled."; return 1 ;; esac
+}
+
 pause() { printf "\n%spress Enter to continue%s " "$C_DIM" "$C_RESET"; read -r _ || true; }
 
 # ---- screens ----------------------------------------------------------------
@@ -90,6 +109,8 @@ component_menu() {
     ext_installed "$d" && st="installed${C_RESET} ${C_DIM}($(ext_status "$d"))"
     printf "  %s%s%s\n  status: %s%s\n\n" "$C_BOLD" "$title" "$C_RESET" "$C_GREEN" "$st"
     printf "  %s%s%s\n\n" "$C_DIM" "$(_jget "$d/extension.json" summary)" "$C_RESET"
+    local caps; caps="$(ext_caps "$d/extension.json")"
+    [ -n "$caps" ] && printf "  %scapabilities:%s %s\n\n" "$C_DIM" "$C_RESET" "$caps"
     local has_dash=""; [ -n "$(_jget "$d/extension.json" dashboard)" ] && has_dash=1
     if [ -n "$has_dash" ]; then
       printf "   1) Install    2) Update    3) Remove    4) Doctor    w) Watch    b) Back\n\n"
@@ -98,8 +119,8 @@ component_menu() {
     fi
     printf "select: "; read -r a || a="b"
     case "$a" in
-      1) ext_run "$d" install || echo "${C_YELLOW}install returned non-zero${C_RESET}"; pause ;;
-      2) ext_run "$d" update  || echo "${C_YELLOW}update returned non-zero${C_RESET}";  pause ;;
+      1) ext_consent "$d" && { ext_run "$d" install || echo "${C_YELLOW}install returned non-zero${C_RESET}"; }; pause ;;
+      2) ext_consent "$d" && { ext_run "$d" update  || echo "${C_YELLOW}update returned non-zero${C_RESET}"; }; pause ;;
       3) ext_run "$d" remove  || echo "${C_YELLOW}remove returned non-zero${C_RESET}";  pause ;;
       4) ext_run "$d" doctor  || true; pause ;;
       w|W) [ -n "$has_dash" ] && { ext_run "$d" watch || true; }; pause ;;

package/assets/scripts/leopold-watch.py

@@ -449,9 +449,9 @@ html,body{margin:0;background:var(--bg);color:var(--fg);font-family:var(--sans);
 .mrow{display:flex;gap:8px;flex-wrap:wrap;margin-top:14px}
 .mchip{font-family:var(--mono);font-size:10px;letter-spacing:.04em;border:1px solid var(--border);border-radius:9999px;padding:3px 10px;color:var(--muted-fg)}
 .meters{display:grid;grid-template-columns:repeat(auto-fit,minmax(150px,1fr));gap:14px}
-.meter .top{display:flex;justify-content:space-between;align-items:baseline}
-.meter .lbl{font-family:var(--mono);font-size:10px;letter-spacing:.08em;text-transform:uppercase;color:var(--muted-fg)}
-.meter .val{font-family:var(--mono);font-size:12px;font-variant-numeric:tabular-nums}
+.meter .top{display:flex;justify-content:space-between;align-items:baseline;gap:8px}
+.meter .lbl{font-family:var(--mono);font-size:10px;letter-spacing:.08em;text-transform:uppercase;color:var(--muted-fg);flex:1;min-width:0;overflow:hidden;text-overflow:ellipsis;white-space:nowrap}
+.meter .val{font-family:var(--mono);font-size:12px;font-variant-numeric:tabular-nums;flex-shrink:0}
 .bar{height:5px;background:var(--secondary);border-radius:9999px;margin-top:7px;overflow:hidden}
 .bar>i{display:block;height:100%;background:var(--success);transition:width .3s}
 .bar.warn>i{background:var(--warnbar)}.bar.full>i{background:var(--destructive)}
@@ -632,7 +632,8 @@ function widgetEl(name,w){
   if(w.kind==="bars"){
     const wrap=el("div","meters"),items=w.items||[],mx=Math.max(1,...items.map(x=>x.max||x.value||0));
     items.forEach(it=>{const m=el("div","meter"),top=el("div","top");
-      top.append(el("span","lbl",it.label),el("span","val tnum",""+it.value));
+      const lbl=el("span","lbl",it.label);lbl.title=it.label;
+      top.append(lbl,el("span","val tnum",""+it.value));
       const bar=el("div","bar"),i=el("i");i.style.width=Math.round(100*(it.value||0)/(it.max||mx))+"%";bar.append(i);
       m.append(top,bar);wrap.append(m);});
     return wrap;

package/dist/budget.js

@@ -0,0 +1,16 @@
+// USD budget hard-stop for the SDK driver. The Claude Code CLI already reports
+// `total_cost_usd` per session, so we never need a model price map: accumulate the
+// real cost per item and stop the run when it crosses the cap. Two checks, like
+// paperclip's evaluateCostEvent + getInvocationBlock: preventive (before an item)
+// and reactive (after each item's cost lands).
+/** Parse a budget value (CLI flag / env) into a positive USD number, or undefined. */
+export function parseBudgetUsd(raw) {
+    if (raw === undefined || raw === "")
+        return undefined;
+    const n = Number(raw);
+    return Number.isFinite(n) && n > 0 ? n : undefined;
+}
+/** True when spend has reached or passed the cap. A missing cap never trips. */
+export function overBudget(spentUsd, capUsd) {
+    return capUsd !== undefined && spentUsd >= capUsd;
+}

package/dist/config.js

@@ -1,6 +1,7 @@
 // Load the brief and run state from .leopold/, and the driver config from env.
 import fs from "node:fs";
 import path from "node:path";
+import { parseBudgetUsd } from "./budget.js";
 export function findLeoDir(cwd) {
     let dir = path.resolve(cwd);
     for (;;) {
@@ -75,6 +76,11 @@ export function clearRunTokens(leoDir) {
         catch { /* ignore */ }
     }
 }
+/** Read `--flag value` from argv (the value is the next token). */
+function flagValue(argv, name) {
+    const i = argv.indexOf(name);
+    return i >= 0 && i + 1 < argv.length ? argv[i + 1] : undefined;
+}
 export function loadConfig(argv) {
     return {
         conductorModel: process.env.LEOPOLD_CONDUCTOR_MODEL || undefined,
@@ -83,5 +89,6 @@ export function loadConfig(argv) {
         webhookUrl: process.env.LEOPOLD_WEBHOOK || undefined,
         dryRun: argv.includes("--dry-run"),
         worktree: argv.includes("--worktree") || process.env.LEOPOLD_WORKTREE === "1",
+        budgetUsd: parseBudgetUsd(flagValue(argv, "--budget-usd") ?? process.env.LEOPOLD_BUDGET_USD),
     };
 }

package/dist/guard.js

@@ -6,7 +6,10 @@ import fs from "node:fs";
 import path from "node:path";
 const GH_PR = /(^|[^\w-])gh(\s.*)?\s(pr\s+(create|merge)|release\s+create)/i;
 const PUBLISH = /(npm|pnpm|yarn)\s+publish|cargo\s+publish|twine\s+upload|pip\s+.*upload/i;
-const PROTECTED_PATH = /(GUARDRAILS\.md|settings\.json|settings\.local\.json|leopold\/hooks\/|\.leopold\/state\.json)/;
+const PROTECTED_PATH = /(GUARDRAILS\.md|settings\.json|settings\.local\.json|leopold\/hooks\/|\.leopold\/state\.json|secrets\.key|secrets\.env)/;
+// The secret vault (.leopold/secrets.env) and master key (~/.claude/leopold/secrets.key)
+// are off-limits to the worker — secrets reach it only as $NAME env vars.
+const SECRET_FILE = /secrets\.key|secrets\.env/;
 // git global options that consume the following token as their value.
 const GIT_VALUE_OPTS = new Set([
     "-c", "-C", "--git-dir", "--work-tree", "--namespace", "--exec-path", "--config-env",
@@ -61,8 +64,15 @@ export function makeGuard(leoDir, onBlock) {
             onBlock(toolName, message);
             return { behavior: "deny", message };
         };
+        if (toolName === "Read") {
+            const p = String(input.file_path ?? "");
+            if (SECRET_FILE.test(p))
+                return deny("Leopold guard: reading the secret vault or master key is forbidden. Secrets are pre-loaded as $NAME env vars.");
+        }
         if (toolName === "Bash") {
             const c = norm(String(input.command ?? ""));
+            if (SECRET_FILE.test(c))
+                return deny("Leopold guard: touching the secret vault/key via shell is forbidden. Secrets are pre-loaded as $NAME env vars.");
             if (isRecursiveForceRm(c))
                 return deny("Leopold guard: recursive+forced rm is forbidden in autonomous mode.");
             if (isFindDelete(c))

package/dist/index.js

@@ -4,6 +4,7 @@
 // into the package at build time; subcommands run them.
 import { runDriver } from "./loop.js";
 import { runInstall, runMenu, runWatch, runExt, runDoctor } from "./harness.js";
+import { runSecrets } from "./secrets.js";
 const sub = process.argv[2];
 const rest = process.argv.slice(3);
 function help() {
@@ -16,13 +17,17 @@ Usage:
   leopold-driver serena [install|doctor]    manage an extension (also: gstack, ovmem)
   leopold-driver doctor                     run every extension's doctor
   leopold-driver update                     reinstall from this package
-  leopold-driver run [--dry-run]            conduct the .leopold run (the SDK driver)
+  leopold-driver run [--worktree] [--budget-usd N] [--dry-run]
+                                            conduct the .leopold run (the SDK driver)
+  leopold-driver secrets set|list [NAME]    manage the run's encrypted secret vault
 
 Most commands run the bundled harness — no repo clone, no make. 'watch' needs Python 3.
 Newer version: npm i -g leopold-driver@latest.
 
 Conducting a run uses your existing Claude Code login (ANTHROPIC_API_KEY only in headless).
-Env: LEOPOLD_CONDUCTOR_MODEL, LEOPOLD_WORKER_MODEL, LEOPOLD_MAX_TURNS_PER_ITEM, LEOPOLD_WEBHOOK
+--worktree isolates the run in a git worktree; --budget-usd stops it at a USD cap.
+Env: LEOPOLD_CONDUCTOR_MODEL, LEOPOLD_WORKER_MODEL, LEOPOLD_MAX_TURNS_PER_ITEM, LEOPOLD_WEBHOOK,
+     LEOPOLD_WORKTREE, LEOPOLD_BUDGET_USD
 `);
 }
 function conduct() {
@@ -47,6 +52,8 @@ switch (sub) {
         process.exit(runExt(sub, rest));
     case "doctor":
         process.exit(runDoctor());
+    case "secrets":
+        process.exit(runSecrets(rest));
     case "--help":
     case "-h":
     case "help":

package/dist/loop.js

@@ -9,6 +9,7 @@ import { logEvent, logDecision, markItemDone, openItems, nextOpenItem } from "./
 import { notify } from "./notify.js";
 import { createWorktree, cleanupWorktree } from "./worktree.js";
 import { reapOrphan } from "./reaper.js";
+import { overBudget } from "./budget.js";
 export async function runDriver(cwd, argv) {
     const cfg = loadConfig(argv);
     const brief = loadBrief(cwd);
@@ -31,13 +32,18 @@ export async function runDriver(cwd, argv) {
         }
     }
     const state = initState(brief);
+    state.budget_usd = cfg.budgetUsd;
+    state.spent_usd = 0;
     if (worktree) {
         state.worktree_path = worktree.path;
         state.worktree_branch = worktree.branch;
-        writeState(brief.leoDir, state);
     }
+    writeState(brief.leoDir, state);
     const recent = [];
-    logEvent(brief.leoDir, { event: "run_start", conductor: cfg.conductorModel, worktree: worktree?.path ?? null });
+    logEvent(brief.leoDir, {
+        event: "run_start", conductor: cfg.conductorModel,
+        worktree: worktree?.path ?? null, budget_usd: cfg.budgetUsd ?? null,
+    });
     console.log(`Leopold is conducting "${brief.root}". Git is locked. touch .leopold/STOP to halt.\n`);
     const stop = (reason) => {
         state.active = false;
@@ -54,6 +60,11 @@ export async function runDriver(cwd, argv) {
             await notify(brief.leoDir, cfg.webhookUrl, "Leopold stopped", "Kill switch hit.");
             return;
         }
+        if (overBudget(state.spent_usd ?? 0, cfg.budgetUsd)) {
+            stop("budget_exceeded");
+            await notify(brief.leoDir, cfg.webhookUrl, "Leopold stopped", `Budget reached: $${(state.spent_usd ?? 0).toFixed(2)} of $${cfg.budgetUsd?.toFixed(2)}. Work so far is staged for your review.`);
+            return;
+        }
         if (state.iteration >= state.max_iterations) {
             stop("iteration_budget");
             await notify(brief.leoDir, cfg.webhookUrl, "Leopold stopped", "Iteration budget reached.");
@@ -84,6 +95,10 @@ export async function runDriver(cwd, argv) {
             item,
             workerPrompt,
             onBlock: (tool, reason) => logEvent(brief.leoDir, { event: "guard_block", tool, reason }),
+            onCost: (usd) => {
+                state.spent_usd = (state.spent_usd ?? 0) + usd;
+                logEvent(brief.leoDir, { event: "cost", item, usd, spent_usd: state.spent_usd });
+            },
             onTurn: async (status) => {
                 logEvent(brief.leoDir, { event: "worker_turn", kind: status.kind, item: status.item || item });
                 const verdict = await decide(cfg, brief, status, recent.slice(-5).join("\n"));

package/dist/secrets.js

@@ -0,0 +1,142 @@
+// Encrypted secret vault for a run. Secrets the work needs are injected into the
+// worker as environment variables (resolved at run time), so they reach the worker's
+// Bash tool as $NAME but never land in the prompt/transcript the model sees.
+//
+// At rest: AES-256-GCM. The 32-byte master key lives at ~/.claude/leopold/secrets.key
+// (mode 0600, generated on demand); the vault is .leopold/secrets.env (the encrypted
+// blob). Mirrors paperclip's local-encrypted-provider, file edition — no DB.
+import crypto from "node:crypto";
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import { findLeoDir } from "./config.js";
+const ALGO = "aes-256-gcm";
+const NAME_RE = /^[A-Za-z_][A-Za-z0-9_]*$/;
+export function keyPath() {
+    const base = process.env.CLAUDE_CONFIG_DIR || path.join(os.homedir(), ".claude");
+    return path.join(base, "leopold", "secrets.key");
+}
+export function vaultPath(leoDir) {
+    return path.join(leoDir, "secrets.env");
+}
+/** Paths the guard must protect from the worker (the vault and the master key). */
+export function secretFilePaths(leoDir) {
+    return [vaultPath(leoDir), keyPath()];
+}
+function loadOrCreateKey() {
+    const kp = keyPath();
+    if (fs.existsSync(kp))
+        return Buffer.from(fs.readFileSync(kp, "utf8").trim(), "base64");
+    const key = crypto.randomBytes(32);
+    fs.mkdirSync(path.dirname(kp), { recursive: true });
+    fs.writeFileSync(kp, key.toString("base64"), { mode: 0o600 });
+    try {
+        fs.chmodSync(kp, 0o600);
+    }
+    catch { /* best effort on platforms without chmod */ }
+    return key;
+}
+function encrypt(key, plaintext) {
+    const iv = crypto.randomBytes(12);
+    const cipher = crypto.createCipheriv(ALGO, key, iv);
+    const data = Buffer.concat([cipher.update(plaintext, "utf8"), cipher.final()]);
+    return JSON.stringify({
+        v: 1, iv: iv.toString("base64"),
+        tag: cipher.getAuthTag().toString("base64"), data: data.toString("base64"),
+    });
+}
+function decrypt(key, blob) {
+    const o = JSON.parse(blob);
+    const decipher = crypto.createDecipheriv(ALGO, key, Buffer.from(o.iv, "base64"));
+    decipher.setAuthTag(Buffer.from(o.tag, "base64"));
+    return Buffer.concat([decipher.update(Buffer.from(o.data, "base64")), decipher.final()]).toString("utf8");
+}
+function readVault(leoDir) {
+    const vp = vaultPath(leoDir);
+    if (!fs.existsSync(vp))
+        return {};
+    try {
+        return JSON.parse(decrypt(loadOrCreateKey(), fs.readFileSync(vp, "utf8")));
+    }
+    catch {
+        return {}; // wrong/rotated key or corrupt vault — fail closed (no secrets)
+    }
+}
+function writeVault(leoDir, secrets) {
+    fs.writeFileSync(vaultPath(leoDir), encrypt(loadOrCreateKey(), JSON.stringify(secrets)), { mode: 0o600 });
+}
+export function isValidName(name) {
+    return NAME_RE.test(name);
+}
+export function setSecret(leoDir, name, value) {
+    if (!isValidName(name))
+        throw new Error(`invalid secret name: ${name}`);
+    const s = readVault(leoDir);
+    s[name] = value;
+    writeVault(leoDir, s);
+}
+/** Decrypt the vault to {NAME: value} for env injection. */
+export function loadSecrets(leoDir) {
+    return readVault(leoDir);
+}
+/** Set the run's secrets into process.env for the duration of one item; returns a
+ *  restore() that puts the previous environment back (so secrets don't outlive the item). */
+export function applySecretsEnv(leoDir) {
+    const secrets = loadSecrets(leoDir);
+    const prev = [];
+    for (const [k, v] of Object.entries(secrets)) {
+        prev.push([k, process.env[k]]);
+        process.env[k] = v;
+    }
+    return {
+        restore() {
+            for (const [k, p] of prev) {
+                if (p === undefined)
+                    delete process.env[k];
+                else
+                    process.env[k] = p;
+            }
+        },
+    };
+}
+export function listSecretNames(leoDir) {
+    return Object.keys(readVault(leoDir)).sort();
+}
+/** `leopold-driver secrets set NAME | list` — the value for `set` is read from
+ *  stdin so it never appears in shell history. */
+export function runSecrets(argv) {
+    const sub = argv[0];
+    let leoDir;
+    try {
+        leoDir = findLeoDir(process.cwd());
+    }
+    catch {
+        console.error("leopold-driver secrets: no .leopold/ here. Run /leopold-brief first.");
+        return 1;
+    }
+    if (sub === "list") {
+        const names = listSecretNames(leoDir);
+        process.stdout.write(names.length ? names.join("\n") + "\n" : "(no secrets)\n");
+        return 0;
+    }
+    if (sub === "set") {
+        const name = argv[1];
+        if (!name || !isValidName(name)) {
+            console.error("usage: leopold-driver secrets set NAME   (NAME = a valid env var name)");
+            return 2;
+        }
+        let value;
+        try {
+            value = fs.readFileSync(0, "utf8").replace(/\r?\n$/, ""); // stdin, strip one trailing newline
+        }
+        catch {
+            console.error("could not read the secret value from stdin");
+            return 1;
+        }
+        setSecret(leoDir, name, value);
+        console.log(`secret '${name}' stored (encrypted) in .leopold/secrets.env`);
+        return 0;
+    }
+    console.error("usage: leopold-driver secrets set NAME | list");
+    return 2;
+}

package/dist/worker.js

@@ -8,10 +8,12 @@ import { query } from "@anthropic-ai/claude-agent-sdk";
 import { InputChannel } from "./channel.js";
 import { parseStatus, isTurnComplete } from "./protocol.js";
 import { makeGuard } from "./guard.js";
+import { applySecretsEnv } from "./secrets.js";
 const WORKER_APPEND = `You are a Leopold worker, conducted by an autonomous orchestrator. No human is watching live. Rules for this session:
 - Do NOT ask the human anything. Decide reversible or charter-clear calls yourself and keep going.
 - Spawned mode: if you invoke gstack skills, auto-pick the recommended option; never prompt.
 - git commit/push/publish are LOCKED by a guard. Never attempt them. Stage with "git add" and report instead.
+- Secrets you may need are pre-loaded as environment variables; use them as $NAME, and never ask for, echo, or print their values.
 - Close EVERY turn with a fenced status block, then stop and wait for the conductor's reply:
 
 \`\`\`leopold-status
@@ -29,10 +31,15 @@ export async function runItem(opts) {
     const channel = new InputChannel();
     channel.push(workerPrompt);
     const guard = makeGuard(brief.leoDir, onBlock);
+    // Inject the run's secrets as env vars for this item: they reach the worker's Bash
+    // tool as $NAME but never enter the prompt. Restored after the loop (runs are
+    // sequential, so there is no env overlap between items).
+    const { restore: restoreSecrets } = applySecretsEnv(brief.leoDir);
     const q = query({
         prompt: channel,
         options: {
             cwd: brief.worktreeRoot ?? brief.root,
+            env: { ...process.env },
             maxTurns: cfg.maxTurnsPerItem,
             permissionMode: "default",
             canUseTool: guard,
@@ -61,6 +68,10 @@ export async function runItem(opts) {
             }
         }
         else if (msg.type === "result") {
+            // The CLI reports the item's real cost here — accumulate it for the budget.
+            const cost = msg.total_cost_usd;
+            if (typeof cost === "number" && Number.isFinite(cost))
+                opts.onCost?.(cost);
             // Session ended (channel closed, or the worker stopped on its own). Flush
             // whatever we have so the conductor can make a final call.
             if (turnText.trim()) {
@@ -72,4 +83,5 @@ export async function runItem(opts) {
             break;
         }
     }
+    restoreSecrets();
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "leopold-driver",
-  "version": "0.6.0",
+  "version": "0.7.1",
   "description": "Leopold SDK driver: a persistent conductor that orchestrates fresh Claude Code workers per task, decides from your charter, and notifies you. Uses your Claude Code auth. Git stays locked.",
   "type": "module",
   "bin": {
@@ -37,7 +37,7 @@
   "scripts": {
     "build": "tsc -p tsconfig.json && node scripts/copy-runtime.mjs",
     "typecheck": "tsc -p tsconfig.json --noEmit",
-    "test": "node --import tsx --test test/protocol.test.ts test/guard.test.ts test/worktree.test.ts test/reaper.test.ts",
+    "test": "node --import tsx --test test/protocol.test.ts test/guard.test.ts test/worktree.test.ts test/reaper.test.ts test/budget.test.ts test/secrets.test.ts",
     "dev": "tsx src/index.ts",
     "start": "node dist/index.js",
     "prepublishOnly": "npm run build"