@mastra/ai-sdk 1.4.2-alpha.0 → 1.4.2

package/dist/index.js

@@ -9,7 +9,7 @@ import { TripWire, MessageList, aiV5ModelMessageToV2PromptMessage } from '@mastr
 import { RequestContext } from '@mastra/core/di';
 import { WorkingMemory, MessageHistory, SemanticRecall } from '@mastra/core/processors';
 
-// ../../packages/_vendored/ai_v5/dist/chunk-UDIOFQR2.js
+// ../../packages/_vendored/ai_v5/dist/chunk-TJMMQ5MY.js
 var marker = "vercel.ai.error";
 var symbol = Symbol.for(marker);
 var _a;
@@ -188,9 +188,6 @@ var ParseError = class extends Error {
     super(message), this.name = "ParseError", this.type = options.type, this.field = options.field, this.value = options.value, this.line = options.line;
   }
 };
-var LF = 10;
-var CR = 13;
-var SPACE = 32;
 function noop(_arg) {
 }
 function createParser(callbacks) {
@@ -198,109 +195,39 @@ function createParser(callbacks) {
     throw new TypeError(
       "`callbacks` must be an object, got a function instead. Did you mean `{onEvent: fn}`?"
     );
-  const { onEvent = noop, onError = noop, onRetry = noop, onComment } = callbacks, pendingFragments = [];
-  let isFirstChunk = true, id, data = "", dataLines = 0, eventType;
-  function feed(chunk) {
-    if (isFirstChunk && (isFirstChunk = false, chunk.charCodeAt(0) === 239 && chunk.charCodeAt(1) === 187 && chunk.charCodeAt(2) === 191 && (chunk = chunk.slice(3))), pendingFragments.length === 0) {
-      const trailing2 = processLines(chunk);
-      trailing2 !== "" && pendingFragments.push(trailing2);
-      return;
-    }
-    if (chunk.indexOf(`
-`) === -1 && chunk.indexOf("\r") === -1) {
-      pendingFragments.push(chunk);
-      return;
-    }
-    pendingFragments.push(chunk);
-    const input = pendingFragments.join("");
-    pendingFragments.length = 0;
-    const trailing = processLines(input);
-    trailing !== "" && pendingFragments.push(trailing);
-  }
-  function processLines(chunk) {
-    let searchIndex = 0;
-    if (chunk.indexOf("\r") === -1) {
-      let lfIndex = chunk.indexOf(`
-`, searchIndex);
-      for (; lfIndex !== -1; ) {
-        if (searchIndex === lfIndex) {
-          dataLines > 0 && onEvent({ id, event: eventType, data }), id = void 0, data = "", dataLines = 0, eventType = void 0, searchIndex = lfIndex + 1, lfIndex = chunk.indexOf(`
-`, searchIndex);
-          continue;
-        }
-        const firstCharCode = chunk.charCodeAt(searchIndex);
-        if (isDataPrefix(chunk, searchIndex, firstCharCode)) {
-          const valueStart = chunk.charCodeAt(searchIndex + 5) === SPACE ? searchIndex + 6 : searchIndex + 5, value = chunk.slice(valueStart, lfIndex);
-          if (dataLines === 0 && chunk.charCodeAt(lfIndex + 1) === LF) {
-            onEvent({ id, event: eventType, data: value }), id = void 0, data = "", eventType = void 0, searchIndex = lfIndex + 2, lfIndex = chunk.indexOf(`
-`, searchIndex);
-            continue;
-          }
-          data = dataLines === 0 ? value : `${data}
-${value}`, dataLines++;
-        } else isEventPrefix(chunk, searchIndex, firstCharCode) ? eventType = chunk.slice(
-          chunk.charCodeAt(searchIndex + 6) === SPACE ? searchIndex + 7 : searchIndex + 6,
-          lfIndex
-        ) || void 0 : parseLine(chunk, searchIndex, lfIndex);
-        searchIndex = lfIndex + 1, lfIndex = chunk.indexOf(`
-`, searchIndex);
-      }
-      return chunk.slice(searchIndex);
-    }
-    for (; searchIndex < chunk.length; ) {
-      const crIndex = chunk.indexOf("\r", searchIndex), lfIndex = chunk.indexOf(`
-`, searchIndex);
-      let lineEnd = -1;
-      if (crIndex !== -1 && lfIndex !== -1 ? lineEnd = crIndex < lfIndex ? crIndex : lfIndex : crIndex !== -1 ? crIndex === chunk.length - 1 ? lineEnd = -1 : lineEnd = crIndex : lfIndex !== -1 && (lineEnd = lfIndex), lineEnd === -1)
-        break;
-      parseLine(chunk, searchIndex, lineEnd), searchIndex = lineEnd + 1, chunk.charCodeAt(searchIndex - 1) === CR && chunk.charCodeAt(searchIndex) === LF && searchIndex++;
-    }
-    return chunk.slice(searchIndex);
-  }
-  function parseLine(chunk, start, end) {
-    if (start === end) {
+  const { onEvent = noop, onError = noop, onRetry = noop, onComment } = callbacks;
+  let incompleteLine = "", isFirstChunk = true, id, data = "", eventType = "";
+  function feed(newChunk) {
+    const chunk = isFirstChunk ? newChunk.replace(/^\xEF\xBB\xBF/, "") : newChunk, [complete, incomplete] = splitLines(`${incompleteLine}${chunk}`);
+    for (const line of complete)
+      parseLine(line);
+    incompleteLine = incomplete, isFirstChunk = false;
+  }
+  function parseLine(line) {
+    if (line === "") {
       dispatchEvent();
       return;
     }
-    const firstCharCode = chunk.charCodeAt(start);
-    if (isDataPrefix(chunk, start, firstCharCode)) {
-      const valueStart = chunk.charCodeAt(start + 5) === SPACE ? start + 6 : start + 5, value2 = chunk.slice(valueStart, end);
-      data = dataLines === 0 ? value2 : `${data}
-${value2}`, dataLines++;
-      return;
-    }
-    if (isEventPrefix(chunk, start, firstCharCode)) {
-      eventType = chunk.slice(chunk.charCodeAt(start + 6) === SPACE ? start + 7 : start + 6, end) || void 0;
-      return;
-    }
-    if (firstCharCode === 105 && chunk.charCodeAt(start + 1) === 100 && chunk.charCodeAt(start + 2) === 58) {
-      const value2 = chunk.slice(chunk.charCodeAt(start + 3) === SPACE ? start + 4 : start + 3, end);
-      id = value2.includes("\0") ? void 0 : value2;
-      return;
-    }
-    if (firstCharCode === 58) {
-      if (onComment) {
-        const line2 = chunk.slice(start, end);
-        onComment(line2.slice(chunk.charCodeAt(start + 1) === SPACE ? 2 : 1));
-      }
+    if (line.startsWith(":")) {
+      onComment && onComment(line.slice(line.startsWith(": ") ? 2 : 1));
       return;
     }
-    const line = chunk.slice(start, end), fieldSeparatorIndex = line.indexOf(":");
-    if (fieldSeparatorIndex === -1) {
-      processField(line, "", line);
+    const fieldSeparatorIndex = line.indexOf(":");
+    if (fieldSeparatorIndex !== -1) {
+      const field = line.slice(0, fieldSeparatorIndex), offset = line[fieldSeparatorIndex + 1] === " " ? 2 : 1, value = line.slice(fieldSeparatorIndex + offset);
+      processField(field, value, line);
       return;
     }
-    const field = line.slice(0, fieldSeparatorIndex), offset = line.charCodeAt(fieldSeparatorIndex + 1) === SPACE ? 2 : 1, value = line.slice(fieldSeparatorIndex + offset);
-    processField(field, value, line);
+    processField(line, "", line);
   }
   function processField(field, value, line) {
     switch (field) {
       case "event":
-        eventType = value || void 0;
+        eventType = value;
         break;
       case "data":
-        data = dataLines === 0 ? value : `${data}
-${value}`, dataLines++;
+        data = `${data}${value}
+`;
         break;
       case "id":
         id = value.includes("\0") ? void 0 : value;
@@ -325,26 +252,37 @@ ${value}`, dataLines++;
     }
   }
   function dispatchEvent() {
-    dataLines > 0 && onEvent({
+    data.length > 0 && onEvent({
       id,
-      event: eventType,
-      data
-    }), id = void 0, data = "", dataLines = 0, eventType = void 0;
+      event: eventType || void 0,
+      // If the data buffer's last character is a U+000A LINE FEED (LF) character,
+      // then remove the last character from the data buffer.
+      data: data.endsWith(`
+`) ? data.slice(0, -1) : data
+    }), id = void 0, data = "", eventType = "";
   }
   function reset(options = {}) {
-    if (options.consume && pendingFragments.length > 0) {
-      const incompleteLine = pendingFragments.join("");
-      parseLine(incompleteLine, 0, incompleteLine.length);
-    }
-    isFirstChunk = true, id = void 0, data = "", dataLines = 0, eventType = void 0, pendingFragments.length = 0;
+    incompleteLine && options.consume && parseLine(incompleteLine), isFirstChunk = true, id = void 0, data = "", eventType = "", incompleteLine = "";
   }
   return { feed, reset };
 }
-function isDataPrefix(chunk, i, firstCharCode) {
-  return firstCharCode === 100 && chunk.charCodeAt(i + 1) === 97 && chunk.charCodeAt(i + 2) === 116 && chunk.charCodeAt(i + 3) === 97 && chunk.charCodeAt(i + 4) === 58;
-}
-function isEventPrefix(chunk, i, firstCharCode) {
-  return firstCharCode === 101 && chunk.charCodeAt(i + 1) === 118 && chunk.charCodeAt(i + 2) === 101 && chunk.charCodeAt(i + 3) === 110 && chunk.charCodeAt(i + 4) === 116 && chunk.charCodeAt(i + 5) === 58;
+function splitLines(chunk) {
+  const lines = [];
+  let incompleteLine = "", searchIndex = 0;
+  for (; searchIndex < chunk.length; ) {
+    const crIndex = chunk.indexOf("\r", searchIndex), lfIndex = chunk.indexOf(`
+`, searchIndex);
+    let lineEnd = -1;
+    if (crIndex !== -1 && lfIndex !== -1 ? lineEnd = Math.min(crIndex, lfIndex) : crIndex !== -1 ? crIndex === chunk.length - 1 ? lineEnd = -1 : lineEnd = crIndex : lfIndex !== -1 && (lineEnd = lfIndex), lineEnd === -1) {
+      incompleteLine = chunk.slice(searchIndex);
+      break;
+    } else {
+      const line = chunk.slice(searchIndex, lineEnd);
+      lines.push(line), searchIndex = lineEnd + 1, chunk[searchIndex - 1] === "\r" && chunk[searchIndex] === `
+` && searchIndex++;
+    }
+  }
+  return [lines, incompleteLine];
 }
 var EventSourceParserStream = class extends TransformStream {
   constructor({ onError, onRetry, onComment } = {}) {
@@ -3314,7 +3252,7 @@ async function getVercelRequestId() {
   var _a932;
   return (_a932 = getContext().headers) == null ? void 0 : _a932["x-vercel-id"];
 }
-var VERSION2 = "2.0.86";
+var VERSION2 = "2.0.88";
 var AI_GATEWAY_PROTOCOL_VERSION = "0.0.1";
 function createGatewayProvider(options = {}) {
   var _a932, _b93;
@@ -5754,9 +5692,9 @@ var ParseError2 = class extends Error {
     super(message), this.name = "ParseError", this.type = options.type, this.field = options.field, this.value = options.value, this.line = options.line;
   }
 };
-var LF2 = 10;
-var CR2 = 13;
-var SPACE2 = 32;
+var LF = 10;
+var CR = 13;
+var SPACE = 32;
 function noop2(_arg) {
 }
 function createParser2(callbacks) {
@@ -5795,17 +5733,17 @@ function createParser2(callbacks) {
           continue;
         }
         const firstCharCode = chunk.charCodeAt(searchIndex);
-        if (isDataPrefix2(chunk, searchIndex, firstCharCode)) {
-          const valueStart = chunk.charCodeAt(searchIndex + 5) === SPACE2 ? searchIndex + 6 : searchIndex + 5, value = chunk.slice(valueStart, lfIndex);
-          if (dataLines === 0 && chunk.charCodeAt(lfIndex + 1) === LF2) {
+        if (isDataPrefix(chunk, searchIndex, firstCharCode)) {
+          const valueStart = chunk.charCodeAt(searchIndex + 5) === SPACE ? searchIndex + 6 : searchIndex + 5, value = chunk.slice(valueStart, lfIndex);
+          if (dataLines === 0 && chunk.charCodeAt(lfIndex + 1) === LF) {
             onEvent({ id, event: eventType, data: value }), id = void 0, data = "", eventType = void 0, searchIndex = lfIndex + 2, lfIndex = chunk.indexOf(`
 `, searchIndex);
             continue;
           }
           data = dataLines === 0 ? value : `${data}
 ${value}`, dataLines++;
-        } else isEventPrefix2(chunk, searchIndex, firstCharCode) ? eventType = chunk.slice(
-          chunk.charCodeAt(searchIndex + 6) === SPACE2 ? searchIndex + 7 : searchIndex + 6,
+        } else isEventPrefix(chunk, searchIndex, firstCharCode) ? eventType = chunk.slice(
+          chunk.charCodeAt(searchIndex + 6) === SPACE ? searchIndex + 7 : searchIndex + 6,
           lfIndex
         ) || void 0 : parseLine(chunk, searchIndex, lfIndex);
         searchIndex = lfIndex + 1, lfIndex = chunk.indexOf(`
@@ -5819,7 +5757,7 @@ ${value}`, dataLines++;
       let lineEnd = -1;
       if (crIndex !== -1 && lfIndex !== -1 ? lineEnd = crIndex < lfIndex ? crIndex : lfIndex : crIndex !== -1 ? crIndex === chunk.length - 1 ? lineEnd = -1 : lineEnd = crIndex : lfIndex !== -1 && (lineEnd = lfIndex), lineEnd === -1)
         break;
-      parseLine(chunk, searchIndex, lineEnd), searchIndex = lineEnd + 1, chunk.charCodeAt(searchIndex - 1) === CR2 && chunk.charCodeAt(searchIndex) === LF2 && searchIndex++;
+      parseLine(chunk, searchIndex, lineEnd), searchIndex = lineEnd + 1, chunk.charCodeAt(searchIndex - 1) === CR && chunk.charCodeAt(searchIndex) === LF && searchIndex++;
     }
     return chunk.slice(searchIndex);
   }
@@ -5829,25 +5767,25 @@ ${value}`, dataLines++;
       return;
     }
     const firstCharCode = chunk.charCodeAt(start);
-    if (isDataPrefix2(chunk, start, firstCharCode)) {
-      const valueStart = chunk.charCodeAt(start + 5) === SPACE2 ? start + 6 : start + 5, value2 = chunk.slice(valueStart, end);
+    if (isDataPrefix(chunk, start, firstCharCode)) {
+      const valueStart = chunk.charCodeAt(start + 5) === SPACE ? start + 6 : start + 5, value2 = chunk.slice(valueStart, end);
       data = dataLines === 0 ? value2 : `${data}
 ${value2}`, dataLines++;
       return;
     }
-    if (isEventPrefix2(chunk, start, firstCharCode)) {
-      eventType = chunk.slice(chunk.charCodeAt(start + 6) === SPACE2 ? start + 7 : start + 6, end) || void 0;
+    if (isEventPrefix(chunk, start, firstCharCode)) {
+      eventType = chunk.slice(chunk.charCodeAt(start + 6) === SPACE ? start + 7 : start + 6, end) || void 0;
       return;
     }
     if (firstCharCode === 105 && chunk.charCodeAt(start + 1) === 100 && chunk.charCodeAt(start + 2) === 58) {
-      const value2 = chunk.slice(chunk.charCodeAt(start + 3) === SPACE2 ? start + 4 : start + 3, end);
+      const value2 = chunk.slice(chunk.charCodeAt(start + 3) === SPACE ? start + 4 : start + 3, end);
       id = value2.includes("\0") ? void 0 : value2;
       return;
     }
     if (firstCharCode === 58) {
       if (onComment) {
         const line2 = chunk.slice(start, end);
-        onComment(line2.slice(chunk.charCodeAt(start + 1) === SPACE2 ? 2 : 1));
+        onComment(line2.slice(chunk.charCodeAt(start + 1) === SPACE ? 2 : 1));
       }
       return;
     }
@@ -5856,7 +5794,7 @@ ${value2}`, dataLines++;
       processField(line, "", line);
       return;
     }
-    const field = line.slice(0, fieldSeparatorIndex), offset = line.charCodeAt(fieldSeparatorIndex + 1) === SPACE2 ? 2 : 1, value = line.slice(fieldSeparatorIndex + offset);
+    const field = line.slice(0, fieldSeparatorIndex), offset = line.charCodeAt(fieldSeparatorIndex + 1) === SPACE ? 2 : 1, value = line.slice(fieldSeparatorIndex + offset);
     processField(field, value, line);
   }
   function processField(field, value, line) {
@@ -5906,10 +5844,10 @@ ${value}`, dataLines++;
   }
   return { feed, reset };
 }
-function isDataPrefix2(chunk, i, firstCharCode) {
+function isDataPrefix(chunk, i, firstCharCode) {
   return firstCharCode === 100 && chunk.charCodeAt(i + 1) === 97 && chunk.charCodeAt(i + 2) === 116 && chunk.charCodeAt(i + 3) === 97 && chunk.charCodeAt(i + 4) === 58;
 }
-function isEventPrefix2(chunk, i, firstCharCode) {
+function isEventPrefix(chunk, i, firstCharCode) {
   return firstCharCode === 101 && chunk.charCodeAt(i + 1) === 118 && chunk.charCodeAt(i + 2) === 101 && chunk.charCodeAt(i + 3) === 110 && chunk.charCodeAt(i + 4) === 116 && chunk.charCodeAt(i + 5) === 58;
 }
 var EventSourceParserStream2 = class extends TransformStream {
@@ -6094,7 +6032,7 @@ function withUserAgentSuffix2(headers, ...userAgentSuffixParts) {
   );
   return Object.fromEntries(normalizedHeaders.entries());
 }
-var VERSION4 = "4.0.26";
+var VERSION4 = "4.0.27";
 var getOriginalFetch3 = () => globalThis.fetch;
 var getFromApi2 = async ({
   url,
@@ -9261,7 +9199,7 @@ async function getVercelRequestId2() {
   var _a932;
   return (_a932 = getContext2().headers) == null ? void 0 : _a932["x-vercel-id"];
 }
-var VERSION5 = "3.0.110";
+var VERSION5 = "3.0.112";
 var AI_GATEWAY_PROTOCOL_VERSION2 = "0.0.1";
 function createGatewayProvider2(options = {}) {
   var _a932, _b93;
@@ -11162,6 +11100,10 @@ function createUIMessageStreamResponse2({
     headers: prepareHeaders2(headers, UI_MESSAGE_STREAM_HEADERS2)
   });
 }
+z.record(
+  z.string(),
+  jsonValueSchema2.optional()
+);
 function isDataUIMessageChunk2(chunk) {
   return chunk.type.startsWith("data-");
 }
@@ -11238,6 +11180,9 @@ function processUIMessageStream2({
               if (options.title !== void 0) {
                 anyPart.title = options.title;
               }
+              if (options.toolMetadata !== void 0) {
+                anyPart.toolMetadata = options.toolMetadata;
+              }
               anyPart.providerExecuted = (_a2222 = anyOptions.providerExecuted) != null ? _a2222 : part.providerExecuted;
               const providerMetadata = anyOptions.providerMetadata;
               if (providerMetadata != null) {
@@ -11254,6 +11199,7 @@ function processUIMessageStream2({
                 toolCallId: options.toolCallId,
                 state: options.state,
                 title: options.title,
+                ...options.toolMetadata !== void 0 ? { toolMetadata: options.toolMetadata } : {},
                 input: anyOptions.input,
                 output: anyOptions.output,
                 rawInput: anyOptions.rawInput,
@@ -11283,6 +11229,9 @@ function processUIMessageStream2({
               if (options.title !== void 0) {
                 anyPart.title = options.title;
               }
+              if (options.toolMetadata !== void 0) {
+                anyPart.toolMetadata = options.toolMetadata;
+              }
               anyPart.providerExecuted = (_b222 = anyOptions.providerExecuted) != null ? _b222 : part.providerExecuted;
               const providerMetadata = anyOptions.providerMetadata;
               if (providerMetadata != null) {
@@ -11305,6 +11254,7 @@ function processUIMessageStream2({
                 preliminary: anyOptions.preliminary,
                 providerExecuted: anyOptions.providerExecuted,
                 title: options.title,
+                ...options.toolMetadata !== void 0 ? { toolMetadata: options.toolMetadata } : {},
                 ...anyOptions.providerMetadata != null && (options.state === "output-available" || options.state === "output-error") ? { resultProviderMetadata: anyOptions.providerMetadata } : {},
                 ...anyOptions.providerMetadata != null && !(options.state === "output-available" || options.state === "output-error") ? { callProviderMetadata: anyOptions.providerMetadata } : {}
               });
@@ -11449,7 +11399,8 @@ function processUIMessageStream2({
                 toolName: chunk.toolName,
                 index: toolInvocations.length,
                 dynamic: chunk.dynamic,
-                title: chunk.title
+                title: chunk.title,
+                toolMetadata: chunk.toolMetadata
               };
               if (chunk.dynamic) {
                 updateDynamicToolPart({
@@ -11459,6 +11410,7 @@ function processUIMessageStream2({
                   input: void 0,
                   providerExecuted: chunk.providerExecuted,
                   title: chunk.title,
+                  toolMetadata: chunk.toolMetadata,
                   providerMetadata: chunk.providerMetadata
                 });
               } else {
@@ -11469,6 +11421,7 @@ function processUIMessageStream2({
                   input: void 0,
                   providerExecuted: chunk.providerExecuted,
                   title: chunk.title,
+                  toolMetadata: chunk.toolMetadata,
                   providerMetadata: chunk.providerMetadata
                 });
               }
@@ -11494,7 +11447,8 @@ function processUIMessageStream2({
                   toolName: partialToolCall.toolName,
                   state: "input-streaming",
                   input: partialArgs,
-                  title: partialToolCall.title
+                  title: partialToolCall.title,
+                  toolMetadata: partialToolCall.toolMetadata
                 });
               } else {
                 updateToolPart({
@@ -11502,7 +11456,8 @@ function processUIMessageStream2({
                   toolName: partialToolCall.toolName,
                   state: "input-streaming",
                   input: partialArgs,
-                  title: partialToolCall.title
+                  title: partialToolCall.title,
+                  toolMetadata: partialToolCall.toolMetadata
                 });
               }
               write();
@@ -11517,7 +11472,8 @@ function processUIMessageStream2({
                   input: chunk.input,
                   providerExecuted: chunk.providerExecuted,
                   providerMetadata: chunk.providerMetadata,
-                  title: chunk.title
+                  title: chunk.title,
+                  toolMetadata: chunk.toolMetadata
                 });
               } else {
                 updateToolPart({
@@ -11527,7 +11483,8 @@ function processUIMessageStream2({
                   input: chunk.input,
                   providerExecuted: chunk.providerExecuted,
                   providerMetadata: chunk.providerMetadata,
-                  title: chunk.title
+                  title: chunk.title,
+                  toolMetadata: chunk.toolMetadata
                 });
               }
               write();
@@ -11549,7 +11506,8 @@ function processUIMessageStream2({
                   input: chunk.input,
                   errorText: chunk.errorText,
                   providerExecuted: chunk.providerExecuted,
-                  providerMetadata: chunk.providerMetadata
+                  providerMetadata: chunk.providerMetadata,
+                  toolMetadata: chunk.toolMetadata
                 });
               } else {
                 updateToolPart({
@@ -11560,7 +11518,8 @@ function processUIMessageStream2({
                   rawInput: chunk.input,
                   errorText: chunk.errorText,
                   providerExecuted: chunk.providerExecuted,
-                  providerMetadata: chunk.providerMetadata
+                  providerMetadata: chunk.providerMetadata,
+                  toolMetadata: chunk.toolMetadata
                 });
               }
               write();
@@ -11591,7 +11550,8 @@ function processUIMessageStream2({
                   preliminary: chunk.preliminary,
                   providerExecuted: chunk.providerExecuted,
                   providerMetadata: chunk.providerMetadata,
-                  title: toolInvocation.title
+                  title: toolInvocation.title,
+                  toolMetadata: toolInvocation.toolMetadata
                 });
               } else {
                 updateToolPart({
@@ -11603,7 +11563,8 @@ function processUIMessageStream2({
                   providerExecuted: chunk.providerExecuted,
                   preliminary: chunk.preliminary,
                   providerMetadata: chunk.providerMetadata,
-                  title: toolInvocation.title
+                  title: toolInvocation.title,
+                  toolMetadata: toolInvocation.toolMetadata
                 });
               }
               write();
@@ -11620,7 +11581,8 @@ function processUIMessageStream2({
                   errorText: chunk.errorText,
                   providerExecuted: chunk.providerExecuted,
                   providerMetadata: chunk.providerMetadata,
-                  title: toolInvocation.title
+                  title: toolInvocation.title,
+                  toolMetadata: toolInvocation.toolMetadata
                 });
               } else {
                 updateToolPart({
@@ -11632,7 +11594,8 @@ function processUIMessageStream2({
                   errorText: chunk.errorText,
                   providerExecuted: chunk.providerExecuted,
                   providerMetadata: chunk.providerMetadata,
-                  title: toolInvocation.title
+                  title: toolInvocation.title,
+                  toolMetadata: toolInvocation.toolMetadata
                 });
               }
               write();
@@ -11909,6 +11872,10 @@ function createUIMessageStream2({
     onError
   });
 }
+z.record(
+  z.string(),
+  jsonValueSchema2.optional()
+);
 createIdGenerator2({ prefix: "aiobj", size: 24 });
 createIdGenerator2({ prefix: "aiobj", size: 24 });
 var wrapLanguageModel2 = ({