@axlsdk/studio 0.18.2 → 0.19.0

package/dist/cli.cjs

@@ -822,6 +822,7 @@ function emptyRetry() {
 function emptyCostData() {
   return {
     totalCost: 0,
+    unpricedCalls: 0,
     totalTokens: { input: 0, output: 0, reasoning: 0 },
     byAgent: {},
     byModel: {},
@@ -834,7 +835,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 };
   }
@@ -842,6 +843,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),
@@ -849,15 +851,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)
@@ -866,10 +878,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;
@@ -900,15 +913,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,