@wrongstack/core 0.275.1 → 0.276.2

package/dist/infrastructure/index.d.ts

@@ -1,9 +1,9 @@
 export { D as DefaultLogger, a as DefaultLoggerOptions, L as LogFormat } from '../logger-D3lV0cUZ.js';
-export { D as DefaultPathResolver, a as DefaultTokenCounter } from '../path-resolver-uVK4BatM.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-MLL6bMlv.js';
+export { D as DefaultPathResolver, a as DefaultTokenCounter } from '../path-resolver-BhIb6mtd.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-CuZGf9fI.js';
 import '../logger-B63L5bTg.js';
-import '../brain-DCkB5_e7.js';
-import '../context-BrLe8pJy.js';
-import '../config-BzFRKkg7.js';
+import '../brain-BLOyN5ZP.js';
+import '../context-CLnUMW5g.js';
+import '../config-Bf5mj-ad.js';
 import '../path-resolver-CPRj4bFY.js';
-import '../compactor-CzSvxM1g.js';
+import '../compactor-DcBpaJsI.js';

package/dist/infrastructure/index.js

@@ -319,6 +319,7 @@ var DefaultTokenCounter = class {
   registry;
   providerId;
   events;
+  sessionId;
   priceCache = /* @__PURE__ */ new Map();
   /** Most recently accounted request's tokens. Used for per-request context pressure. */
   lastInput = 0;
@@ -327,8 +328,13 @@ var DefaultTokenCounter = class {
     this.registry = opts.registry;
     this.providerId = opts.providerId;
     this.events = opts.events;
+    this.sessionId = opts.sessionId;
+  }
+  setSessionId(sessionId) {
+    this.sessionId = sessionId;
   }
   account(usage, model) {
+    const eventSessionId = this.currentSessionId();
     this.input += usage.input;
     this.output += usage.output;
     this.cacheRead += usage.cacheRead ?? 0;
@@ -338,7 +344,7 @@ var DefaultTokenCounter = class {
     const price = model ? this.priceCache.get(model) : void 0;
     if (price) {
       this.applyPrice(usage, price);
-      this.emitAccounted();
+      this.emitAccounted(eventSessionId);
       return;
     }
     if (this.registry && this.providerId && model) {
@@ -352,18 +358,22 @@ var DefaultTokenCounter = class {
           this.priceCache.set(model, p);
           this.applyPrice(usage, p);
         }
-        this.emitAccounted();
+        this.emitAccounted(eventSessionId);
       }).catch(() => {
-        this.events?.emit("token.cost_estimate_unavailable", { model: model ?? "<unknown>" });
-        this.emitAccounted();
+        this.events?.emit("token.cost_estimate_unavailable", {
+          ...eventSessionId ? { sessionId: eventSessionId } : {},
+          model: model ?? "<unknown>"
+        });
+        this.emitAccounted(eventSessionId);
         return void 0;
       });
       return;
     }
-    this.emitAccounted();
+    this.emitAccounted(eventSessionId);
   }
   /** Synchronous variant for code paths that have already resolved the model. */
   accountWithModel(usage, resolved) {
+    const eventSessionId = this.currentSessionId();
     this.input += usage.input;
     this.output += usage.output;
     this.cacheRead += usage.cacheRead ?? 0;
@@ -377,7 +387,7 @@ var DefaultTokenCounter = class {
     }
     this.priceCache.set(resolved.modelId, price);
     this.applyPrice(usage, price);
-    this.emitAccounted();
+    this.emitAccounted(eventSessionId);
   }
   total() {
     return {
@@ -390,6 +400,10 @@ var DefaultTokenCounter = class {
   currentRequestTokens() {
     return { input: this.lastInput, cacheRead: this.lastCacheRead };
   }
+  setCurrentRequestTokens(input, cacheRead) {
+    this.lastInput = input;
+    this.lastCacheRead = cacheRead ?? 0;
+  }
   estimateCost() {
     return {
       input: round4(this.costInput),
@@ -410,12 +424,17 @@ var DefaultTokenCounter = class {
   invalidateCache() {
     this.priceCache.clear();
   }
-  emitAccounted() {
+  emitAccounted(sessionId = this.currentSessionId()) {
     this.events?.emit("token.accounted", {
+      ...sessionId ? { sessionId } : {},
       usage: this.total(),
       cost: { input: this.costInput, output: this.costOutput, total: this.costInput + this.costOutput }
     });
   }
+  currentSessionId() {
+    const value = typeof this.sessionId === "function" ? this.sessionId() : this.sessionId;
+    return typeof value === "string" && value.length > 0 ? value : void 0;
+  }
   reset() {
     this.input = 0;
     this.output = 0;