@mastra/server 1.58.0-alpha.1 → 1.58.0-alpha.3

package/dist/{a2a-CHi9AJ0n.js → a2a-_D2EA4oa.js}

@@ -3,7 +3,7 @@ import { n as createRoute } from "./route-builder-CIk5k35M.js";
 import { r as convertInstructionsToString } from "./utils-BUK03Frl.js";
 import { getAgentFromSystem } from "./server/handlers/agents.js";
 import { TaskStoreVersionConflictError } from "./server/a2a/store.js";
-import { a as agentExecutionResponseSchema, i as agentExecutionBodySchema, r as agentCardResponseSchema, t as a2aAgentIdPathParams } from "./a2a-CnU9zaGR.js";
+import { a as agentExecutionResponseSchema, i as agentExecutionBodySchema, r as agentCardResponseSchema, t as a2aAgentIdPathParams } from "./a2a-8C4O2dVJ.js";
 import { getPublicOrigin } from "./server/handlers/auth.js";
 import { z } from "zod/v4";
 import { isDeepStrictEqual } from "util";
@@ -420,7 +420,9 @@ var a2a_exports = /* @__PURE__ */ __exportAll({
 	handleSetTaskPushNotificationConfig: () => handleSetTaskPushNotificationConfig,
 	handleTaskCancel: () => handleTaskCancel,
 	handleTaskGet: () => handleTaskGet,
-	handleTaskResubscribe: () => handleTaskResubscribe
+	handleTaskList: () => handleTaskList,
+	handleTaskResubscribe: () => handleTaskResubscribe,
+	resolveA2AProtocolVersion: () => resolveA2AProtocolVersion
 });
 const messagePartSchema = z.discriminatedUnion("kind", [
 	z.object({
@@ -462,6 +464,7 @@ const messageSendParamsSchema = z.object({
 	configuration: z.object({
 		acceptedOutputModes: z.array(z.string()).optional(),
 		blocking: z.boolean().optional(),
+		returnImmediately: z.boolean().optional(),
 		historyLength: z.number().optional(),
 		pushNotificationConfig: z.object({
 			url: z.string(),
@@ -476,6 +479,159 @@ const messageSendParamsSchema = z.object({
 });
 const defaultPushNotificationStore = new InMemoryPushNotificationStore();
 const defaultPushNotificationSender = new DefaultPushNotificationSender(defaultPushNotificationStore);
+const V1_TASK_STATES = {
+	submitted: "TASK_STATE_SUBMITTED",
+	working: "TASK_STATE_WORKING",
+	completed: "TASK_STATE_COMPLETED",
+	failed: "TASK_STATE_FAILED",
+	canceled: "TASK_STATE_CANCELED",
+	"input-required": "TASK_STATE_INPUT_REQUIRED",
+	rejected: "TASK_STATE_REJECTED",
+	"auth-required": "TASK_STATE_AUTH_REQUIRED"
+};
+function normalizeV1Part(part) {
+	if ("text" in part) return {
+		kind: "text",
+		text: part.text,
+		metadata: part.metadata
+	};
+	if ("raw" in part) return {
+		kind: "file",
+		file: {
+			bytes: part.raw,
+			mimeType: part.mediaType,
+			name: part.filename
+		},
+		metadata: part.metadata
+	};
+	if ("url" in part) return {
+		kind: "file",
+		file: {
+			uri: part.url,
+			mimeType: part.mediaType,
+			name: part.filename
+		},
+		metadata: part.metadata
+	};
+	return {
+		kind: "data",
+		data: part.data,
+		metadata: part.metadata
+	};
+}
+function normalizeV1Params(params) {
+	if (!params?.message) return params;
+	const configuration = params.configuration;
+	return {
+		...params,
+		message: {
+			...params.message,
+			kind: "message",
+			role: params.message.role === "ROLE_AGENT" ? "agent" : "user",
+			parts: params.message.parts.map((part) => normalizeV1Part(part))
+		},
+		configuration: configuration ? {
+			...configuration,
+			blocking: configuration.returnImmediately === void 0 ? void 0 : !configuration.returnImmediately,
+			pushNotificationConfig: configuration.taskPushNotificationConfig
+		} : void 0
+	};
+}
+function toV1Part(part) {
+	if (part.kind === "text") return {
+		text: part.text,
+		metadata: part.metadata
+	};
+	if (part.kind === "file") return "uri" in part.file ? {
+		url: part.file.uri,
+		filename: part.file.name,
+		mediaType: part.file.mimeType,
+		metadata: part.metadata
+	} : {
+		raw: part.file.bytes,
+		filename: part.file.name,
+		mediaType: part.file.mimeType,
+		metadata: part.metadata
+	};
+	return {
+		data: part.data,
+		metadata: part.metadata
+	};
+}
+function toV1Message(message) {
+	if (!message) return void 0;
+	return {
+		messageId: message.messageId,
+		contextId: message.contextId,
+		taskId: message.taskId,
+		role: message.role === "agent" ? "ROLE_AGENT" : "ROLE_USER",
+		parts: message.parts?.map(toV1Part),
+		metadata: message.metadata,
+		extensions: message.extensions,
+		referenceTaskIds: message.referenceTaskIds
+	};
+}
+function toV1Task(task, { includeArtifacts = true, historyLength } = {}) {
+	return {
+		id: task.id,
+		contextId: task.contextId,
+		status: {
+			state: V1_TASK_STATES[task.status.state] ?? "TASK_STATE_UNSPECIFIED",
+			message: toV1Message(task.status.message),
+			timestamp: task.status.timestamp
+		},
+		artifacts: includeArtifacts ? task.artifacts?.map((artifact) => ({
+			artifactId: artifact.artifactId,
+			name: artifact.name,
+			description: artifact.description,
+			parts: artifact.parts?.map(toV1Part),
+			metadata: artifact.metadata,
+			extensions: artifact.extensions
+		})) : void 0,
+		history: task.history?.slice(historyLength === void 0 ? 0 : -historyLength).map(toV1Message),
+		metadata: task.metadata
+	};
+}
+function toV1Result(result, method) {
+	if (method === "message/send") return result?.kind === "message" ? { message: toV1Message(result) } : { task: toV1Task(result) };
+	if (method === "message/stream" || method === "tasks/resubscribe") {
+		if (result?.kind === "status-update") return { statusUpdate: {
+			taskId: result.taskId,
+			contextId: result.contextId,
+			status: {
+				state: V1_TASK_STATES[result.status.state] ?? "TASK_STATE_UNSPECIFIED",
+				message: toV1Message(result.status.message),
+				timestamp: result.status.timestamp
+			},
+			final: result.final,
+			metadata: result.metadata
+		} };
+		if (result?.kind === "artifact-update") return { artifactUpdate: {
+			taskId: result.taskId,
+			contextId: result.contextId,
+			artifact: {
+				...result.artifact,
+				parts: result.artifact.parts?.map(toV1Part)
+			},
+			append: result.append,
+			lastChunk: result.lastChunk,
+			metadata: result.metadata
+		} };
+		return result?.kind === "message" ? { message: toV1Message(result) } : { task: toV1Task(result) };
+	}
+	if (method === "tasks/get" || method === "tasks/cancel") return toV1Task(result);
+	return result;
+}
+function convertV1Response(response, method) {
+	if (!response || !("result" in response)) return response;
+	return {
+		...response,
+		result: toV1Result(response.result, method)
+	};
+}
+async function* convertV1Stream(stream, method) {
+	for await (const response of stream) yield convertV1Response(response, method);
+}
 function createAgentCardDefaults({ pushNotifications = false } = {}) {
 	return {
 		protocolVersion: "0.3.0",
@@ -1095,6 +1251,25 @@ async function handleTaskGet({ requestId, taskStore, agentId, taskId }) {
 	if (!task) throw MastraA2AError.taskNotFound(taskId);
 	return createSuccessResponse(requestId, task);
 }
+function handleTaskList({ requestId, taskStore, agentId, params }) {
+	const requestedPageSize = params.pageSize ?? 50;
+	const offset = params.pageToken ? Number.parseInt(params.pageToken, 10) : 0;
+	const start = Number.isFinite(offset) && offset >= 0 ? offset : 0;
+	const status = typeof params.status === "string" ? params.status.replace(/^TASK_STATE_/, "").toLowerCase().replaceAll("_", "-") : void 0;
+	const timestampAfter = params.statusTimestampAfter ? Date.parse(params.statusTimestampAfter) : void 0;
+	const matchingTasks = taskStore.list({ agentId }).filter((task) => !params.contextId || task.contextId === params.contextId).filter((task) => !status || task.status.state === status).filter((task) => !timestampAfter || task.status.timestamp !== void 0 && Date.parse(task.status.timestamp) >= timestampAfter);
+	const tasks = matchingTasks.slice(start, start + requestedPageSize).map((task) => toV1Task(task, {
+		includeArtifacts: params.includeArtifacts ?? false,
+		historyLength: params.historyLength
+	}));
+	const nextOffset = start + tasks.length;
+	return createSuccessResponse(requestId, {
+		tasks,
+		nextPageToken: nextOffset < matchingTasks.length ? String(nextOffset) : "",
+		pageSize: requestedPageSize,
+		totalSize: matchingTasks.length
+	});
+}
 async function loadTaskOrThrow({ taskStore, agentId, taskId }) {
 	const task = await taskStore.load({
 		agentId,
@@ -1612,63 +1787,87 @@ async function handleTaskCancel({ requestId, taskStore, pushNotificationSender,
 	}
 	throw MastraA2AError.invalidRequest(`Task ${taskId} was updated concurrently. Retry the request.`);
 }
-async function getAgentExecutionHandler({ requestId, mastra, agentId, requestContext, method, params, taskStore, pushNotificationStore, pushNotificationSender, logger, abortSignal }) {
+async function getAgentExecutionHandler({ requestId, mastra, agentId, requestContext, method, params, taskStore, pushNotificationStore, pushNotificationSender, logger, abortSignal, protocolVersion = "0.3" }) {
 	const agent = await getAgentFromSystem({
 		mastra,
 		agentId
 	});
+	const protocolParams = protocolVersion === "1.0" ? normalizeV1Params(params) : params;
 	const { pushNotificationStore: resolvedPushNotificationStore, pushNotificationSender: resolvedPushNotificationSender } = resolvePushNotificationPair({
 		pushNotificationStore,
 		pushNotificationSender
 	});
 	let taskId;
 	try {
-		taskId = getTaskIdFromParams(params);
+		taskId = getTaskIdFromParams(protocolParams);
 		switch (method) {
-			case "message/send": return await handleMessageSend({
-				requestId,
-				params,
-				taskStore,
-				pushNotificationStore: resolvedPushNotificationStore,
-				pushNotificationSender: resolvedPushNotificationSender,
-				agent,
-				agentId,
-				logger,
-				requestContext
-			});
-			case "message/stream": return await handleMessageStream({
-				requestId,
-				taskStore,
-				params,
-				pushNotificationStore: resolvedPushNotificationStore,
-				pushNotificationSender: resolvedPushNotificationSender,
-				agent,
-				agentId,
-				logger,
-				requestContext,
-				abortSignal
-			});
-			case "tasks/get": return await handleTaskGet({
-				requestId,
-				taskStore,
-				agentId,
-				taskId: taskId || "No task ID provided"
-			});
-			case "tasks/cancel": return await handleTaskCancel({
-				requestId,
-				taskStore,
-				pushNotificationSender: resolvedPushNotificationSender,
-				agentId,
-				taskId: taskId || "No task ID provided",
-				logger
-			});
-			case "tasks/resubscribe": return await handleTaskResubscribe({
-				requestId,
-				taskStore,
-				agentId,
-				taskId: taskId || "No task ID provided",
-				abortSignal
-			});
+			case "message/send": {
+				const result = await handleMessageSend({
+					requestId,
+					params: protocolParams,
+					taskStore,
+					pushNotificationStore: resolvedPushNotificationStore,
+					pushNotificationSender: resolvedPushNotificationSender,
+					agent,
+					agentId,
+					logger,
+					requestContext
+				});
+				return protocolVersion === "1.0" ? convertV1Response(result, method) : result;
+			}
+			case "message/stream": {
+				const result = await handleMessageStream({
+					requestId,
+					taskStore,
+					params: protocolParams,
+					pushNotificationStore: resolvedPushNotificationStore,
+					pushNotificationSender: resolvedPushNotificationSender,
+					agent,
+					agentId,
+					logger,
+					requestContext,
+					abortSignal
+				});
+				return protocolVersion === "1.0" ? convertV1Stream(result, method) : result;
+			}
+			case "tasks/get": {
+				const result = await handleTaskGet({
+					requestId,
+					taskStore,
+					agentId,
+					taskId: taskId || "No task ID provided"
+				});
+				return protocolVersion === "1.0" ? convertV1Response(result, method) : result;
+			}
+			case "tasks/list":
+				if (protocolVersion !== "1.0") throw MastraA2AError.methodNotFound(method);
+				return handleTaskList({
+					requestId,
+					taskStore,
+					agentId,
+					params: protocolParams ?? {}
+				});
+			case "tasks/cancel": {
+				const result = await handleTaskCancel({
+					requestId,
+					taskStore,
+					pushNotificationSender: resolvedPushNotificationSender,
+					agentId,
+					taskId: taskId || "No task ID provided",
+					logger
+				});
+				return protocolVersion === "1.0" ? convertV1Response(result, method) : result;
+			}
+			case "tasks/resubscribe": {
+				const result = handleTaskResubscribe({
+					requestId,
+					taskStore,
+					agentId,
+					taskId: taskId || "No task ID provided",
+					abortSignal
+				});
+				return protocolVersion === "1.0" ? convertV1Stream(result, method) : result;
+			}
 			case "tasks/pushNotificationConfig/set": return await handleSetTaskPushNotificationConfig({
 				requestId,
 				taskStore,
@@ -1705,6 +1904,12 @@ async function getAgentExecutionHandler({ requestId, mastra, agentId, requestCon
 		return normalizeError(error, requestId, taskId, logger);
 	}
 }
+function resolveA2AProtocolVersion(request) {
+	const version = request?.headers.get("A2A-Version")?.trim();
+	if (!version || version === "0.3") return "0.3";
+	if (version === "1.0") return "1.0";
+	throw MastraA2AError.versionNotSupported(version);
+}
 const GET_AGENT_CARD_ROUTE = createRoute({
 	method: "GET",
 	path: "/.well-known/:agentId/agent-card.json",
@@ -1741,23 +1946,31 @@ const AGENT_EXECUTION_ROUTE = createRoute({
 	description: "Executes an agent action via JSON-RPC 2.0 over A2A protocol",
 	tags: ["Agent-to-Agent"],
 	requiresAuth: true,
-	handler: async ({ mastra, agentId, requestContext, taskStore, abortSignal, ...bodyParams }) => {
+	handler: async ({ mastra, agentId, requestContext, taskStore, abortSignal, request, ...bodyParams }) => {
 		const { id: requestId, method } = bodyParams;
+		const params = "params" in bodyParams ? bodyParams.params : void 0;
+		let protocolVersion;
+		try {
+			protocolVersion = resolveA2AProtocolVersion(request);
+		} catch (error) {
+			return createA2AJsonResponse(normalizeError(error, requestId));
+		}
 		const result = await getAgentExecutionHandler({
 			requestId,
 			mastra,
 			agentId,
 			requestContext,
 			method,
-			params: "params" in bodyParams ? bodyParams.params : void 0,
+			params,
 			taskStore,
-			abortSignal
+			abortSignal,
+			protocolVersion
 		});
 		if (method === "message/stream" || method === "tasks/resubscribe") return createA2ASSEResponse(result);
 		return createA2AJsonResponse(result);
 	}
 });
 //#endregion
-export { getAgentExecutionHandler as a, handleListTaskPushNotificationConfig as c, handleSetTaskPushNotificationConfig as d, handleTaskCancel as f, getAgentCardByIdHandler as i, handleMessageSend as l, handleTaskResubscribe as m, GET_AGENT_CARD_ROUTE as n, handleDeleteTaskPushNotificationConfig as o, handleTaskGet as p, a2a_exports as r, handleGetTaskPushNotificationConfig as s, AGENT_EXECUTION_ROUTE as t, handleMessageStream as u };
+export { getAgentExecutionHandler as a, handleListTaskPushNotificationConfig as c, handleSetTaskPushNotificationConfig as d, handleTaskCancel as f, resolveA2AProtocolVersion as g, handleTaskResubscribe as h, getAgentCardByIdHandler as i, handleMessageSend as l, handleTaskList as m, GET_AGENT_CARD_ROUTE as n, handleDeleteTaskPushNotificationConfig as o, handleTaskGet as p, a2a_exports as r, handleGetTaskPushNotificationConfig as s, AGENT_EXECUTION_ROUTE as t, handleMessageStream as u };
 
-//# sourceMappingURL=a2a-CHi9AJ0n.js.map
+//# sourceMappingURL=a2a-_D2EA4oa.js.map