troxy-cli 1.29.36 → 1.29.38

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "troxy-cli",
-  "version": "1.29.36",
+  "version": "1.29.38",
   "description": "A secure control layer for AI agents: policies across payments, messages, logins, destructive actions, model usage, and secrets, all enforceable from the CLI",
   "homepage": "https://troxy.io",
   "bugs": {

package/src/mods/troxy-guard/hooks/register.js

@@ -113,22 +113,24 @@ function troxyCommand(args, paused) {
 }
 
 
-// Session savings tracking: cumulative % savings from routing
-let sessionSavingsPercent = 0;
-let routingCount = 0;
-
-// Cost ratio of models (relative to Opus=1.0)
-const MODEL_COST = {
-  'claude-haiku-4-5-20251001': 0.3,
-  'claude-sonnet-5-5': 0.6,
-  'claude-opus-5-5': 1.0,
-};
+// Session usage tracking, one count per human prompt (not per step).
+// Weights relative to Opus: list prices put Haiku at 0.25 and Sonnet at 0.5; a
+// Pro-plan meter test (30 prompts each) put Haiku at ~0.33. Haiku sits between.
+// ponytail: turn-weighted and ignores cache writes after a model switch, so it
+// is an estimate; weight by r.usage tokens if the number must be exact.
+const MODEL_COST = { haiku: 0.3, sonnet: 0.5, opus: 1.0 };
+const stats = { prompts: 0, routed: 0, cost: 0 }; // cost: actual spend in "requested-model prompts"
+
+function costRatio(fromModel, toModel) {
+  const f = MODEL_COST[tierOf(fromModel)], t = MODEL_COST[tierOf(toModel)];
+  return f && t ? t / f : 1;
+}
 
-function calculateSavings(fromModel, toModel) {
-  const fromCost = MODEL_COST[fromModel] || 1.0;
-  const toCost = MODEL_COST[toModel] || 1.0;
-  if (toCost >= fromCost) return 0; // no savings if routing upward
-  return Math.round(((fromCost - toCost) / fromCost) * 100);
+// e.g. [████████░░░░░░░░░░░░] Troxy extended your usage by 87% · 2 of 3 prompts on a cheaper model
+function usageLine({ prompts, routed, cost }) {
+  const extended = cost > 0 ? Math.round((prompts / cost - 1) * 100) : 0;
+  const filled = prompts ? Math.round((routed / prompts) * 20) : 0;
+  return `[${'█'.repeat(filled)}${'░'.repeat(20 - filled)}] Troxy extended your usage by ${extended}% · ${routed} of ${prompts} prompts on a cheaper model`;
 }
 
 // Where the local daemon looks for this session's routing note (see interceptor.js).
@@ -159,6 +161,10 @@ async function scorePrompt($, e, next) {
     { model: 'haiku', system: SCORER_SYSTEM, prompt: e.text.slice(0, SCORER_PROMPT_CHARS), maxTokens: 8, effort: 'low', timeoutMs: SCORE_TIMEOUT_MS },
   );
   const score = r.isAnswered ? parseScore(r.text) : null;
+  $.ui.log(`troxy scorer: ${r.isAnswered ? JSON.stringify(r.text) : r.reason} -> ${score}`, { to: 'debug' });
+  stats.prompts += 1;
+  stats.cost += 1; // until routeStep says otherwise
+  $.ui.status(usageLine(stats));
   if (score !== null) {
     const u = await $.session.usage();
     decision = { score, tokens: estimateTokens(u.context && u.context.tokens, e.text) };
@@ -171,11 +177,11 @@ async function* routeStep($, e, next) {
   if (!decision || e.agentId) return yield* next(e);
   const model = pickModel(decision.score, e.model, decision.tokens);
   if (model !== e.model) {
-    const savings = calculateSavings(e.model, model);
-    if (savings > 0) {
-      routingCount += 1;
-      sessionSavingsPercent = Math.round((sessionSavingsPercent * (routingCount - 1) + savings) / routingCount);
-      $.ui.toast(`Troxy routed to cheaper model · -${savings}% on this turn`);
+    if (!decision.counted) {
+      decision.counted = true;
+      stats.routed += 1;
+      stats.cost += costRatio(e.model, model) - 1;
+      $.ui.status(usageLine(stats));
     }
     const sid = await $.session.id();
     const file = modRouteFile($.plugin.root, sid);
@@ -197,22 +203,9 @@ export function register(on) {
   on('command.run', { command: 'troxy' }, async ($, e) => {
     const r = troxyCommand(e.args, Boolean(await $.store.get('paused')));
     await $.store.set('paused', r.paused);
+    if (r.paused) $.ui.status(undefined);
     return { text: r.text };
   });
-  // Show session savings bar on every turn
-  on('turn.start', async ($, e, next) => {
-    const paused = Boolean(await $.store.get('paused'));
-
-    if (!paused && routingCount > 0) {
-      const barLength = 20;
-      const filledLength = Math.round((sessionSavingsPercent / 100) * barLength);
-      const emptyLength = barLength - filledLength;
-      const bar = '█'.repeat(filledLength) + '░'.repeat(emptyLength);
-      $.ui.toast(`[${bar}] Session: -${sessionSavingsPercent}% saved`);
-    }
-
-    return next(e);
-  });
   // Routing fails open: a scorer or hook error must never break the session,
   // it just means the requested model is used as-is.
   on('prompt.submit', scorePrompt).catch(($, e, next) => { decision = null; return next(e); });
@@ -220,4 +213,4 @@ export function register(on) {
 }
 
 // Exported for tests only - not part of the mod's runtime contract.
-export const _internal = { toDomainAndFields, guardBash, failClosed, parseScore, pickModel, troxyCommand, modRouteFile, estimateTokens, MODELS };
+export const _internal = { toDomainAndFields, guardBash, failClosed, parseScore, pickModel, troxyCommand, modRouteFile, estimateTokens, MODELS, usageLine, costRatio };

package/src/tests/mods-troxy-guard.test.js

@@ -62,3 +62,13 @@ test('real subprocess: a real Bash command still runs normally through the mod (
     fs.rmSync(workdir, { recursive: true, force: true });
   }
 });
+
+test('usageLine: counts every prompt, extension reflects routed share', () => {
+  const { usageLine, costRatio, MODELS } = _internal;
+  assert.equal(costRatio(MODELS.opus, MODELS.haiku), 0.3);
+  assert.equal(costRatio('some-new-model', MODELS.haiku), 1);
+  // 3 prompts, 2 routed opus->haiku: cost 1 + 0.3 + 0.3 = 1.6, 3/1.6 = +88%
+  assert.match(usageLine({ prompts: 3, routed: 2, cost: 1.6 }), /extended your usage by 88% · 2 of 3 prompts/);
+  assert.match(usageLine({ prompts: 3, routed: 0, cost: 3 }), /by 0% · 0 of 3/);
+  assert.match(usageLine({ prompts: 0, routed: 0, cost: 0 }), /by 0% · 0 of 0/);
+});