@cognipeer/agent-sdk 0.8.7 → 0.8.8

package/dist/index.cjs

@@ -812,6 +812,28 @@ function normalizeUsage(raw) {
   };
   return normalized;
 }
+function recordUsage(state, modelName, normalized) {
+  const usageState = state.usage || { perRequest: [], totals: {} };
+  const id = `${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
+  const cachedInput = normalized.prompt_tokens_details?.cached_tokens;
+  usageState.perRequest.push({
+    id,
+    modelName,
+    usage: normalized,
+    timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+    turn: usageState.perRequest.length + 1,
+    cachedInput
+  });
+  const previous = usageState.totals[modelName] || { input: 0, output: 0, total: 0, cachedInput: 0 };
+  usageState.totals[modelName] = {
+    input: previous.input + (Number(normalized.prompt_tokens) || 0),
+    output: previous.output + (Number(normalized.completion_tokens) || 0),
+    total: previous.total + (Number(normalized.total_tokens) || 0),
+    cachedInput: previous.cachedInput + (Number(cachedInput) || 0)
+  };
+  state.usage = usageState;
+  return id;
+}
 function num(...vals) {
   for (const v of vals) {
     if (v === 0) return 0;
@@ -3181,24 +3203,7 @@ function createAgentCoreNode(opts) {
       messageList: messagesWithResponse
     });
     if (normalized) {
-      const usageState = state.usage || { perRequest: [], totals: {} };
-      const id = `${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
-      const turn = usageState.perRequest.length + 1;
-      const timestamp = (/* @__PURE__ */ new Date()).toISOString();
-      const cachedInputTok = normalized.prompt_tokens_details?.cached_tokens;
-      usageState.perRequest.push({ id, modelName, usage: normalized, timestamp, turn, cachedInput: cachedInputTok });
-      const inputTok = normalized.prompt_tokens;
-      const outputTok = normalized.completion_tokens;
-      const totalTok = normalized.total_tokens;
-      const key = modelName;
-      const agg = usageState.totals[key] || { input: 0, output: 0, total: 0, cachedInput: 0 };
-      usageState.totals[key] = {
-        input: agg.input + (Number(inputTok) || 0),
-        output: agg.output + (Number(outputTok) || 0),
-        total: agg.total + (Number(totalTok) || 0),
-        cachedInput: agg.cachedInput + (Number(cachedInputTok) || 0)
-      };
-      state.usage = usageState;
+      recordUsage(state, modelName, normalized);
     }
     return { messages: messagesWithResponse, usage: state.usage };
   };
@@ -5891,24 +5896,7 @@ function createAgentNode(opts) {
       messageList: messagesWithResponse
     });
     if (normalized) {
-      const usageState = state.usage || { perRequest: [], totals: {} };
-      const id = `${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
-      const turn = usageState.perRequest.length + 1;
-      const timestamp = (/* @__PURE__ */ new Date()).toISOString();
-      const cachedInputTok = normalized.prompt_tokens_details.cached_tokens;
-      usageState.perRequest.push({ id, modelName, usage: normalized, timestamp, turn, cachedInput: cachedInputTok });
-      const inputTok = normalized.prompt_tokens;
-      const outputTok = normalized.completion_tokens;
-      const totalTok = normalized.total_tokens;
-      const key = modelName;
-      const agg = usageState.totals[key] || { input: 0, output: 0, total: 0, cachedInput: 0 };
-      usageState.totals[key] = {
-        input: agg.input + (Number(inputTok) || 0),
-        output: agg.output + (Number(outputTok) || 0),
-        total: agg.total + (Number(totalTok) || 0),
-        cachedInput: agg.cachedInput + (Number(cachedInputTok) || 0)
-      };
-      state.usage = usageState;
+      recordUsage(state, modelName, normalized);
     }
     return { messages: messagesWithResponse, usage: state.usage };
   };
@@ -6358,6 +6346,7 @@ ${canonicalFacts.length > 0 ? canonicalFacts.map((fact) => `- ${fact.key}: ${fac
         outputTokensActual = normalized.completion_tokens;
         totalTokensActual = normalized.total_tokens;
         cachedInputTokens = normalized.prompt_tokens_details?.cached_tokens;
+        recordUsage(state, getModelName(model) || "unknown_model", normalized);
       }
     } catch (err) {
       durationMs = Date.now() - startTime;