reasonix 0.12.19 → 0.12.20

package/dist/cli/index.js

@@ -548,8 +548,8 @@ var defaultSelector = (samples) => {
   })[0];
 };
 async function runBranches(client, request, opts = {}) {
-  const budget = Math.max(1, opts.budget ?? 1);
-  const temperatures = resolveTemperatures(budget, opts.temperatures);
+  const budget2 = Math.max(1, opts.budget ?? 1);
+  const temperatures = resolveTemperatures(budget2, opts.temperatures);
   const selector = opts.selector ?? defaultSelector;
   const samples = await Promise.all(
     temperatures.map(async (temperature, index) => {
@@ -586,12 +586,12 @@ function aggregateBranchUsage(samples) {
     promptCacheMissTokens
   };
 }
-function resolveTemperatures(budget, custom) {
-  if (custom && custom.length >= budget) return [...custom.slice(0, budget)];
-  if (budget === 1) return [0];
+function resolveTemperatures(budget2, custom) {
+  if (custom && custom.length >= budget2) return [...custom.slice(0, budget2)];
+  if (budget2 === 1) return [0];
   const out = [];
-  for (let i = 0; i < budget; i++) {
-    out.push(Number((i / (budget - 1)).toFixed(2)));
+  for (let i = 0; i < budget2; i++) {
+    out.push(Number((i / (budget2 - 1)).toFixed(2)));
   }
   return out;
 }
@@ -1276,29 +1276,29 @@ function truncateForModelByTokens(s, maxTokens) {
 
 ${tail}`;
 }
-function sizePrefixToTokens(s, budget) {
-  if (budget <= 0 || s.length === 0) return "";
-  let size = Math.min(s.length, budget * 4);
+function sizePrefixToTokens(s, budget2) {
+  if (budget2 <= 0 || s.length === 0) return "";
+  let size = Math.min(s.length, budget2 * 4);
   for (let iter = 0; iter < 6; iter++) {
     if (size <= 0) return "";
     const slice = s.slice(0, size);
     const count = countTokens(slice);
-    if (count <= budget) return slice;
-    const next = Math.floor(size * (budget / count) * 0.95);
+    if (count <= budget2) return slice;
+    const next = Math.floor(size * (budget2 / count) * 0.95);
     if (next >= size) return s.slice(0, Math.max(0, size - 1));
     size = next;
   }
   return s.slice(0, Math.max(0, size));
 }
-function sizeSuffixToTokens(s, budget) {
-  if (budget <= 0 || s.length === 0) return "";
-  let size = Math.min(s.length, budget * 4);
+function sizeSuffixToTokens(s, budget2) {
+  if (budget2 <= 0 || s.length === 0) return "";
+  let size = Math.min(s.length, budget2 * 4);
   for (let iter = 0; iter < 6; iter++) {
     if (size <= 0) return "";
     const slice = s.slice(-size);
     const count = countTokens(slice);
-    if (count <= budget) return slice;
-    const next = Math.floor(size * (budget / count) * 0.95);
+    if (count <= budget2) return slice;
+    const next = Math.floor(size * (budget2 / count) * 0.95);
     if (next >= size) return s.slice(-Math.max(0, size - 1));
     size = next;
   }
@@ -1999,6 +1999,19 @@ var CacheFirstLoop = class {
    * flip it live alongside `model`.
    */
   autoEscalate = true;
+  /**
+   * Soft USD budget — see {@link CacheFirstLoopOptions.budgetUsd}.
+   * Mutable so `/budget` slash can set / change / clear it mid-session.
+   * `null` (the default) disables all budget checks.
+   */
+  budgetUsd;
+  /**
+   * Set the first time a turn crosses 80% of the budget so the warning
+   * doesn't repeat every turn afterwards. Cleared by `setBudget` (any
+   * change re-arms the warning, including raising the cap above the
+   * current spend).
+   */
+  _budgetWarned = false;
   sessionName;
   /**
    * Hook list, mutable so `/hooks reload` can swap it without
@@ -2064,6 +2077,7 @@ var CacheFirstLoop = class {
     this.model = opts.model ?? "deepseek-v4-flash";
     this.reasoningEffort = opts.reasoningEffort ?? "max";
     if (opts.autoEscalate !== void 0) this.autoEscalate = opts.autoEscalate;
+    this.budgetUsd = typeof opts.budgetUsd === "number" && opts.budgetUsd > 0 ? opts.budgetUsd : null;
     this.maxToolIters = opts.maxToolIters ?? 64;
     this.hooks = opts.hooks ?? [];
     this.hookCwd = opts.hookCwd ?? process.cwd();
@@ -2303,6 +2317,16 @@ var CacheFirstLoop = class {
     }
     this.stream = this.branchEnabled ? false : this._streamPreference;
   }
+  /**
+   * Set / change / clear the soft USD budget. `null` (or any non-
+   * positive number) disables the cap entirely. Re-arms the 80%
+   * warning so a user who bumps the cap mid-session sees a fresh
+   * threshold message at the new boundary.
+   */
+  setBudget(usd) {
+    this.budgetUsd = typeof usd === "number" && usd > 0 ? usd : null;
+    this._budgetWarned = false;
+  }
   /**
    * Arm pro for the next turn (consumed at turn start). Called by
    * `/pro`. Idempotent — repeated calls stay armed, `disarmPro()`
@@ -2464,6 +2488,26 @@ var CacheFirstLoop = class {
     return userText;
   }
   async *step(userInput) {
+    if (this.budgetUsd !== null) {
+      const spent = this.stats.totalCost;
+      if (spent >= this.budgetUsd) {
+        yield {
+          turn: this._turn,
+          role: "error",
+          content: "",
+          error: `session budget exhausted \u2014 spent $${spent.toFixed(4)} \u2265 cap $${this.budgetUsd.toFixed(2)}. Bump the cap with /budget <usd>, clear it with /budget off, or end the session.`
+        };
+        return;
+      }
+      if (!this._budgetWarned && spent >= this.budgetUsd * 0.8) {
+        this._budgetWarned = true;
+        yield {
+          turn: this._turn,
+          role: "warning",
+          content: `\u25B2 budget 80% used \u2014 $${spent.toFixed(4)} of $${this.budgetUsd.toFixed(2)}. Next turn or two likely trips the cap.`
+        };
+      }
+    }
     this._turn++;
     this.scratch.reset();
     this.repair.resetStorm();
@@ -2560,14 +2604,14 @@ var CacheFirstLoop = class {
       let preHarvestedPlanState;
       try {
         if (this.branchEnabled) {
-          const budget = this.branchOptions.budget ?? 1;
+          const budget2 = this.branchOptions.budget ?? 1;
           yield {
             turn: this._turn,
             role: "branch_start",
             content: "",
             branchProgress: {
               completed: 0,
-              total: budget,
+              total: budget2,
               latestIndex: -1,
               latestTemperature: -1,
               latestUncertainties: -1
@@ -2601,7 +2645,7 @@ var CacheFirstLoop = class {
               onSampleDone
             }
           );
-          for (let k = 0; k < budget; k++) {
+          for (let k = 0; k < budget2; k++) {
             const sample = queue.shift() ?? await new Promise((resolve13) => {
               waiter = resolve13;
             });
@@ -2611,7 +2655,7 @@ var CacheFirstLoop = class {
               content: "",
               branchProgress: {
                 completed: k + 1,
-                total: budget,
+                total: budget2,
                 latestIndex: sample.index,
                 latestTemperature: sample.temperature,
                 latestUncertainties: sample.planState.uncertainties.length
@@ -11401,13 +11445,13 @@ var MIN_DIFF_ROWS = 8;
 function EditConfirm({ block, onChoose }) {
   const { stdout: stdout3 } = useStdout2();
   const rows = stdout3?.rows ?? 40;
-  const budget = Math.max(MIN_DIFF_ROWS, rows - MODAL_OVERHEAD_ROWS);
+  const budget2 = Math.max(MIN_DIFF_ROWS, rows - MODAL_OVERHEAD_ROWS);
   const allLines = useMemo(
     () => formatEditBlockDiff(block, { contextLines: 2, maxLines: 1e5, indent: "  " }),
     [block]
   );
   const [scroll, setScroll] = useState2(0);
-  const maxScroll = Math.max(0, allLines.length - budget);
+  const maxScroll = Math.max(0, allLines.length - budget2);
   const effectiveScroll = Math.min(scroll, maxScroll);
   useKeystroke((ev) => {
     if (ev.paste) return;
@@ -11438,11 +11482,11 @@ function EditConfirm({ block, onChoose }) {
       return;
     }
     if (key.pageDown || input === " " || input === "f") {
-      setScroll((s) => Math.min(maxScroll, s + Math.max(1, budget - 2)));
+      setScroll((s) => Math.min(maxScroll, s + Math.max(1, budget2 - 2)));
       return;
     }
     if (key.pageUp || input === "b") {
-      setScroll((s) => Math.max(0, s - Math.max(1, budget - 2)));
+      setScroll((s) => Math.max(0, s - Math.max(1, budget2 - 2)));
       return;
     }
     if (input === "g") {
@@ -11458,9 +11502,9 @@ function EditConfirm({ block, onChoose }) {
   const removed = isNew ? 0 : (block.search.match(/\n/g)?.length ?? 0) + 1;
   const added = block.replace === "" ? 0 : (block.replace.match(/\n/g)?.length ?? 0) + 1;
   const tag = isNew ? "NEW" : "EDIT";
-  const visibleLines = allLines.slice(effectiveScroll, effectiveScroll + budget);
+  const visibleLines = allLines.slice(effectiveScroll, effectiveScroll + budget2);
   const hiddenAbove = effectiveScroll;
-  const hiddenBelow = Math.max(0, allLines.length - effectiveScroll - budget);
+  const hiddenBelow = Math.max(0, allLines.length - effectiveScroll - budget2);
   const totalLines = allLines.length;
   const showScrollHud = hiddenAbove + hiddenBelow > 0;
   const subtitleParts = [`-${removed} +${added} lines`];
@@ -13547,13 +13591,13 @@ function buildViewport(line, cursorCol, visibleCells, pastes) {
   return clipAroundCursor(line, cursorCol, visibleCells, pastes);
 }
 function clipFromLeft(line, visibleCells, pastes) {
-  const budget = Math.max(1, visibleCells - 1);
+  const budget2 = Math.max(1, visibleCells - 1);
   let used = 0;
   let end = 0;
   while (end < line.length) {
     const ch = line[end];
     const cw = charCellsAt(line, end, pastes);
-    if (used + cw > budget) break;
+    if (used + cw > budget2) break;
     used += cw;
     end++;
   }
@@ -13561,10 +13605,10 @@ function clipFromLeft(line, visibleCells, pastes) {
   return { segments, cursorCell: null, hiddenLeft: false, hiddenRight: end < line.length };
 }
 function clipAroundCursor(line, cursorCol, visibleCells, pastes) {
-  let budget = visibleCells;
+  let budget2 = visibleCells;
   const reservedForMarkers = 2;
-  budget = Math.max(1, budget - reservedForMarkers);
-  const halfBudget = Math.floor(budget / 2);
+  budget2 = Math.max(1, budget2 - reservedForMarkers);
+  const halfBudget = Math.floor(budget2 / 2);
   let start = cursorCol;
   let leftCells = 0;
   while (start > 0 && leftCells < halfBudget) {
@@ -13573,7 +13617,7 @@ function clipAroundCursor(line, cursorCol, visibleCells, pastes) {
     start--;
     leftCells += cw;
   }
-  const rightBudget = budget - leftCells;
+  const rightBudget = budget2 - leftCells;
   let end = cursorCol;
   let rightCells = 0;
   const cursorChar = cursorCol < line.length ? charCellsAt(line, cursorCol, pastes) : 1;
@@ -14080,7 +14124,8 @@ function StatsPanel({
   busy,
   proArmed,
   escalated,
-  dashboardUrl
+  dashboardUrl,
+  budgetUsd
 }) {
   const branchOn = (branchBudget ?? 1) > 1;
   const ctxMax = DEEPSEEK_CONTEXT_TOKENS[model2] ?? DEFAULT_CONTEXT_TOKENS;
@@ -14126,7 +14171,12 @@ function StatsPanel({
       balance,
       coldStart
     }
-  ));
+  ), budgetUsd !== null && budgetUsd !== void 0 ? /* @__PURE__ */ React21.createElement(BudgetRow, { spent: summary.totalCostUsd, cap: budgetUsd }) : null);
+}
+function BudgetRow({ spent, cap }) {
+  const pct2 = Math.max(0, spent / cap * 100);
+  const color = pct2 >= 100 ? "#f87171" : pct2 >= 80 ? "#fbbf24" : "#94a3b8";
+  return /* @__PURE__ */ React21.createElement(Box19, null, /* @__PURE__ */ React21.createElement(Text17, { dimColor: true }, "  budget  "), /* @__PURE__ */ React21.createElement(Text17, { color }, `$${spent.toFixed(4)} / $${cap.toFixed(2)}`, /* @__PURE__ */ React21.createElement(Text17, { dimColor: true }, `  (${pct2.toFixed(0)}%)`)));
 }
 function Header({
   model: model2,
@@ -14828,6 +14878,12 @@ var SLASH_COMMANDS = [
     summary: "arm v4-pro for the NEXT turn only (one-shot \xB7 auto-disarms after turn)",
     argCompleter: ["off"]
   },
+  {
+    cmd: "budget",
+    argsHint: "[usd|off]",
+    summary: "session USD cap \u2014 warns at 80%, refuses next turn at 100%. Off by default. /budget alone shows status",
+    argCompleter: ["off", "1", "5", "10", "20", "50"]
+  },
   { cmd: "mcp", summary: "list MCP servers + tools attached to this session" },
   {
     cmd: "resource",
@@ -16317,6 +16373,42 @@ var pro = (args, loop2, ctx) => {
   };
 };
 var ESCALATION_MODEL_ID = "deepseek-v4-pro";
+var budget = (args, loop2) => {
+  const arg = args[0]?.trim() ?? "";
+  if (arg === "") {
+    if (loop2.budgetUsd === null) {
+      return {
+        info: "no session budget set \u2014 Reasonix will keep going until you stop it. Set one with: /budget <usd>   (e.g. /budget 5)"
+      };
+    }
+    const spent2 = loop2.stats.totalCost;
+    const pct2 = spent2 / loop2.budgetUsd * 100;
+    return {
+      info: `budget: $${spent2.toFixed(4)} of $${loop2.budgetUsd.toFixed(2)} (${pct2.toFixed(1)}%) \xB7 /budget off to clear, /budget <usd> to change`
+    };
+  }
+  if (arg === "off" || arg === "none" || arg === "0") {
+    loop2.setBudget(null);
+    return { info: "budget \u2192 off (no cap)" };
+  }
+  const cleaned = arg.replace(/^\$/, "");
+  const usd = Number(cleaned);
+  if (!Number.isFinite(usd) || usd <= 0) {
+    return {
+      info: `usage: /budget <usd>   (got "${arg}" \u2014 must be a positive number, e.g. /budget 5 or /budget 12.50)`
+    };
+  }
+  loop2.setBudget(usd);
+  const spent = loop2.stats.totalCost;
+  if (spent >= usd) {
+    return {
+      info: `\u25B2 budget \u2192 $${usd.toFixed(2)} but already spent $${spent.toFixed(4)}. Next turn will be refused \u2014 bump the cap higher to keep going, or end the session.`
+    };
+  }
+  return {
+    info: `budget \u2192 $${usd.toFixed(2)}  (so far: $${spent.toFixed(4)} \xB7 warns at 80%, refuses next turn at 100% \xB7 /budget off to clear)`
+  };
+};
 var handlers9 = {
   model,
   models,
@@ -16324,7 +16416,8 @@ var handlers9 = {
   preset,
   branch,
   effort,
-  pro
+  pro,
+  budget
 };
 
 // src/cli/ui/slash/handlers/observability.ts
@@ -17577,6 +17670,7 @@ function App({
   transcript,
   harvest: harvest3,
   branch: branch2,
+  budgetUsd,
   session,
   tools,
   mcpSpecs,
@@ -17790,6 +17884,7 @@ function App({
       model: model2,
       harvest: harvest3,
       branch: branch2,
+      budgetUsd,
       session,
       hooks: hookList,
       hookCwd: currentRootDir,
@@ -17800,7 +17895,7 @@ function App({
     });
     loopRef.current = l;
     return l;
-  }, [model2, system, harvest3, branch2, session, tools, codeMode]);
+  }, [model2, system, harvest3, branch2, budgetUsd, session, tools, codeMode]);
   useEffect6(() => {
     loop2.hooks = hookList;
   }, [loop2, hookList]);
@@ -20045,7 +20140,8 @@ Continue executing from the next pending step. Call mark_step_complete after eac
         updateAvailable,
         proArmed,
         escalated: turnOnPro,
-        dashboardUrl
+        dashboardUrl,
+        budgetUsd: loop2.budgetUsd
       }
     ), /* @__PURE__ */ React24.createElement(Static, { items: historical }, (item) => /* @__PURE__ */ React24.createElement(EventRow, { key: item.id, event: item, projectRoot: currentRootDir })), !historical.some((e) => e.role === "user" || e.role === "assistant") && !busy && !streaming ? /* @__PURE__ */ React24.createElement(WelcomeBanner, { inCodeMode: !!codeMode }) : null, !PLAIN_UI && !pendingShell && !pendingWorkspace && !pendingPlan && !stagedInput && !pendingEditReview && !pendingCheckpoint && !stagedCheckpointRevise && streaming ? /* @__PURE__ */ React24.createElement(Box22, { marginY: 1 }, /* @__PURE__ */ React24.createElement(EventRow, { event: streaming, projectRoot: currentRootDir })) : null, !PLAIN_UI && !pendingShell && !pendingWorkspace && !pendingPlan && !stagedInput && !pendingEditReview && !pendingCheckpoint && !stagedCheckpointRevise && ongoingTool ? /* @__PURE__ */ React24.createElement(OngoingToolRow, { tool: ongoingTool, progress: toolProgress }) : null, !PLAIN_UI && !pendingShell && !pendingWorkspace && !pendingPlan && !stagedInput && !pendingEditReview && !pendingCheckpoint && !stagedCheckpointRevise && subagentActivity ? /* @__PURE__ */ React24.createElement(SubagentRow, { activity: subagentActivity }) : null, !PLAIN_UI && !pendingShell && !pendingWorkspace && !pendingPlan && !stagedInput && !pendingEditReview && !pendingCheckpoint && !stagedCheckpointRevise && !ongoingTool && statusLine ? /* @__PURE__ */ React24.createElement(StatusRow, { text: statusLine }) : null, !PLAIN_UI && undoBanner && !pendingShell && !pendingWorkspace && !pendingPlan && !stagedInput && !pendingEditReview && !pendingCheckpoint && !stagedCheckpointRevise && !pendingChoice && !stagedChoiceCustom && !pendingRevision ? /* @__PURE__ */ React24.createElement(UndoBanner, { banner: undoBanner }) : null, !PLAIN_UI && !pendingShell && !pendingWorkspace && !pendingPlan && !stagedInput && !pendingEditReview && !pendingCheckpoint && !stagedCheckpointRevise && busy && !streaming && !ongoingTool && !statusLine ? /* @__PURE__ */ React24.createElement(StatusRow, { text: "processing\u2026" }) : null, stagedInput ? /* @__PURE__ */ React24.createElement(
       PlanRefineInput,
@@ -20316,6 +20412,7 @@ function Root({
       transcript: appProps.transcript,
       harvest: appProps.harvest,
       branch: appProps.branch,
+      budgetUsd: appProps.budgetUsd,
       session: appProps.session,
       tools,
       mcpSpecs,
@@ -20477,6 +20574,7 @@ async function codeCommand(opts = {}) {
   await chatCommand({
     model: opts.model ?? "deepseek-v4-flash",
     harvest: opts.harvest ?? false,
+    budgetUsd: opts.budgetUsd,
     system: codeSystemPrompt2(rootDir, { hasSemanticSearch: semantic2.enabled }),
     transcript: opts.transcript,
     session,
@@ -21213,7 +21311,8 @@ async function runCommand2(opts) {
     tools,
     model: opts.model,
     harvest: opts.harvest,
-    branch: opts.branch
+    branch: opts.branch,
+    budgetUsd: opts.budgetUsd
   });
   const prefixHash = prefix.fingerprint;
   let transcriptStream = null;
@@ -21769,6 +21868,17 @@ Your training data has a cutoff. When an answer's correctness depends on somethi
 The signal isn't a topic list \u2014 it's: "if I'm wrong about this, is it because reality moved on?". If yes, ground the answer in fresh evidence; if no (definitions, mechanisms, well-established APIs), answer from memory.
 
 ${ESCALATION_CONTRACT}`;
+function parseBudgetFlag(raw) {
+  if (raw === void 0) return void 0;
+  if (!Number.isFinite(raw) || raw <= 0) {
+    process.stderr.write(
+      `\u25B2 ignoring --budget=${raw} (must be a positive number) \u2014 running with no cap
+`
+    );
+    return void 0;
+  }
+  return raw;
+}
 var program = new Command();
 program.name("reasonix").description("DeepSeek-native agent framework \u2014 built for cache hits and cheap tokens.").version(VERSION).option(
   "-c, --continue",
@@ -21806,6 +21916,10 @@ program.command("code [dir]").description(
 ).option("-m, --model <id>", "Override default model (v4-flash)").option("--no-session", "Disable session persistence for this run").option("-r, --resume", "Skip the session picker \u2014 always continue prior messages").option("-n, --new", "Skip the session picker \u2014 always wipe prior messages and start fresh").option("--transcript <path>", "Write a JSONL transcript to this path").option(
   "--harvest",
   "Opt-in Pillar-2 plan-state extraction. Adds one flash call per turn; off by default (no preset enables it)."
+).option(
+  "--budget <usd>",
+  "Soft USD cap on session spend. Off by default. Warns at 80%, refuses next turn at 100%. Mid-session: /budget <usd> or /budget off.",
+  (v) => Number.parseFloat(v)
 ).option(
   "--no-dashboard",
   "Suppress the auto-launched embedded web dashboard. Default behavior boots it on TUI mount and shows the URL in the status bar (clickable in OSC-8-aware terminals)."
@@ -21818,6 +21932,7 @@ program.command("code [dir]").description(
     forceResume: !!opts.resume,
     forceNew: !!opts.new,
     harvest: !!opts.harvest,
+    budgetUsd: parseBudgetFlag(opts.budget),
     noDashboard: opts.dashboard === false
   });
 });
@@ -21831,6 +21946,10 @@ program.command("chat").description("Interactive Ink TUI with live cache/cost pa
   "--branch <n>",
   "Self-consistency: run N parallel samples per turn (N\xD7 cost). Manual only \u2014 never auto-enabled.",
   (v) => Number.parseInt(v, 10)
+).option(
+  "--budget <usd>",
+  "Soft USD cap on session spend. Off by default. Warns at 80%, refuses next turn at 100%. Mid-session: /budget <usd> or /budget off.",
+  (v) => Number.parseFloat(v)
 ).option("--session <name>", "Use a named session (default: from config, usually 'default').").option("--no-session", "Disable session persistence for this run (ephemeral chat)").option("-r, --resume", "Skip the session picker \u2014 always continue prior messages").option(
   "-c, --continue",
   "Resume the most-recently-used session (any name) without showing the picker."
@@ -21868,6 +21987,7 @@ program.command("chat").description("Interactive Ink TUI with live cache/cost pa
     transcript: opts.transcript,
     harvest: defaults.harvest,
     branch: defaults.branch,
+    budgetUsd: parseBudgetFlag(opts.budget),
     session: continueOpts.session,
     mcp: defaults.mcp,
     mcpPrefix: opts.mcpPrefix,
@@ -21880,6 +22000,10 @@ program.command("run <task>").description("Run a single task non-interactively,
   "--branch <n>",
   "Self-consistency: run N parallel samples per turn and pick the most confident",
   (v) => Number.parseInt(v, 10)
+).option(
+  "--budget <usd>",
+  "Soft USD cap on session spend. Off by default. Refuses to start a new turn once cumulative cost \u2265 cap.",
+  (v) => Number.parseFloat(v)
 ).option("--transcript <path>", "Write a JSONL transcript to this path for replay/diff").option(
   "--mcp <spec>",
   'MCP server spec; repeatable. "name=cmd args...", "cmd args...", or a URL (http/https \u2192 SSE).',
@@ -21903,6 +22027,7 @@ program.command("run <task>").description("Run a single task non-interactively,
     system: applyMemoryStack(opts.system, process.cwd()),
     harvest: defaults.harvest,
     branch: defaults.branch,
+    budgetUsd: parseBudgetFlag(opts.budget),
     transcript: opts.transcript,
     mcp: defaults.mcp,
     mcpPrefix: opts.mcpPrefix