@ai-sdk/harness-claude-code 1.0.116 → 1.0.118

package/CHANGELOG.md

@@ -1,5 +1,27 @@
 # @ai-sdk/harness-claude-code
 
+## 1.0.118
+
+### Patch Changes
+
+- Updated dependencies [215b25e]
+- Updated dependencies [d4d96bf]
+- Updated dependencies [3456e2c]
+- Updated dependencies [c4e76de]
+  - @ai-sdk/provider-utils@5.0.43
+  - @ai-sdk/harness@1.0.114
+
+## 1.0.117
+
+### Patch Changes
+
+- 5145174: fix(harness-claude-code): use message delta usage for per-step metering
+- Updated dependencies [91c2128]
+- Updated dependencies [2cd80b3]
+- Updated dependencies [2fa5e0e]
+  - @ai-sdk/provider-utils@5.0.42
+  - @ai-sdk/harness@1.0.113
+
 ## 1.0.116
 
 ### Patch Changes

package/dist/bridge/index.mjs

@@ -701,6 +701,8 @@ function createClaudeStreamEventState() {
     partialBlocks: /* @__PURE__ */ new Map(),
     stepUsage: void 0,
     pendingStepToolUseIds: /* @__PURE__ */ new Set(),
+    pendingStepAssistantUsage: void 0,
+    pendingStepDeltaUsage: void 0,
     pendingStepUsage: void 0,
     stepOpen: false,
     mcpToolUseIds: /* @__PURE__ */ new Set(),
@@ -778,7 +780,7 @@ function createEmitStreamEvent({
       return;
     }
     if (type === "assistant" && msg.message?.content) {
-      const usage = mapUsage(msg.message.usage);
+      const usage = toUsageRecord(msg.message.usage);
       const toolUseIds = [];
       let opensStep = false;
       for (const block of msg.message.content) {
@@ -820,7 +822,10 @@ function createEmitStreamEvent({
       }
       if (opensStep || toolUseIds.length === 0) {
         state.stepOpen = true;
-        if (usage) state.pendingStepUsage = usage;
+        if (usage) {
+          state.pendingStepAssistantUsage = usage;
+          updatePendingStepUsage(state);
+        }
       }
       return;
     }
@@ -886,6 +891,8 @@ function emitFinishStep({
     usage: usage ?? defaultUsage()
   });
   state.stepUsage = usage ?? state.stepUsage;
+  state.pendingStepAssistantUsage = void 0;
+  state.pendingStepDeltaUsage = void 0;
   state.pendingStepUsage = void 0;
   state.pendingStepToolUseIds = /* @__PURE__ */ new Set();
   state.stepOpen = false;
@@ -920,7 +927,19 @@ function handleStreamEvent({
   send,
   toCommonName: toCommonName2
 }) {
-  if (!event || typeof event.index !== "number") return;
+  if (!event) return;
+  if (event.type === "message_delta") {
+    const usage = toUsageRecord(event.usage);
+    if (usage) {
+      state.pendingStepDeltaUsage = mergeNonNullUsage(
+        state.pendingStepDeltaUsage,
+        usage
+      );
+      updatePendingStepUsage(state);
+    }
+    return;
+  }
+  if (typeof event.index !== "number") return;
   const index = event.index;
   const partialBlocks = state.partialBlocks;
   if (event.type === "content_block_start") {
@@ -988,6 +1007,25 @@ function handleStreamEvent({
     }
   }
 }
+function toUsageRecord(usage) {
+  return usage != null && typeof usage === "object" ? usage : void 0;
+}
+function mergeNonNullUsage(current, update) {
+  const merged = { ...current };
+  for (const [key, value] of Object.entries(update)) {
+    if (value != null) {
+      merged[key] = value;
+    }
+  }
+  return merged;
+}
+function updatePendingStepUsage(state) {
+  const assistantUsage = state.pendingStepAssistantUsage;
+  const deltaUsage = state.pendingStepDeltaUsage;
+  state.pendingStepUsage = mapUsage(
+    assistantUsage || deltaUsage ? { ...assistantUsage, ...deltaUsage } : void 0
+  );
+}
 function isTextEntry(entry) {
   return entry != null && typeof entry === "object" && "text" in entry;
 }