@tanstack/ai 0.61.0 → 0.64.0

package/src/utilities/ag-ui-wire.ts

@@ -14,12 +14,14 @@ import type {
   StructuredOutputPart,
   SubagentPart,
   TanStackMessageMetadata,
+  ToolResultOutcome,
   UIMessage,
   UIResourcePart,
 } from '../types'
 import type { MetadataRecord } from './merge-metadata'
 import { tanstackMetadata } from './merge-metadata'
 import { isProviderExecutedToolCall } from './provider-executed'
+import { REDACTED_THINKING_ID_PREFIX } from './reasoning-encrypted-value'
 import { normalizeToolResult } from './tool-result'
 import { wireSubagentInfo, wireSubagentRunId } from './subagent-wire'
 import type { SubagentWireInfo } from './subagent-wire'
@@ -49,6 +51,7 @@ function rebuiltToolMetadata(
   id: string | undefined,
   content: string | null | Array<ContentPart>,
   anchorOwnsUiResources = false,
+  outcome?: ToolResultOutcome,
 ): MetadataRecord | undefined {
   const source: MetadataRecord = isRecord(metadata) ? metadata : {}
   const tanstack = isRecord(source.tanstack) ? { ...source.tanstack } : {}
@@ -61,7 +64,11 @@ function rebuiltToolMetadata(
   }
   const result = {
     ...source,
-    tanstack: { ...tanstack, toolResult },
+    tanstack: {
+      ...tanstack,
+      ...(outcome !== undefined && { toolResultOutcome: outcome }),
+      toolResult,
+    },
   }
   return Object.keys(result).length ? result : undefined
 }
@@ -262,6 +269,7 @@ export function uiMessagesToWire(
           part.id,
           part.content,
           true,
+          part.outcome,
         )
         wire.push({
           role: 'tool',
@@ -306,6 +314,8 @@ export function uiMessagesToWire(
             undefined,
             undefined,
             result,
+            false,
+            part.approval?.approved === false ? 'denied' : undefined,
           ),
         })
       }
@@ -385,6 +395,8 @@ function messageMetadata(
     tanstack.model = previousTanstack.model
   if (previousTanstack?.runId !== undefined)
     tanstack.runId = previousTanstack.runId
+  if (previousTanstack?.run?.id !== undefined)
+    tanstack.run = { id: previousTanstack.run.id }
   if (previousTanstack?.signature !== undefined)
     tanstack.signature = previousTanstack.signature
   const createdAt = coerceCreatedAt(msg.createdAt)
@@ -603,7 +615,10 @@ function collectToolCalls(
 }
 
 function deriveReasoningId(messageId: string, part: MessagePart): string {
-  return `${messageId}-reasoning-${(part as { id?: string }).id ?? hashContent((part as { content: string }).content)}`
+  const id = `${messageId}-reasoning-${(part as { id?: string }).id ?? hashContent((part as { content: string }).content)}`
+  return part.type === 'thinking' && part.redacted
+    ? `${REDACTED_THINKING_ID_PREFIX}${id}`
+    : id
 }
 
 function deriveToolMessageId(toolCallId: string): string {

package/src/utilities/merge-streams.ts

@@ -0,0 +1,34 @@
+/**
+ * Yield the items of several async streams as they arrive. If the reader
+ * stops early, every stream that is still open is closed, so its `finally`
+ * runs.
+ */
+export async function* mergeStreams<T>(streams: Array<AsyncIterable<T>>) {
+  const readers = streams.map((stream) => {
+    const iterator = stream[Symbol.asyncIterator]()
+    return {
+      iterator,
+      next: iterator.next(),
+    }
+  })
+
+  try {
+    while (readers.length > 0) {
+      const indexed = readers.map((reader, index) =>
+        reader.next.then((result) => ({ index, result, reader })),
+      )
+      const winner = await Promise.race(indexed)
+      if (winner.result.done) {
+        readers.splice(winner.index, 1)
+        continue
+      }
+      yield winner.result.value
+      winner.reader.next = winner.reader.iterator.next()
+    }
+  } finally {
+    // The reader stopped early. Close every stream so its finally runs.
+    for (const reader of readers) {
+      void reader.iterator.return?.().catch(() => {})
+    }
+  }
+}

package/src/utilities/reasoning-encrypted-value.ts

@@ -16,3 +16,15 @@ export function reasoningEncryptedValue(opts: {
     ...(opts.timestamp !== undefined ? { timestamp: opts.timestamp } : {}),
   }
 }
+
+/**
+ * Id prefix for the reasoning message of a redacted thinking block (Anthropic
+ * `redacted_thinking`). AG-UI's `encryptedValue` does not say what kind of
+ * bytes it holds (ag-ui-protocol/ag-ui#2884), so the id that `entityId` points
+ * to says it. AG-UI clients keep message ids, but they drop event metadata.
+ */
+export const REDACTED_THINKING_ID_PREFIX = 'redacted_thinking-'
+
+export function isRedactedThinkingId(id: unknown): boolean {
+  return typeof id === 'string' && id.startsWith(REDACTED_THINKING_ID_PREFIX)
+}

package/src/utilities/tool-result.ts

@@ -1,4 +1,4 @@
-import type { ContentPart } from '../types'
+import type { ContentPart, ToolResultOutcome } from '../types'
 
 const CONTENT_PART_TYPES = new Set([
   'text',
@@ -8,6 +8,12 @@ const CONTENT_PART_TYPES = new Set([
   'document',
 ])
 
+export function isToolResultOutcome(
+  value: unknown,
+): value is ToolResultOutcome {
+  return value === 'cancelled' || value === 'denied'
+}
+
 /**
  * Structural check for a single `ContentPart`. A text part must carry a string
  * `content`. Every other part carries a source with a string `value`; a file