fix(ai): preserve providerMetadata through client-side UIMessage pipeline

packages/typescript/ai/src/activities/chat/messages.ts

@@ -138,6 +138,7 @@ interface AssistantSegment {
     id: string
     type: 'function'
     function: { name: string; arguments: string }
+    providerMetadata?: Record<string, unknown>
   }>
 }
 
@@ -205,6 +206,9 @@ function buildAssistantMessages(uiMessage: UIMessage): Array<ModelMessage> {
               name: part.name,
               arguments: part.arguments,
             },
+            ...(part.providerMetadata && {
+              providerMetadata: part.providerMetadata,
+            }),
           })
         }
         break
@@ -340,6 +344,9 @@ export function modelMessageToUIMessage(
         name: toolCall.function.name,
         arguments: toolCall.function.arguments,
         state: 'input-complete', // Model messages have complete arguments
+        ...(toolCall.providerMetadata && {
+          providerMetadata: toolCall.providerMetadata,
+        }),
       })
     }
   }

packages/typescript/ai/src/activities/chat/stream/message-updaters.ts

@@ -55,6 +55,7 @@ export function updateToolCallPart(
     name: string
     arguments: string
     state: ToolCallState
+    providerMetadata?: Record<string, unknown>
   },
 ): Array<UIMessage> {
   return messages.map((msg) => {
@@ -67,6 +68,10 @@ export function updateToolCallPart(
       (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCall.id,
     )
 
+    // Carry forward providerMetadata from either the new toolCall or the existing part
+    const providerMetadata =
+      toolCall.providerMetadata ?? existing?.providerMetadata
+
     const toolCallPart: ToolCallPart = {
       type: 'tool-call',
       id: toolCall.id,
@@ -76,6 +81,7 @@ export function updateToolCallPart(
       // Carry forward approval and output from the existing part
       ...(existing?.approval && { approval: { ...existing.approval } }),
       ...(existing?.output !== undefined && { output: existing.output }),
+      ...(providerMetadata && { providerMetadata }),
     }
 
     if (existing) {

packages/typescript/ai/src/activities/chat/stream/processor.ts

@@ -856,6 +856,9 @@ export class StreamProcessor {
         state: initialState,
         parsedArguments: undefined,
         index: chunk.index ?? state.toolCalls.size,
+        ...(chunk.providerMetadata && {
+          providerMetadata: chunk.providerMetadata,
+        }),
       }
 
       state.toolCalls.set(toolCallId, newToolCall)
@@ -870,6 +873,9 @@ export class StreamProcessor {
         name: chunk.toolName,
         arguments: '',
         state: initialState,
+        ...(chunk.providerMetadata && {
+          providerMetadata: chunk.providerMetadata,
+        }),
       })
       this.emitMessagesChange()
 
@@ -1367,6 +1373,9 @@ export class StreamProcessor {
               name: tc.name,
               arguments: tc.arguments,
             },
+            ...(tc.providerMetadata && {
+              providerMetadata: tc.providerMetadata,
+            }),
           })
         }
       }

packages/typescript/ai/src/activities/chat/stream/types.ts

@@ -25,6 +25,8 @@ export interface InternalToolCallState {
   state: ToolCallState
   parsedArguments?: any
   index: number
+  /** Provider-specific metadata (e.g. Gemini thoughtSignature) */
+  providerMetadata?: Record<string, unknown>
 }
 
 /**

packages/typescript/ai/src/types.ts

@@ -295,6 +295,8 @@ export interface ToolCallPart {
   }
   /** Tool execution output (for client tools or after approval) */
   output?: any
+  /** Provider-specific metadata carried through the tool call lifecycle (e.g. Gemini thoughtSignature) */
+  providerMetadata?: Record<string, unknown>
 }
 
 export interface ToolResultPart {