@mastra/client-js 1.41.0-alpha.8 → 1.41.0

package/dist/index.js

@@ -1389,13 +1389,22 @@ var Agent = class extends BaseResource {
 		let currentTextPart = void 0;
 		let currentReasoningPart = void 0;
 		let currentReasoningTextDetail = void 0;
-		function updateToolInvocationPart(toolCallId, invocation) {
+		function updateToolInvocationPart(toolCallId, invocation, partProviderMetadata) {
 			const part = message.parts.find((part) => part.type === "tool-invocation" && part.toolInvocation.toolCallId === toolCallId);
-			if (part != null) part.toolInvocation = invocation;
-			else message.parts.push({
-				type: "tool-invocation",
-				toolInvocation: invocation
-			});
+			if (part != null) {
+				part.toolInvocation = invocation;
+				if (partProviderMetadata !== void 0) part.providerMetadata = {
+					...part.providerMetadata ?? {},
+					...partProviderMetadata
+				};
+			} else {
+				const newPart = {
+					type: "tool-invocation",
+					toolInvocation: invocation
+				};
+				if (partProviderMetadata !== void 0) newPart.providerMetadata = partProviderMetadata;
+				message.parts.push(newPart);
+			}
 		}
 		const data = [];
 		let messageAnnotations = replaceLastMessage ? lastMessage?.annotations : void 0;
@@ -1533,7 +1542,7 @@ var Agent = class extends BaseResource {
 					if (message.toolInvocations == null) message.toolInvocations = [];
 					message.toolInvocations.push(invocation);
 				}
-				updateToolInvocationPart(value.toolCallId, invocation);
+				updateToolInvocationPart(value.toolCallId, invocation, value.providerMetadata);
 				execUpdate();
 				if (onToolCall) {
 					const result = await onToolCall({ toolCall: value });
@@ -1545,7 +1554,7 @@ var Agent = class extends BaseResource {
 							result
 						};
 						message.toolInvocations[message.toolInvocations.length - 1] = invocation;
-						updateToolInvocationPart(value.toolCallId, invocation);
+						updateToolInvocationPart(value.toolCallId, invocation, value.providerMetadata);
 						execUpdate();
 					}
 				}
@@ -1561,7 +1570,7 @@ var Agent = class extends BaseResource {
 					...value
 				};
 				toolInvocations[toolInvocationIndex] = invocation;
-				updateToolInvocationPart(value.toolCallId, invocation);
+				updateToolInvocationPart(value.toolCallId, invocation, value.providerMetadata);
 				execUpdate();
 			},
 			onDataPart(value) {
@@ -1641,13 +1650,22 @@ var Agent = class extends BaseResource {
 		let currentTextPart = void 0;
 		let currentReasoningPart = void 0;
 		let currentReasoningTextDetail = void 0;
-		function updateToolInvocationPart(toolCallId, invocation) {
+		function updateToolInvocationPart(toolCallId, invocation, partProviderMetadata) {
 			const part = message.parts.find((part) => part.type === "tool-invocation" && part.toolInvocation.toolCallId === toolCallId);
-			if (part != null) part.toolInvocation = invocation;
-			else message.parts.push({
-				type: "tool-invocation",
-				toolInvocation: invocation
-			});
+			if (part != null) {
+				part.toolInvocation = invocation;
+				if (partProviderMetadata !== void 0) part.providerMetadata = {
+					...part.providerMetadata ?? {},
+					...partProviderMetadata
+				};
+			} else {
+				const newPart = {
+					type: "tool-invocation",
+					toolInvocation: invocation
+				};
+				if (partProviderMetadata !== void 0) newPart.providerMetadata = partProviderMetadata;
+				message.parts.push(newPart);
+			}
 		}
 		const data = [];
 		let messageAnnotations = replaceLastMessage ? lastMessage?.annotations : void 0;
@@ -1743,7 +1761,7 @@ var Agent = class extends BaseResource {
 							if (message.toolInvocations == null) message.toolInvocations = [];
 							message.toolInvocations.push(invocation);
 						}
-						updateToolInvocationPart(chunk.payload.toolCallId, invocation);
+						updateToolInvocationPart(chunk.payload.toolCallId, invocation, chunk.payload.providerMetadata);
 						execUpdate();
 						if (onToolCall) {
 							const result = await onToolCall({ toolCall: chunk.payload });
@@ -1755,7 +1773,7 @@ var Agent = class extends BaseResource {
 									result
 								};
 								message.toolInvocations[message.toolInvocations.length - 1] = invocation;
-								updateToolInvocationPart(chunk.payload.toolCallId, invocation);
+								updateToolInvocationPart(chunk.payload.toolCallId, invocation, chunk.payload.providerMetadata);
 								execUpdate();
 							}
 						}
@@ -1812,7 +1830,7 @@ var Agent = class extends BaseResource {
 							...chunk.payload
 						};
 						toolInvocations[toolInvocationIndex] = invocation;
-						updateToolInvocationPart(chunk.payload.toolCallId, invocation);
+						updateToolInvocationPart(chunk.payload.toolCallId, invocation, chunk.payload.providerMetadata);
 						execUpdate();
 						break;
 					}
@@ -5630,22 +5648,29 @@ const KNOWN_AGENT_CONTROLLER_EVENT_TYPES = new Set(Object.keys({
 	message_start: true,
 	message_update: true,
 	message_end: true,
+	state_changed: true,
 	tool_input_start: true,
 	tool_input_delta: true,
 	tool_input_end: true,
 	tool_start: true,
 	tool_update: true,
 	shell_output: true,
+	command_exit: true,
 	tool_end: true,
 	tool_approval_required: true,
 	tool_suspended: true,
+	tool_suspension_cancelled: true,
 	mode_changed: true,
 	model_changed: true,
 	thread_changed: true,
 	thread_created: true,
 	thread_deleted: true,
 	subagent_start: true,
+	subagent_text_delta: true,
+	subagent_tool_start: true,
+	subagent_tool_end: true,
 	subagent_end: true,
+	subagent_model_changed: true,
 	task_updated: true,
 	notification: true,
 	notification_summary: true,
@@ -5676,19 +5701,42 @@ const KNOWN_AGENT_CONTROLLER_EVENT_TYPES = new Set(Object.keys({
 function isKnownAgentControllerEvent(event) {
 	return KNOWN_AGENT_CONTROLLER_EVENT_TYPES.has(event.type);
 }
+const toDate = (value) => value instanceof Date ? value : new Date(value);
 function hydrateMessage(message) {
 	return {
 		...message,
-		createdAt: message.createdAt instanceof Date ? message.createdAt : new Date(message.createdAt)
+		createdAt: toDate(message.createdAt)
 	};
 }
-function hydrateEventMessage(event) {
-	if (event.type !== "message_start" && event.type !== "message_update" && event.type !== "message_end") return event;
+function hydrateThread(thread) {
 	return {
-		...event,
-		message: hydrateMessage(event.message)
+		...thread,
+		createdAt: toDate(thread.createdAt),
+		updatedAt: toDate(thread.updatedAt)
 	};
 }
+function isKnownParsedEvent(event) {
+	return KNOWN_AGENT_CONTROLLER_EVENT_TYPES.has(event.type);
+}
+/** The stream carries every timestamp as an ISO string; give consumers back the `Date`s {@link Hydrated} promises. */
+function hydrateKnownEvent(event) {
+	switch (event.type) {
+		case "message_start":
+		case "message_update":
+		case "message_end": return {
+			...event,
+			message: hydrateMessage(event.message)
+		};
+		case "thread_created": return {
+			...event,
+			thread: hydrateThread(event.thread)
+		};
+		default: return event;
+	}
+}
+function hydrateEventTimestamps(event) {
+	return isKnownParsedEvent(event) ? hydrateKnownEvent(event) : event;
+}
 /**
 * A session bound to a `resourceId` within one agent controller. Sessions are
 * get-or-create on the server, so re-creating the same resourceId (and scope)
@@ -5820,7 +5868,7 @@ var AgentControllerSession = class extends BaseResource {
 							if (!data) continue;
 							let event;
 							try {
-								event = hydrateEventMessage(JSON.parse(data));
+								event = hydrateEventTimestamps(JSON.parse(data));
 							} catch {
 								continue;
 							}