@genesislcap/ai-assistant 15.7.3-alpha-e7d4aa5.0 → 15.8.0

package/src/main/main.types.ts

@@ -93,6 +93,32 @@ export const AiAssistantEvent = {
  */
 export type AiAssistantEvent = (typeof AiAssistantEvent)[keyof typeof AiAssistantEvent];
 
+/**
+ * One rendered row of the settings-modal budget meter (GENC-1464) — a vendor
+ * that currently has budget figures, pre-formatted so the template stays a pure
+ * binding layer (and so tests can pin the exact text a user reads).
+ *
+ * @beta
+ */
+export interface SettingsBudgetRow {
+  /** The vendor the row describes (an `AIProviderType` member). */
+  vendor: string;
+  /** Display name from `VENDOR_LABELS`, e.g. `Anthropic`. */
+  label: string;
+  /**
+   * The real figures — `$45.00 / $50.00 (90%)`. The percentage is NOT clamped:
+   * spend can genuinely overshoot the cap (the last allowed call may cross it),
+   * and the text tells the truth about it.
+   */
+  figures: string;
+  /**
+   * The bar's value, clamped to `0..100` — a progress bar past its end renders
+   * nonsense, so only the bar clamps (mirroring the context-usage indicator's
+   * `Math.min(100, …)`).
+   */
+  barValue: number;
+}
+
 /**
  * State of the chat suggestions feature.
  *

package/src/state/ai-assistant-slice.ts

@@ -27,6 +27,23 @@ export interface InputOverride {
   mode: ChatInputDuringExecutionMode;
 }
 
+/**
+ * One vendor's AI-spend figures, as fed by the host from the budget endpoint
+ * (GENC-1464 — the settings-modal budget meter).
+ *
+ * Dollars, not a percentage: the meter's text shows the real figures (spend CAN
+ * exceed the cap — the last allowed call may overshoot it), and only the bar
+ * clamps. Storing a pre-computed percentage would destroy the overshoot.
+ *
+ * @beta
+ */
+export interface VendorBudgetFigures {
+  /** The vendor's budget cap in USD. Finite and `>= 0`. */
+  budgetUsd: number;
+  /** Metered spend against that cap in USD. Finite and `>= 0`; may exceed the cap. */
+  spentUsd: number;
+}
+
 /**
  * Shape of a single AI assistant session's serializable state.
  *
@@ -182,6 +199,20 @@ export interface AiAssistantSessionState {
    * copy composed from the vendor's display name.
    */
   blockedVendorReasons: Partial<Record<AIProviderType, string>>;
+  /**
+   * Per-vendor AI-spend figures for the settings-modal budget meter (GENC-1464),
+   * fed by the host via the element's `setVendorBudget`. A vendor with no entry
+   * renders no meter row — which is also how "unlimited" is expressed: the host
+   * simply never feeds a vendor the server marks unlimited.
+   *
+   * NOT persisted to the session snapshot, deliberately. The server is
+   * authoritative for spend and the host re-feeds on boot (and after each turn),
+   * so a persisted copy could only ever show stale dollars across a reload —
+   * wrong money is worse than a beat of no meter. The snapshot assembler
+   * (`session-snapshot.ts`) builds from an explicit field list, so absence is
+   * structural, not an accident of serialization.
+   */
+  vendorBudgets: Partial<Record<AIProviderType, VendorBudgetFigures>>;
   /** Draft text in the input box — preserved across pop-in/pop-out cycles. */
   inputValue: string;
   /** Live trace from a currently-executing sub-agent. Cleared on sub-agent-stop. */
@@ -249,6 +280,7 @@ export function createDefaultSessionState(): AiAssistantSessionState {
     blockedVendors: [],
     sweptVendors: [],
     blockedVendorReasons: {},
+    vendorBudgets: {},
     inputValue: '',
     liveSubAgentTrace: [],
     liveSubAgentName: null,
@@ -427,6 +459,41 @@ export const aiAssistantSlice = createSlice({
         delete state.blockedVendorReasons[vendor];
       }
     },
+    /**
+     * Store (or clear, with `null`) one vendor's budget figures for the meter.
+     *
+     * Validation lives at the element boundary (`setVendorBudget` on the
+     * element), not here — the reducer is the storage layer and trusts its one
+     * caller, exactly as `setVendorBlocked` above trusts its.
+     *
+     * Idempotent by VALUE, not just by effect: re-feeding the figures a vendor
+     * already holds leaves the stored object untouched. The documented feed
+     * re-runs after every turn, so without this every turn would publish a new
+     * store reference for the same dollars and re-render a meter that has not
+     * changed.
+     */
+    setVendorBudget(
+      state,
+      action: PayloadAction<{ vendor: AIProviderType; figures: VendorBudgetFigures | null }>,
+    ) {
+      const { vendor, figures } = action.payload;
+      if (figures === null) {
+        delete state.vendorBudgets[vendor];
+        return;
+      }
+      const existing = state.vendorBudgets[vendor];
+      if (
+        existing &&
+        existing.budgetUsd === figures.budgetUsd &&
+        existing.spentUsd === figures.spentUsd
+      ) {
+        return;
+      }
+      // A fresh object rather than the caller's: the store may freeze what it
+      // holds, and adopting the host's object would freeze state the host still
+      // owns (and let later host mutations bypass the store).
+      state.vendorBudgets[vendor] = { budgetUsd: figures.budgetUsd, spentUsd: figures.spentUsd };
+    },
     setInputValue(state, action: PayloadAction<string>) {
       state.inputValue = action.payload;
     },
@@ -522,6 +589,13 @@ export const aiAssistantSlice = createSlice({
         blockedVendors: state.blockedVendors,
         sweptVendors: state.sweptVendors,
         blockedVendorReasons: state.blockedVendorReasons,
+        // Category 2, beside the blocked latch: the budget figures are a fact
+        // about the USER'S DEPLOYMENT (what the server meters), not about this
+        // conversation — "Clear" refills no budget, so wiping them would blank
+        // the meter until the host's next re-feed while the dollars are
+        // unchanged. Opposite side from `contextTokens`, which IS conversation
+        // state (a new chat genuinely has an empty context window).
+        vendorBudgets: state.vendorBudgets,
       } satisfies Partial<AiAssistantSessionState>;
       Object.assign(state, createDefaultSessionState(), preservedAcrossReset);
     },

package/src/utils/format-usd.ts

@@ -0,0 +1,26 @@
+/**
+ * The ONE place this package renders a USD amount as text.
+ *
+ * Extracted (GENC-1464 budget meter) from the ad-hoc `.toFixed(2)` call sites
+ * that had grown in `formatBlockedReason` and the settings modal's cost blocks,
+ * so the blocked banner, the session-cost figures and the budget meter cannot
+ * drift to different precisions. Two decimals always — money columns jitter
+ * when "$5.1" sits above "$5.10".
+ *
+ * @internal
+ */
+
+/** Decimal places for every USD figure this package renders. */
+export const USD_DECIMALS = 2;
+
+/**
+ * The bare amount, no currency sign — `(12.3) => "12.30"`.
+ *
+ * Exists for the settings modal's cost blocks, where the `$` is a separately
+ * styled `<span class="prefix">` and folding it into the text would change the
+ * DOM those styles target.
+ */
+export const formatUsdAmount = (value: number): string => value.toFixed(USD_DECIMALS);
+
+/** A signed USD figure — `(12.3) => "$12.30"`. */
+export const formatUsd = (value: number): string => `$${formatUsdAmount(value)}`;