@mastra/ai-sdk 1.9.0 → 1.9.1

package/dist/index.js

@@ -13962,6 +13962,7 @@ async function handleChatStream({ mastra, agentId, agentVersion, params, default
 * @param {boolean} [options.sendSources=false] - Whether to include source citations in the stream
 * @param {number} [options.heartbeatMs] - Target interval for periodic SSE comment heartbeats. Already-buffered source events and stream lifecycle signals take priority. Values up to 0 disable heartbeats. `NaN`, positive infinity, and values above 2,147,483,647 throw a `RangeError`.
 * @param {(error: unknown) => string} [options.onError] - Custom error serializer streamed to the client. When omitted, errors are passed through a default serializer that strips sensitive fields (e.g. `APICallError.requestBodyValues`, which holds the system prompt) before they reach the client.
+* @param {Function} [options.messageMetadata] - Maps stream parts to metadata attached to AI SDK start and finish message parts.
 *
 * @returns {ReturnType<typeof registerApiRoute>} A registered API route handler
 *
@@ -13993,7 +13994,7 @@ async function handleChatStream({ mastra, agentId, agentVersion, params, default
 * - If both `agent` and `:agentId` are present, a warning is logged and the fixed `agent` takes precedence
 * - Request context from the incoming request overrides `defaultOptions.requestContext` if both are present
 */
-function chatRoute({ path = "/chat/:agentId", agent, defaultOptions, experimentalTransform, version = "v5", agentVersion, sendStart = true, sendFinish = true, sendReasoning = false, sendSources = false, heartbeatMs, onError }) {
+function chatRoute({ path = "/chat/:agentId", agent, defaultOptions, experimentalTransform, version = "v5", agentVersion, sendStart = true, sendFinish = true, sendReasoning = false, sendSources = false, heartbeatMs, onError, messageMetadata }) {
 	if (!agent && !path.includes("/:agentId")) throw new Error("Path must include :agentId to route to the correct agent or pass the agent explicitly");
 	assertValidHeartbeatMs(heartbeatMs);
 	return registerApiRoute(path, {
@@ -14127,13 +14128,18 @@ function chatRoute({ path = "/chat/:agentId", agent, defaultOptions, experimenta
 			let response;
 			if (version === "v7") response = createUIMessageStreamResponse({ stream: await handleChatStream({
 				...handlerOptions,
-				version: "v7"
+				version: "v7",
+				messageMetadata
 			}) });
 			else if (version === "v6") response = createUIMessageStreamResponse$1({ stream: await handleChatStream({
 				...handlerOptions,
-				version: "v6"
+				version: "v6",
+				messageMetadata
+			}) });
+			else response = createUIMessageStreamResponse$2({ stream: await handleChatStream({
+				...handlerOptions,
+				messageMetadata
 			}) });
-			else response = createUIMessageStreamResponse$2({ stream: await handleChatStream(handlerOptions) });
 			return withSseHeartbeat(response, heartbeatMs);
 		}
 	});