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

package/dist/index.js

@@ -442,7 +442,7 @@ var inboundMessageSchema = z.discriminatedUnion("type", [
 ]);
 
 // src/version.ts
-var VERSION = true ? "1.0.116" : "0.0.0-test";
+var VERSION = true ? "1.0.118" : "0.0.0-test";
 
 // src/claude-code-harness.ts
 var CLAUDE_CODE_CLIENT_APP = `ai-sdk/harness-claude-code/${VERSION}`;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/harness-claude-code",
-  "version": "1.0.116",
+  "version": "1.0.118",
   "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,
@@ -26,8 +26,8 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/harness": "1.0.112",
-    "@ai-sdk/provider-utils": "5.0.41",
+    "@ai-sdk/harness": "1.0.114",
+    "@ai-sdk/provider-utils": "5.0.43",
     "ws": "^8.21.0"
   },
   "peerDependencies": {

package/src/bridge/create-emit-stream-event.ts

@@ -21,6 +21,7 @@ export type ClaudeMessage = {
   event?: {
     type?: string;
     index?: number;
+    usage?: Record<string, unknown>;
     content_block?: {
       type?: string;
       id?: string;
@@ -72,6 +73,8 @@ export type ClaudeStreamEventState = {
   partialBlocks: Map<number, PartialBlock>;
   stepUsage: Record<string, unknown> | undefined;
   pendingStepToolUseIds: Set<string>;
+  pendingStepAssistantUsage: Record<string, unknown> | undefined;
+  pendingStepDeltaUsage: Record<string, unknown> | undefined;
   pendingStepUsage: Record<string, unknown> | undefined;
   stepOpen: boolean;
   /*
@@ -97,6 +100,8 @@ export function createClaudeStreamEventState(): ClaudeStreamEventState {
     partialBlocks: new Map(),
     stepUsage: undefined,
     pendingStepToolUseIds: new Set(),
+    pendingStepAssistantUsage: undefined,
+    pendingStepDeltaUsage: undefined,
     pendingStepUsage: undefined,
     stepOpen: false,
     mcpToolUseIds: new Set(),
@@ -228,7 +233,7 @@ export function createEmitStreamEvent({
     }
 
     if (type === 'assistant' && msg.message?.content) {
-      const usage = mapUsage(msg.message.usage);
+      const usage = toUsageRecord(msg.message.usage);
       const toolUseIds: string[] = [];
       let opensStep = false;
       for (const block of msg.message.content) {
@@ -274,7 +279,10 @@ export function createEmitStreamEvent({
       }
       if (opensStep || toolUseIds.length === 0) {
         state.stepOpen = true;
-        if (usage) state.pendingStepUsage = usage;
+        if (usage) {
+          state.pendingStepAssistantUsage = usage;
+          updatePendingStepUsage(state);
+        }
       }
       return;
     }
@@ -360,6 +368,8 @@ export function emitFinishStep({
     usage: usage ?? defaultUsage(),
   });
   state.stepUsage = usage ?? state.stepUsage;
+  state.pendingStepAssistantUsage = undefined;
+  state.pendingStepDeltaUsage = undefined;
   state.pendingStepUsage = undefined;
   state.pendingStepToolUseIds = new Set();
   state.stepOpen = false;
@@ -412,7 +422,21 @@ function handleStreamEvent({
   send: Emit;
   toCommonName: (nativeName: string) => string;
 }): void {
-  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;
 
@@ -502,6 +526,35 @@ function handleStreamEvent({
   }
 }
 
+function toUsageRecord(usage: unknown): Record<string, unknown> | undefined {
+  return usage != null && typeof usage === 'object'
+    ? (usage as Record<string, unknown>)
+    : undefined;
+}
+
+function mergeNonNullUsage(
+  current: Record<string, unknown> | undefined,
+  update: Record<string, unknown>,
+): Record<string, unknown> {
+  const merged = { ...current };
+  for (const [key, value] of Object.entries(update)) {
+    if (value != null) {
+      merged[key] = value;
+    }
+  }
+  return merged;
+}
+
+function updatePendingStepUsage(state: ClaudeStreamEventState): void {
+  const assistantUsage = state.pendingStepAssistantUsage;
+  const deltaUsage = state.pendingStepDeltaUsage;
+  state.pendingStepUsage = mapUsage(
+    assistantUsage || deltaUsage
+      ? { ...assistantUsage, ...deltaUsage }
+      : undefined,
+  );
+}
+
 function isTextEntry(entry: unknown): entry is { text?: unknown } {
   return entry != null && typeof entry === 'object' && 'text' in entry;
 }