@wrongstack/core 0.267.0 → 0.269.0

package/dist/infrastructure/index.d.ts

@@ -1,9 +1,9 @@
 export { D as DefaultLogger, a as DefaultLoggerOptions, L as LogFormat } from '../logger-DmmQhf4P.js';
-export { D as DefaultPathResolver, a as DefaultTokenCounter } from '../path-resolver-COIMLCQL.js';
-export { C as ContextManagerAction, a as ContextManagerInput, b as ContextManagerResult, c as ContextManagerToolOptions, d as allServers, e as awsServer, f as blockServer, g as braveSearchServer, h as context7Server, i as contextManagerTool, j as createContextManagerTool, k as everArtServer, l as filesystemServer, m as githubServer, n as googleMapsServer, o as miniMaxVisionServer, p as playwrightServer, s as sentinelServer, q as slackServer, z as zaiVisionServer } from '../mcp-servers-C2cBTxUR.js';
+export { D as DefaultPathResolver, a as DefaultTokenCounter } from '../path-resolver-CnQ8SIfh.js';
+export { C as ContextManagerAction, a as ContextManagerInput, b as ContextManagerResult, c as ContextManagerToolOptions, d as allServers, e as awsServer, f as blockServer, g as braveSearchServer, h as context7Server, i as contextManagerTool, j as createContextManagerTool, k as everArtServer, l as filesystemServer, m as githubServer, n as googleMapsServer, o as miniMaxVisionServer, p as playwrightServer, s as sentinelServer, q as slackServer, r as sshManagerServer, z as zaiVisionServer } from '../mcp-servers-DJdZiRcv.js';
 import '../logger-B63L5bTg.js';
-import '../brain-Cdg77tVN.js';
-import '../context-dT5Ueund.js';
-import '../config-Du3pYYln.js';
+import '../brain-BYcK__Ym.js';
+import '../context-Dp87Bcaq.js';
+import '../config-C_ae2k86.js';
 import '../path-resolver-CPRj4bFY.js';
-import '../compactor-iMZ84CXq.js';
+import '../compactor-C2RKEBtC.js';

package/dist/infrastructure/index.js

@@ -287,11 +287,10 @@ var DefaultTokenCounter = class {
     const price = model ? this.priceCache.get(model) : void 0;
     if (price) {
       this.applyPrice(usage, price);
-      this.events?.emit("token.accounted", {
-        usage: this.total(),
-        cost: { input: this.costInput, output: this.costOutput, total: this.costInput + this.costOutput }
-      });
-    } else if (this.registry && this.providerId && model) {
+      this.emitAccounted();
+      return;
+    }
+    if (this.registry && this.providerId && model) {
       if (this.priceCache.size >= PRICE_CACHE_MAX_SIZE) {
         const keys = [...this.priceCache.keys()];
         this.priceCache.delete(keys[0] ?? "");
@@ -301,16 +300,16 @@ var DefaultTokenCounter = class {
           const p = priceFromModel(m);
           this.priceCache.set(model, p);
           this.applyPrice(usage, p);
-          this.events?.emit("token.accounted", {
-            usage: this.total(),
-            cost: { input: this.costInput, output: this.costOutput, total: this.costInput + this.costOutput }
-          });
         }
+        this.emitAccounted();
       }).catch(() => {
         this.events?.emit("token.cost_estimate_unavailable", { model: model ?? "<unknown>" });
+        this.emitAccounted();
         return void 0;
       });
+      return;
     }
+    this.emitAccounted();
   }
   /** Synchronous variant for code paths that have already resolved the model. */
   accountWithModel(usage, resolved) {
@@ -327,10 +326,7 @@ var DefaultTokenCounter = class {
     }
     this.priceCache.set(resolved.modelId, price);
     this.applyPrice(usage, price);
-    this.events?.emit("token.accounted", {
-      usage: this.total(),
-      cost: { input: this.costInput, output: this.costOutput, total: this.costInput + this.costOutput }
-    });
+    this.emitAccounted();
   }
   total() {
     return {
@@ -363,6 +359,12 @@ var DefaultTokenCounter = class {
   invalidateCache() {
     this.priceCache.clear();
   }
+  emitAccounted() {
+    this.events?.emit("token.accounted", {
+      usage: this.total(),
+      cost: { input: this.costInput, output: this.costOutput, total: this.costInput + this.costOutput }
+    });
+  }
   reset() {
     this.input = 0;
     this.output = 0;
@@ -370,6 +372,9 @@ var DefaultTokenCounter = class {
     this.cacheWrite = 0;
     this.costInput = 0;
     this.costOutput = 0;
+    this.lastInput = 0;
+    this.lastCacheRead = 0;
+    this.emitAccounted();
   }
   applyPrice(usage, price) {
     if (price.input) this.costInput += usage.input / 1e6 * price.input;
@@ -377,8 +382,14 @@ var DefaultTokenCounter = class {
     if (usage.cacheRead && price.cacheRead) {
       this.costInput += usage.cacheRead / 1e6 * price.cacheRead;
     }
-    if (usage.cacheWrite && price.cacheWrite) {
-      this.costInput += usage.cacheWrite / 1e6 * price.cacheWrite;
+    const hasCacheWriteSplit = usage.cacheWrite5m !== void 0 || usage.cacheWrite1h !== void 0;
+    const cacheWrite5m = usage.cacheWrite5m ?? (hasCacheWriteSplit ? 0 : usage.cacheWrite);
+    const cacheWrite1h = usage.cacheWrite1h ?? 0;
+    if (cacheWrite5m && (price.cacheWrite5m ?? price.cacheWrite)) {
+      this.costInput += cacheWrite5m / 1e6 * (price.cacheWrite5m ?? price.cacheWrite ?? 0);
+    }
+    if (cacheWrite1h && (price.cacheWrite1h ?? price.cacheWrite)) {
+      this.costInput += cacheWrite1h / 1e6 * (price.cacheWrite1h ?? price.cacheWrite ?? 0);
     }
   }
 };
@@ -387,7 +398,9 @@ function priceFromModel(m) {
     input: m.cost?.input,
     output: m.cost?.output,
     cacheRead: m.cost?.cache_read,
-    cacheWrite: m.cost?.cache_write
+    cacheWrite: m.cost?.cache_write,
+    cacheWrite5m: m.cost?.cache_write_5m ?? m.cost?.cache_write,
+    cacheWrite1h: m.cost?.cache_write_1h ?? (m.cost?.input !== void 0 ? m.cost.input * 2 : void 0)
   };
 }
 function round4(n) {
@@ -513,6 +526,19 @@ var miniMaxVisionServer = () => ({
   allowedTools: ["understand_image"],
   permission: "auto"
 });
+var sshManagerServer = () => ({
+  name: "ssh",
+  description: "Remote SSH management \u2014 execute commands, transfer files, tunnels, health checks (mcp-ssh-manager)",
+  transport: "stdio",
+  command: "npx",
+  args: ["-y", "mcp-ssh-manager"],
+  env: {
+    MCP_SSH_COMPACT_JSON: "true",
+    MCP_SSH_DEFAULT_TIMEOUT: "120000"
+  },
+  permission: "confirm",
+  requestTimeoutMs: 18e4
+});
 var allServers = () => ({
   filesystem: { ...filesystemServer(), enabled: false },
   github: { ...githubServer(), enabled: false },
@@ -526,7 +552,8 @@ var allServers = () => ({
   sentinel: { ...sentinelServer(), enabled: false },
   "zai-vision": { ...zaiVisionServer(), enabled: false },
   "minimax-vision": { ...miniMaxVisionServer(), enabled: false },
-  playwright: { ...playwrightServer(), enabled: false }
+  playwright: { ...playwrightServer(), enabled: false },
+  ssh: { ...sshManagerServer(), enabled: false }
 });
 
 // src/utils/expect-defined.ts
@@ -601,7 +628,7 @@ function hasToolResult(msg) {
 }
 function toolUseIds(msg) {
   const ids = /* @__PURE__ */ new Set();
-  if (!msg || msg.role !== "assistant") return ids;
+  if (msg?.role !== "assistant") return ids;
   for (const block of contentBlocks(msg)) {
     if (block.type === "tool_use") ids.add(block.id);
   }
@@ -609,7 +636,7 @@ function toolUseIds(msg) {
 }
 function toolResultIds(msg) {
   const ids = /* @__PURE__ */ new Set();
-  if (!msg || msg.role !== "user") return ids;
+  if (msg?.role !== "user") return ids;
   for (const block of contentBlocks(msg)) {
     if (block.type === "tool_result") ids.add(block.tool_use_id);
   }
@@ -714,7 +741,7 @@ function calState(key) {
   return state;
 }
 var ESTIMATE_CACHE = /* @__PURE__ */ new Map();
-var ESTIMATE_CACHE_MAX_SIZE = 1e4;
+var ESTIMATE_CACHE_MAX_SIZE = 5e4;
 function getCachedEstimate(key, compute) {
   const existing = ESTIMATE_CACHE.get(key);
   if (existing !== void 0) return existing;
@@ -1080,6 +1107,6 @@ function createContextManagerTool(opts = {}) {
 }
 var contextManagerTool = createContextManagerTool();
 
-export { DefaultLogger, DefaultPathResolver, DefaultTokenCounter, allServers, awsServer, blockServer, braveSearchServer, context7Server, contextManagerTool, createContextManagerTool, everArtServer, filesystemServer, githubServer, googleMapsServer, miniMaxVisionServer, playwrightServer, sentinelServer, slackServer, zaiVisionServer };
+export { DefaultLogger, DefaultPathResolver, DefaultTokenCounter, allServers, awsServer, blockServer, braveSearchServer, context7Server, contextManagerTool, createContextManagerTool, everArtServer, filesystemServer, githubServer, googleMapsServer, miniMaxVisionServer, playwrightServer, sentinelServer, slackServer, sshManagerServer, zaiVisionServer };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map