@invergent/agent-chat-react 2.15.9 → 2.16.1

package/dist/index.d.cts

@@ -61,6 +61,10 @@ interface AgentChatMessage {
     status: AgentChatMessageStatus;
     toolCalls?: AgentChatToolCallInfo[];
     reasoning?: string;
+    /** Provider-reported reasoning usage for this iteration, when available. */
+    reasoningTokens?: number;
+    /** Incoming reasoning chunks counted live, before provider usage arrives. */
+    reasoningDeltaCount?: number;
     systemKind?: AgentChatSystemKind;
     systemMeta?: Record<string, unknown>;
     errorInfo?: AgentChatErrorInfo;

package/dist/index.d.ts

@@ -61,6 +61,10 @@ interface AgentChatMessage {
     status: AgentChatMessageStatus;
     toolCalls?: AgentChatToolCallInfo[];
     reasoning?: string;
+    /** Provider-reported reasoning usage for this iteration, when available. */
+    reasoningTokens?: number;
+    /** Incoming reasoning chunks counted live, before provider usage arrives. */
+    reasoningDeltaCount?: number;
     systemKind?: AgentChatSystemKind;
     systemMeta?: Record<string, unknown>;
     errorInfo?: AgentChatErrorInfo;

package/dist/index.js

@@ -5255,7 +5255,11 @@ function cancelledToolLabel(toolName) {
     // Arbor research-mission tools (the /auto-research coordinator loop).
     idea_tree: "Idea tree",
     dispatch_experiments: "Dispatch experiments",
-    merge_experiment: "Merge experiment"
+    merge_experiment: "Merge experiment",
+    kb_search_pages: "Knowledge base search",
+    kb_list_pages: "Knowledge base index",
+    kb_read_page: "Knowledge base page",
+    vision_analyze: "Image analysis"
   };
   if (map[toolName]) return map[toolName];
   if (toolName.startsWith("mcp__")) return formatMcpToolLabel(toolName);
@@ -5318,16 +5322,27 @@ function extractToolDetail(tc) {
       return code3 ? truncate3(code3.split("\n")[0] ?? "", 40) : null;
     }
     case "web_search":
-    case "web_crawl":
+    case "session_search":
       return stringArg("query");
+    // web_crawl takes a start ``url``, not a query.
+    case "web_crawl":
     case "web_extract": {
       const url = stringArg("url");
-      if (!url) return null;
-      try {
-        return new URL(url).hostname.replace(/^www\./, "");
-      } catch {
-        return truncate3(url, 40);
-      }
+      return url ? hostOf2(url) : null;
+    }
+    case "kb_search_pages": {
+      const query = stringArg("query");
+      return query ? truncate3(query, 60) : null;
+    }
+    case "kb_read_page": {
+      const path = stringArg("path");
+      if (!path) return null;
+      const pages = stringArg("pages");
+      return pages ? `${kbPageName(path)} \xB7 pages ${pages}` : kbPageName(path);
+    }
+    case "vision_analyze": {
+      const image = stringArg("image");
+      return image && !image.startsWith("data:") ? lastPathSegment(image) : null;
     }
     case "skill_view":
     case "skill_manage":
@@ -5360,14 +5375,7 @@ function extractToolDetail(tc) {
         const title = stringArg("title");
         if (title) return truncate3(title, 60);
         const url = stringArg("url");
-        if (url) {
-          try {
-            return new URL(url).hostname.replace(/^www\./, "");
-          } catch {
-            return truncate3(url, 40);
-          }
-        }
-        return null;
+        return url ? hostOf2(url) : null;
       }
       if (action === "retrieve") {
         return stringArg("query");
@@ -5400,6 +5408,34 @@ function lastPathSegment(path) {
 function truncate3(s, max) {
   return s.length > max ? `${s.slice(0, max - 1)}\u2026` : s;
 }
+function hostOf2(url) {
+  try {
+    return new URL(url).hostname.replace(/^www\./, "") || truncate3(url, 40);
+  } catch {
+    return truncate3(url, 40);
+  }
+}
+function kbPageName(path) {
+  return lastPathSegment(path).replace(/\.(md|json)$/, "");
+}
+var _PLURAL_LABEL = {
+  read_file: (n) => `Read ${n} files`,
+  write_file: (n) => `Wrote ${n} files`,
+  patch: (n) => `Edited ${n} files`,
+  search_files: (n) => `Ran ${n} file searches`,
+  terminal: (n) => `Ran ${n} commands`,
+  web_search: (n) => `Ran ${n} web searches`,
+  web_extract: (n) => `Fetched ${n} pages`,
+  session_search: (n) => `Ran ${n} session searches`,
+  kb_search_pages: (n) => `Ran ${n} knowledge base searches`,
+  kb_read_page: (n) => `Read ${n} knowledge base pages`,
+  vision_analyze: (n) => `Looked at ${n} images`,
+  consult_expert: (n) => `Consulted ${n} experts`,
+  delegate_task: (n) => `Delegated ${n} tasks`
+};
+function sameToolGroupLabel(toolName, count) {
+  return _PLURAL_LABEL[toolName]?.(count) ?? `${cancelledToolLabel(toolName)} \xD7 ${count}`;
+}
 function toolRowLabel(tc) {
   const detail = extractToolDetail(tc);
   switch (tc.toolName) {
@@ -5423,7 +5459,22 @@ function toolRowLabel(tc) {
     case "web_search":
       return detail ? `Searched the web for "${detail}"` : "Searched the web";
     case "web_crawl":
-      return detail ? `Crawled "${detail}"` : "Crawled the web";
+      return detail ? `Crawled ${detail}` : "Crawled a site";
+    case "kb_search_pages": {
+      if (!detail) return "Searched the knowledge base";
+      return stringField(parseArgs(tc.args), "mode") === "documents" ? `Looked up document "${detail}"` : `Searched the knowledge base for "${detail}"`;
+    }
+    case "kb_list_pages":
+      return "Browsed the knowledge base index";
+    case "kb_read_page": {
+      const args = parseArgs(tc.args);
+      const path = stringField(args, "path");
+      if (!path) return "Read a knowledge base page";
+      const pages = stringField(args, "pages");
+      return pages ? `Read ${kbPageName(path)}, pages ${pages}` : `Read ${kbPageName(path)}`;
+    }
+    case "vision_analyze":
+      return detail ? `Looked at ${detail}` : "Looked at an image";
     case "web_extract":
       return detail ? `Fetched ${detail}` : "Fetched a page";
     case "session_search":
@@ -10578,6 +10629,17 @@ function useSmoothStream(target, isStreaming) {
   return displayedLen >= target.length ? target : target.slice(0, displayedLen);
 }
 
+// src/runtime/reasoning-tokens.ts
+function formatReasoningTokens(tokens) {
+  if (tokens < 1e3) return String(tokens);
+  return `${(tokens / 1e3).toFixed(1).replace(/\.0$/, "")}k`;
+}
+function reasoningTokenLabel(message, live) {
+  const tokens = message.reasoningTokens ?? message.reasoningDeltaCount;
+  if (tokens === void 0) return live ? "Thinking..." : "Thought";
+  return `${live ? "Thinking..." : "Thought \xB7"} ${formatReasoningTokens(tokens)} tokens`;
+}
+
 // src/lib/next-action.ts
 var _NEXT_ACTION_RE = /<next_action([^>]*)>([\s\S]*?)<\/next_action>\s*/gi;
 var _PARTIAL_NEXT_ACTION_FOOTER_RE = /(?:^|\n)\s*<(?:n(?:e(?:x(?:t(?:_(?:a(?:c(?:t(?:i(?:o(?:n[\s\S]*)?)?)?)?)?)?)?)?)?)?)?$/i;
@@ -12133,7 +12195,7 @@ function SimpleFinalAnswer({
 function deriveIterationLabel(message) {
   const calls = visibleToolCalls(message);
   if (calls.length === 0) {
-    return message.reasoning ? "Thought through the problem" : null;
+    return message.reasoning || message.reasoningTokens ? reasoningTokenLabel(message, false) : null;
   }
   if (calls.every((tc) => tc.toolName === "skill_view")) {
     return skillViewLabel(calls);
@@ -12145,8 +12207,7 @@ function deriveIterationLabel(message) {
   const firstName = calls[0].toolName;
   const allSame = calls.every((tc) => tc.toolName === firstName);
   if (allSame) {
-    const human = cancelledToolLabel(firstName);
-    return `${human} \xD7 ${calls.length}`;
+    return sameToolGroupLabel(firstName, calls.length);
   }
   return `Used ${calls.length} tools`;
 }
@@ -12165,11 +12226,28 @@ function isIterationLive(message) {
   }
   return message.status === "streaming";
 }
+function hasInlineRunningIndicator(messages, viewMode) {
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const message = messages[i];
+    if (message.role === "user") return false;
+    if (message.role !== "assistant") continue;
+    if (viewMode === "simple") {
+      if (message.toolCalls?.some((tc) => tc.toolName === "ask_user_question")) {
+        return false;
+      }
+      return isIterationLive(message) && (!!message.toolCalls?.length || !message.content || message.content === message.reasoning);
+    }
+    return messageToEntries(message, i === messages.length - 1, {}).some(
+      (entry) => entry.kind === "reasoning" && entry.isStreaming
+    );
+  }
+  return false;
+}
 function liveIterationLabel(message) {
   const running = (message.toolCalls ?? []).filter(
     (tc) => tc.status === "running" && !isHiddenSimpleTool(tc)
   );
-  if (running.length === 0) return "Thinking\u2026";
+  if (running.length === 0) return reasoningTokenLabel(message, true);
   if (running.every((tc) => tc.toolName === "skill_view")) {
     return `${skillViewLabel(running)}\u2026`;
   }
@@ -12209,7 +12287,11 @@ var _TOOL_ROW_ICON = {
   research_memory: BookOpenIcon,
   research_outline: ListIcon,
   run_coding_agent: BotIcon,
-  code_run: BotIcon
+  code_run: BotIcon,
+  kb_search_pages: LibraryIcon,
+  kb_list_pages: LibraryIcon,
+  kb_read_page: LibraryIcon,
+  vision_analyze: EyeIcon
 };
 function _toolRowIcon(toolName) {
   return _TOOL_ROW_ICON[toolName] ?? WrenchIcon;
@@ -12226,44 +12308,16 @@ function SimpleDetailRow({ icon: Icon, children }) {
     /* @__PURE__ */ jsx("div", { className: "min-w-0 flex-1 leading-relaxed text-foreground/60", children })
   ] });
 }
-function ClampedReasoning({
-  text,
-  previewParagraphs = 2
-}) {
-  const paragraphs = splitReasoningParagraphs(text);
-  const [expanded, setExpanded] = useState(false);
-  const hasMore = paragraphs.length > previewParagraphs;
-  const visible = expanded || !hasMore ? paragraphs : paragraphs.slice(0, previewParagraphs);
-  return /* @__PURE__ */ jsxs("div", { className: "space-y-2", children: [
-    visible.map((paragraph, i) => /* @__PURE__ */ jsx("p", { className: "whitespace-pre-wrap", children: paragraph }, i)),
-    hasMore && /* @__PURE__ */ jsx(
-      "button",
-      {
-        type: "button",
-        onClick: () => setExpanded((prev) => !prev),
-        className: "text-xs font-medium text-foreground hover:text-foreground/70 cursor-pointer",
-        children: expanded ? "Show less" : "Show more..."
-      }
-    )
-  ] });
-}
-function splitReasoningParagraphs(text) {
-  const trimmed = text.trim();
-  if (!trimmed) return [];
-  const blankSplit = trimmed.split(/\n\s*\n+/).map((p) => p.trim()).filter((p) => p.length > 0);
-  if (blankSplit.length > 1) return blankSplit;
-  return trimmed.split(/\n+/).map((p) => p.trim()).filter((p) => p.length > 0);
-}
 function IterationExpanded({
   message
 }) {
   const calls = visibleToolCalls(message);
-  const reasoning = (message.reasoning ?? "").trim();
+  const hasReasoning = !!message.reasoning || !!message.reasoningTokens;
   const anyVisibleRunning = calls.some((tc) => tc.status === "running");
-  const showDone = !anyVisibleRunning && (calls.length > 0 || !!reasoning);
-  if (!reasoning && calls.length === 0) return null;
+  const showDone = !anyVisibleRunning && (calls.length > 0 || hasReasoning);
+  if (!hasReasoning && calls.length === 0) return null;
   return /* @__PURE__ */ jsxs("div", { className: "ml-2 space-y-3 border-l border-border/40 pl-4 py-1", children: [
-    reasoning && /* @__PURE__ */ jsx(SimpleDetailRow, { icon: ClockIcon, children: /* @__PURE__ */ jsx(ClampedReasoning, { text: reasoning }) }),
+    hasReasoning && /* @__PURE__ */ jsx(SimpleDetailRow, { icon: ClockIcon, children: /* @__PURE__ */ jsx("span", { children: reasoningTokenLabel(message, false) }) }),
     calls.map((tc) => {
       if (tc.toolName === "todo") {
         return /* @__PURE__ */ jsx("div", { className: "pr-1", children: /* @__PURE__ */ jsx(TodoToolBlock, { tc }) }, tc.id);
@@ -12278,6 +12332,7 @@ function IterationExpanded({
 }
 function IterationGroup({
   message,
+  isActive = true,
   sessionId,
   artifactFallbacks,
   onFileSelect
@@ -12297,7 +12352,7 @@ function IterationGroup({
       }
     ) });
   }
-  if (isIterationLive(message)) {
+  if (isActive && isIterationLive(message)) {
     return /* @__PURE__ */ jsx(
       OrbShimmerLabel,
       {
@@ -12311,6 +12366,9 @@ function IterationGroup({
   if (!label) {
     return null;
   }
+  if (!summary && visibleToolCalls(message).length === 0) {
+    return /* @__PURE__ */ jsx("p", { className: "py-0.5 text-sm text-muted-foreground", children: label });
+  }
   const labelTone = summary ? "text-muted-foreground italic" : "text-muted-foreground";
   return /* @__PURE__ */ jsxs("div", { className: "space-y-1", children: [
     /* @__PURE__ */ jsxs(
@@ -12404,6 +12462,7 @@ function LoopResultAffordance({ message }) {
 }
 function SimpleAssistantGroup({
   messages,
+  activeAssistantId,
   isRunning,
   sessionId,
   artifactFallbacks,
@@ -12415,7 +12474,7 @@ function SimpleAssistantGroup({
   const tail = assistantMessages[assistantMessages.length - 1];
   const tailHasTools = !!(tail?.toolCalls && tail.toolCalls.length > 0);
   const tailIsTextOnly = !!tail && !tailHasTools;
-  const finalText = tailIsTextOnly && tail.content ? tail.content : "";
+  const finalText = tailIsTextOnly && tail.content !== tail.reasoning ? tail.content : "";
   const showErrorInfo = !!tail && tail.status === "error" && !!tail.errorInfo;
   const turnComplete = !isRunning && tail?.status === "complete";
   const summaryPref = finalText !== null ? stripAndParseNextAction(finalText).action?.summary ?? "hide" : "hide";
@@ -12452,7 +12511,7 @@ function SimpleAssistantGroup({
         );
       }
       if (message === tail && tailIsTextOnly && finalText) return null;
-      const rawContent = (message.content ?? "").trim();
+      const rawContent = message.content === message.reasoning ? "" : (message.content ?? "").trim();
       const { action, inferredNarration, cleaned } = stripAndParseNextAction(rawContent);
       const narrationLine = action ? action.body.trim().toLowerCase() === "done" ? null : action.body : inferredNarration;
       const askBody = hasUserFacingAskContent(message) ? cleaned : null;
@@ -12462,6 +12521,7 @@ function SimpleAssistantGroup({
           IterationGroup,
           {
             message,
+            isActive: message.id === activeAssistantId,
             sessionId,
             artifactFallbacks,
             onFileSelect
@@ -12494,6 +12554,7 @@ function SimpleAssistantGroup({
 }
 function AssistantGroup({
   messages,
+  activeAssistantId,
   lastGlobalIndex,
   totalMessages,
   isRunning,
@@ -12509,6 +12570,7 @@ function AssistantGroup({
       SimpleAssistantGroup,
       {
         messages,
+        activeAssistantId,
         isRunning,
         sessionId,
         artifactFallbacks,
@@ -12640,6 +12702,15 @@ function ChatThread({
   onOpenIntegrations
 }) {
   const groups = useMemo(() => groupMessages(messages), [messages]);
+  const activeAssistantId = useMemo(() => {
+    if (!isRunning) return void 0;
+    for (let i = messages.length - 1; i >= 0; i--) {
+      const message = messages[i];
+      if (message.role === "user") return void 0;
+      if (message.role === "assistant") return message.id;
+    }
+    return void 0;
+  }, [messages, isRunning]);
   const pendingAsk = useMemo(() => pendingAskCall(messages), [messages]);
   const askPolicy = useMemo(
     () => askComposerPolicy(pendingAsk),
@@ -12647,13 +12718,26 @@ function ChatThread({
   );
   const awaitingInput = askPolicy !== null;
   const orbActivity = useMemo(
-    () => deriveOrbActivity(
-      messages,
-      viewMode === "simple" ? SIMPLE_MODE_ORB_OPTIONS : void 0
-    ),
+    () => {
+      const activity = deriveOrbActivity(
+        messages,
+        viewMode === "simple" ? SIMPLE_MODE_ORB_OPTIONS : void 0
+      );
+      if (viewMode === "simple" && activity.state === "solving") {
+        const latest = [...messages].reverse().find((message) => message.role === "assistant");
+        if (latest && !latest.toolCalls?.some((tc) => tc.status === "running")) {
+          return { ...activity, label: reasoningTokenLabel(latest, true) };
+        }
+      }
+      return activity;
+    },
     [messages, viewMode]
   );
-  const showWorkingOnIt = useDelayedRunningIndicator(isRunning) && !awaitingInput;
+  const inlineRunningIndicator = useMemo(
+    () => hasInlineRunningIndicator(messages, viewMode),
+    [messages, viewMode]
+  );
+  const showWorkingOnIt = useDelayedRunningIndicator(isRunning) && !awaitingInput && !inlineRunningIndicator;
   const composerLocked = askPolicy?.locked ?? false;
   const composerDisabled = disabled || composerLocked;
   const composerDisabledReason = composerLocked ? askPolicy?.reason : disabledReason;
@@ -12839,6 +12923,7 @@ function ChatThread({
             AssistantGroup,
             {
               messages: group.messages,
+              activeAssistantId,
               lastGlobalIndex: group.lastGlobalIndex,
               totalMessages: messages.length,
               isRunning,
@@ -13081,8 +13166,17 @@ function applyAgentChatEvent(state, event) {
       return collectResearchSource(withResult, toolName, event.data);
     }
     case "harness.wake":
-    case "llm.request":
       return { ...nextState, terminal: false, isRunning: true };
+    case "llm.request":
+      return {
+        ...nextState,
+        terminal: false,
+        isRunning: true,
+        hadDeltas: false,
+        messages: nextState.messages.map(
+          (message) => message.role === "assistant" && message.status === "streaming" && !message.toolCalls?.some((tc) => tc.status === "running") ? { ...message, status: "complete" } : message
+        )
+      };
     case "harness.crash":
     case "stream.timeout":
       return nextState;
@@ -13299,22 +13393,31 @@ function browserMarker(event) {
 function applyLlmDelta(state, event) {
   const deltaContent = stringValue2(event.data.content);
   const deltaReasoning = stringValue2(event.data.reasoning);
+  const reasoningTokens = typeof event.data.reasoning_tokens === "number" && Number.isInteger(event.data.reasoning_tokens) && event.data.reasoning_tokens >= 0 ? event.data.reasoning_tokens : void 0;
   const { turnId, iterationIndex } = readTurnMeta(event.data);
   const messages = [...state.messages];
   const lastIdx = findLastAssistantIndex(messages);
   const lastMsg = lastIdx >= 0 ? messages[lastIdx] : null;
   const hasUserAfter = lastIdx >= 0 && hasUserAfterIndex(messages, lastIdx);
+  const matchesIteration = (turnId === void 0 || lastMsg?.turnId === turnId) && (iterationIndex === void 0 || lastMsg?.iterationIndex === iterationIndex);
+  const sameIteration = turnId !== void 0 && iterationIndex !== void 0 && matchesIteration;
+  if (reasoningTokens !== void 0 && !deltaContent && !deltaReasoning && lastMsg && !hasUserAfter && matchesIteration) {
+    messages[lastIdx] = { ...lastMsg, reasoningTokens };
+    return { ...state, messages };
+  }
   const allToolsDone = Boolean(
     lastMsg?.toolCalls?.length && lastMsg.toolCalls.every((tc) => tc.status !== "running")
   );
   const canAppend = Boolean(
-    lastMsg && lastMsg.status === "streaming" && !allToolsDone && !hasUserAfter
+    lastMsg && lastMsg.status === "streaming" && (!allToolsDone || sameIteration) && !hasUserAfter && matchesIteration
   );
   if (canAppend && lastMsg) {
     messages[lastIdx] = {
       ...lastMsg,
       content: deltaContent ? lastMsg.content + deltaContent : lastMsg.content,
       reasoning: deltaReasoning ? (lastMsg.reasoning ?? "") + deltaReasoning : lastMsg.reasoning,
+      reasoningTokens: reasoningTokens ?? lastMsg.reasoningTokens,
+      reasoningDeltaCount: deltaReasoning ? (lastMsg.reasoningDeltaCount ?? 0) + 1 : lastMsg.reasoningDeltaCount,
       turnId: turnId ?? lastMsg.turnId,
       iterationIndex: iterationIndex ?? lastMsg.iterationIndex
     };
@@ -13324,6 +13427,8 @@ function applyLlmDelta(state, event) {
       role: "assistant",
       content: deltaContent,
       reasoning: deltaReasoning || void 0,
+      reasoningTokens,
+      reasoningDeltaCount: deltaReasoning ? 1 : void 0,
       createdAt: /* @__PURE__ */ new Date(),
       status: "streaming",
       turnId,
@@ -13399,6 +13504,11 @@ function applyLlmResponse(state, event) {
   }
   const inputTokens = numberValue(event.data.input_tokens);
   const outputTokens = numberValue(event.data.output_tokens);
+  const reasoningTokens = numberValue(event.data.reasoning_tokens);
+  if (reasoningTokens > 0) {
+    const responseIdx = findLastAssistantIndex(messages);
+    messages[responseIdx] = { ...messages[responseIdx], reasoningTokens };
+  }
   return {
     ...state,
     messages,
@@ -13407,7 +13517,7 @@ function applyLlmResponse(state, event) {
     tokenUsage: {
       inputTokens,
       outputTokens,
-      reasoningTokens: numberValue(event.data.reasoning_tokens),
+      reasoningTokens,
       cachedInputTokens: numberValue(event.data.cache_read_tokens),
       totalTokens: inputTokens + outputTokens,
       contextWindow: numberValue(event.data.context_window),
@@ -13426,6 +13536,8 @@ ${addition}`;
 }
 function applyLlmThinking(state, event) {
   const reasoningText = stringValue2(event.data.reasoning) || stringValue2(event.data.content);
+  const reasoningTokens = optionalNumberValue(event.data.reasoning_tokens);
+  const reasoningDeltaCount = optionalNumberValue(event.data.reasoning_delta_count);
   const { turnId, iterationIndex } = readTurnMeta(event.data);
   const messages = [...state.messages];
   const idx = findLastAssistantIndex(messages);
@@ -13434,12 +13546,16 @@ function applyLlmThinking(state, event) {
     prev?.toolCalls?.length && prev.toolCalls.every((tc) => tc.status !== "running")
   );
   const hasUserAfter = idx >= 0 && hasUserAfterIndex(messages, idx);
-  if (!prev || allToolsDone || hasUserAfter) {
+  const sameIteration = turnId !== void 0 && iterationIndex !== void 0 && (prev?.turnId === turnId && prev.iterationIndex === iterationIndex || prev?.turnId === void 0 && prev?.iterationIndex === void 0 && prev?.llmResponseEventId === void 0);
+  const differentIteration = !!prev && (turnId !== void 0 && prev.turnId !== void 0 && turnId !== prev.turnId || iterationIndex !== void 0 && prev.iterationIndex !== void 0 && iterationIndex !== prev.iterationIndex);
+  if (!prev || allToolsDone && !sameIteration || hasUserAfter || differentIteration) {
     messages.push({
       id: `evt-${event.eventId}`,
       role: "assistant",
       content: "",
       reasoning: reasoningText,
+      reasoningTokens: reasoningTokens ?? void 0,
+      reasoningDeltaCount: reasoningDeltaCount ?? void 0,
       createdAt: /* @__PURE__ */ new Date(),
       status: "streaming",
       turnId,
@@ -13449,6 +13565,8 @@ function applyLlmThinking(state, event) {
     messages[idx] = {
       ...prev,
       reasoning: appendText(prev.reasoning, reasoningText),
+      reasoningTokens: reasoningTokens ?? prev.reasoningTokens,
+      reasoningDeltaCount: reasoningDeltaCount ?? prev.reasoningDeltaCount,
       turnId: turnId ?? prev.turnId,
       iterationIndex: iterationIndex ?? prev.iterationIndex
     };