troxy-cli 1.29.39 → 1.29.40

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "troxy-cli",
-  "version": "1.29.39",
+  "version": "1.29.40",
   "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/.claude-plugin/plugin.json

@@ -2,5 +2,8 @@
   "name": "troxy-guard",
   "version": "0.1.0",
   "description": "Troxy keeps your Claude usage bar going longer. Type /troxy off to pause it.",
-  "author": { "name": "Troxy" }
+  "author": {
+    "name": "Troxy"
+  },
+  "types": "./types/index.d.ts"
 }

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

@@ -119,7 +119,16 @@ function troxyCommand(args, paused) {
 // 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"
+// Kept in $.state (per session) so a `troxy update` hot reload keeps the counts.
+// cost: actual spend in "requested-model prompts".
+const STATS = { plugin: 'troxy-guard', key: 'stats' };
+
+async function addStats($, prompts, routed, cost) {
+  const { value: s = { prompts: 0, routed: 0, cost: 0 } } = await $.state.get(STATS);
+  const next = { prompts: s.prompts + prompts, routed: s.routed + routed, cost: s.cost + cost };
+  await $.state.set(STATS, next);
+  $.ui.status(usageLine(next));
+}
 
 function costRatio(fromModel, toModel) {
   const f = MODEL_COST[tierOf(fromModel)], t = MODEL_COST[tierOf(toModel)];
@@ -162,9 +171,7 @@ async function scorePrompt($, e, next) {
   );
   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));
+  await addStats($, 1, 0, 1); // full cost until routeStep says otherwise
   if (score !== null) {
     const u = await $.session.usage();
     decision = { score, tokens: estimateTokens(u.context && u.context.tokens, e.text) };
@@ -179,9 +186,7 @@ async function* routeStep($, e, next) {
   if (model !== e.model) {
     if (!decision.counted) {
       decision.counted = true;
-      stats.routed += 1;
-      stats.cost += costRatio(e.model, model) - 1;
-      $.ui.status(usageLine(stats));
+      await addStats($, 0, 1, costRatio(e.model, model) - 1);
     }
     const sid = await $.session.id();
     const file = modRouteFile($.plugin.root, sid);
@@ -198,6 +203,9 @@ export function register(on) {
   on('tool.call', { tool: 'Bash' }, guardBash).catch(failClosed);
   on('session.start', async ($, e, next) => {
     await $.command.register({ name: 'troxy', description: 'Pause or resume Troxy model routing', argumentHint: '[on|off|status]' });
+    // session.start fires again on a hot reload: put the bar back from the kept counts
+    const { value } = await $.state.get(STATS);
+    if (value && value.prompts && !(await $.store.get('paused'))) $.ui.status(usageLine(value));
     return next(e);
   });
   on('command.run', { command: 'troxy' }, async ($, e) => {

package/src/mods/troxy-guard/types/index.d.ts

@@ -0,0 +1,7 @@
+export type UsageStats = { prompts: number; routed: number; cost: number }
+
+declare module 'claude-code' {
+  interface PluginState {
+    'troxy-guard': { stats: UsageStats }
+  }
+}