@axlsdk/studio 0.18.2 → 0.19.0

package/dist/middleware.d.cts

@@ -1,5 +1,5 @@
 import { AxlRuntime } from '@axlsdk/axl';
-import { B as BroadcastTarget, C as ConnectionManager, a as BufferCaps, S as StudioEnv } from './connection-manager-8TQqoUtk.cjs';
+import { B as BroadcastTarget, C as ConnectionManager, a as BufferCaps, S as StudioEnv } from './connection-manager-DMjH2G5n.cjs';
 import * as hono_types from 'hono/types';
 import * as hono from 'hono';
 import { IncomingMessage, ServerResponse, Server } from 'node:http';

package/dist/middleware.d.ts

@@ -1,5 +1,5 @@
 import { AxlRuntime } from '@axlsdk/axl';
-import { B as BroadcastTarget, C as ConnectionManager, a as BufferCaps, S as StudioEnv } from './connection-manager-8TQqoUtk.js';
+import { B as BroadcastTarget, C as ConnectionManager, a as BufferCaps, S as StudioEnv } from './connection-manager-DMjH2G5n.js';
 import * as hono_types from 'hono/types';
 import * as hono from 'hono';
 import { IncomingMessage, ServerResponse, Server } from 'node:http';

package/dist/middleware.js

@@ -1,7 +1,7 @@
 import {
   createServer,
   handleWsMessage
-} from "./chunk-RLG3DDQZ.js";
+} from "./chunk-S6HQOGOQ.js";
 
 // src/middleware.ts
 import { resolve as resolve2, dirname } from "path";

package/dist/server/index.cjs

@@ -825,6 +825,7 @@ function emptyRetry() {
 function emptyCostData() {
   return {
     totalCost: 0,
+    unpricedCalls: 0,
     totalTokens: { input: 0, output: 0, reasoning: 0 },
     byAgent: {},
     byModel: {},
@@ -837,7 +838,7 @@ function reduceCost(acc, event) {
   const isWorkflowStart = event.type === "workflow_start";
   if (isWorkflowStart && event.workflow) {
     const byWorkflow2 = { ...acc.byWorkflow };
-    const prev = byWorkflow2[event.workflow] ?? { cost: 0, executions: 0 };
+    const prev = byWorkflow2[event.workflow] ?? { cost: 0, executions: 0, unpricedCalls: 0 };
     byWorkflow2[event.workflow] = { ...prev, executions: prev.executions + 1 };
     return { ...acc, byWorkflow: byWorkflow2 };
   }
@@ -845,6 +846,7 @@ function reduceCost(acc, event) {
   const cost = (0, import_axl2.eventCostContribution)(event);
   if (event.type === "ask_end") return acc;
   const tokens = event.tokens ?? {};
+  const unpricedLeaf = (0, import_axl2.isUnpricedLeaf)(event);
   const totalTokens = event.type === "agent_call_end" ? {
     input: acc.totalTokens.input + finite(tokens.input),
     output: acc.totalTokens.output + finite(tokens.output),
@@ -852,15 +854,25 @@ function reduceCost(acc, event) {
   } : acc.totalTokens;
   const byAgent = { ...acc.byAgent };
   if (event.agent) {
-    const prev = byAgent[event.agent] ?? { cost: 0, calls: 0 };
-    byAgent[event.agent] = { cost: prev.cost + cost, calls: prev.calls + 1 };
+    const prev = byAgent[event.agent] ?? { cost: 0, calls: 0, unpricedCalls: 0 };
+    byAgent[event.agent] = {
+      cost: prev.cost + cost,
+      calls: prev.calls + 1,
+      unpricedCalls: prev.unpricedCalls + (unpricedLeaf ? 1 : 0)
+    };
   }
   const byModel = { ...acc.byModel };
   if (event.model) {
-    const prev = byModel[event.model] ?? { cost: 0, calls: 0, tokens: { input: 0, output: 0 } };
+    const prev = byModel[event.model] ?? {
+      cost: 0,
+      calls: 0,
+      unpricedCalls: 0,
+      tokens: { input: 0, output: 0 }
+    };
     byModel[event.model] = {
       cost: prev.cost + cost,
       calls: prev.calls + 1,
+      unpricedCalls: prev.unpricedCalls + (unpricedLeaf ? 1 : 0),
       tokens: {
         input: prev.tokens.input + finite(tokens.input),
         output: prev.tokens.output + finite(tokens.output)
@@ -869,10 +881,11 @@ function reduceCost(acc, event) {
   }
   const byWorkflow = { ...acc.byWorkflow };
   if (event.workflow) {
-    const prev = byWorkflow[event.workflow] ?? { cost: 0, executions: 0 };
+    const prev = byWorkflow[event.workflow] ?? { cost: 0, executions: 0, unpricedCalls: 0 };
     byWorkflow[event.workflow] = {
       cost: prev.cost + cost,
-      executions: prev.executions + (isWorkflowStart ? 1 : 0)
+      executions: prev.executions + (isWorkflowStart ? 1 : 0),
+      unpricedCalls: prev.unpricedCalls + (unpricedLeaf ? 1 : 0)
     };
   }
   let retry = acc.retry;
@@ -903,15 +916,17 @@ function reduceCost(acc, event) {
     byEmbedder = { ...acc.byEmbedder };
     const modelKey = usage?.model ?? "unknown";
     const embedTokens = typeof usage?.tokens === "number" ? finite(usage.tokens) : 0;
-    const prev = byEmbedder[modelKey] ?? { cost: 0, calls: 0, tokens: 0 };
+    const prev = byEmbedder[modelKey] ?? { cost: 0, calls: 0, unpricedCalls: 0, tokens: 0 };
     byEmbedder[modelKey] = {
       cost: prev.cost + cost,
       calls: prev.calls + 1,
+      unpricedCalls: prev.unpricedCalls + (unpricedLeaf ? 1 : 0),
       tokens: prev.tokens + embedTokens
     };
   }
   return {
     totalCost: acc.totalCost + cost,
+    unpricedCalls: acc.unpricedCalls + (unpricedLeaf ? 1 : 0),
     totalTokens,
     byAgent,
     byModel,