@mastra/server 1.0.4-alpha.2 → 1.1.0-alpha.0

package/dist/chunk-GUG6KRAJ.js

@@ -0,0 +1,666 @@
+import { convertInstructionsToString } from './chunk-ED6NQBHF.js';
+import { getAgentFromSystem } from './chunk-3QUACCLU.js';
+import { createRoute } from './chunk-D5VCL56B.js';
+import { __export } from './chunk-PR4QN5HX.js';
+import { MastraA2AError } from '@mastra/core/a2a';
+import z, { z as z$1 } from 'zod';
+
+// src/server/handlers/a2a.ts
+var a2a_exports = {};
+__export(a2a_exports, {
+  AGENT_EXECUTION_ROUTE: () => AGENT_EXECUTION_ROUTE,
+  GET_AGENT_CARD_ROUTE: () => GET_AGENT_CARD_ROUTE,
+  getAgentCardByIdHandler: () => getAgentCardByIdHandler,
+  getAgentExecutionHandler: () => getAgentExecutionHandler,
+  handleMessageSend: () => handleMessageSend,
+  handleMessageStream: () => handleMessageStream,
+  handleTaskCancel: () => handleTaskCancel,
+  handleTaskGet: () => handleTaskGet
+});
+function normalizeError(error, reqId, taskId, logger) {
+  let a2aError;
+  if (error instanceof MastraA2AError) {
+    a2aError = error;
+  } else if (error instanceof Error) {
+    a2aError = MastraA2AError.internalError(error.message, { stack: error.stack });
+  } else {
+    a2aError = MastraA2AError.internalError("An unknown error occurred.", error);
+  }
+  if (taskId && !a2aError.taskId) {
+    a2aError.taskId = taskId;
+  }
+  logger?.error(`Error processing request (Task: ${a2aError.taskId ?? "N/A"}, ReqID: ${reqId ?? "N/A"}):`, a2aError);
+  return createErrorResponse(reqId, a2aError.toJSONRPCError());
+}
+function createErrorResponse(id, error) {
+  return {
+    jsonrpc: "2.0",
+    id,
+    // Can be null if request ID was invalid/missing
+    error
+  };
+}
+function createSuccessResponse(id, result) {
+  if (!id) {
+    throw MastraA2AError.internalError("Cannot create success response for null ID.");
+  }
+  return {
+    jsonrpc: "2.0",
+    id,
+    result
+  };
+}
+function convertToCoreMessage(message) {
+  return {
+    role: message.role === "user" ? "user" : "assistant",
+    content: message.parts.map((msg) => convertToCoreMessagePart(msg))
+  };
+}
+function convertToCoreMessagePart(part) {
+  switch (part.kind) {
+    case "text":
+      return {
+        type: "text",
+        text: part.text
+      };
+    case "file":
+      return {
+        type: "file",
+        data: "uri" in part.file ? new URL(part.file.uri) : part.file.bytes,
+        mimeType: part.file.mimeType
+      };
+    case "data":
+      throw new Error("Data parts are not supported in core messages");
+  }
+}
+
+// src/server/a2a/tasks.ts
+function isTaskStatusUpdate(update) {
+  return "state" in update && !("parts" in update);
+}
+function isArtifactUpdate(update) {
+  return "kind" in update && update.kind === "artifact-update";
+}
+function applyUpdateToTask(current, update) {
+  let newTask = structuredClone(current);
+  if (isTaskStatusUpdate(update)) {
+    newTask.status = {
+      ...newTask.status,
+      // Keep existing properties if not overwritten
+      ...update,
+      // Apply updates
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    };
+  } else if (isArtifactUpdate(update)) {
+    if (!newTask.artifacts) {
+      newTask.artifacts = [];
+    } else {
+      newTask.artifacts = [...newTask.artifacts];
+    }
+    const artifact = update.artifact;
+    const existingIndex = newTask.artifacts.findIndex((a) => a.name === artifact.name);
+    const existingArtifact = newTask.artifacts[existingIndex];
+    if (existingArtifact) {
+      if (update.append) {
+        const appendedArtifact = JSON.parse(JSON.stringify(existingArtifact));
+        appendedArtifact.parts.push(...artifact.parts);
+        if (artifact.metadata) {
+          appendedArtifact.metadata = {
+            ...appendedArtifact.metadata || {},
+            ...artifact.metadata
+          };
+        }
+        if (artifact.description) appendedArtifact.description = artifact.description;
+        newTask.artifacts[existingIndex] = appendedArtifact;
+      } else {
+        newTask.artifacts[existingIndex] = { ...artifact };
+      }
+    } else {
+      newTask.artifacts.push({ ...artifact });
+    }
+  }
+  return newTask;
+}
+async function loadOrCreateTask({
+  agentId,
+  taskId,
+  taskStore,
+  message,
+  contextId,
+  metadata,
+  logger
+}) {
+  const data = await taskStore.load({ agentId, taskId });
+  if (!data) {
+    const initialTask = {
+      id: taskId,
+      contextId: contextId || crypto.randomUUID(),
+      status: {
+        state: "submitted",
+        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+        message: void 0
+      },
+      artifacts: [],
+      history: [message],
+      metadata,
+      kind: "task"
+    };
+    logger?.info(`[Task ${taskId}] Created new task.`);
+    await taskStore.save({ agentId, data: initialTask });
+    return initialTask;
+  }
+  logger?.info(`[Task ${taskId}] Loaded existing task.`);
+  let updatedData = data;
+  updatedData.history = [...data.history || [], message];
+  const { status } = data;
+  const finalStates = ["completed", "failed", "canceled"];
+  if (finalStates.includes(status.state)) {
+    logger?.warn(`[Task ${taskId}] Received message for task in final state ${status.state}. Restarting.`);
+    updatedData = applyUpdateToTask(updatedData, {
+      state: "submitted",
+      message: void 0
+    });
+  } else if (status.state === "input-required") {
+    logger?.info(`[Task ${taskId}] Changing state from 'input-required' to 'working'.`);
+    updatedData = applyUpdateToTask(updatedData, { state: "working" });
+  } else if (status.state === "working") {
+    logger?.warn(`[Task ${taskId}] Received message while already 'working'. Proceeding.`);
+  }
+  await taskStore.save({ agentId, data: updatedData });
+  return updatedData;
+}
+function createTaskContext({
+  task,
+  userMessage,
+  history,
+  activeCancellations
+}) {
+  return {
+    task: structuredClone(task),
+    userMessage,
+    history: structuredClone(history),
+    isCancelled: () => activeCancellations.has(task.id)
+  };
+}
+var a2aAgentIdPathParams = z.object({
+  agentId: z.string().describe("Unique identifier for the agent")
+});
+a2aAgentIdPathParams.extend({
+  taskId: z.string().describe("Unique identifier for the task")
+});
+var pushNotificationAuthenticationInfoSchema = z.object({
+  schemes: z.array(z.string()).describe("Supported authentication schemes - e.g. Basic, Bearer"),
+  credentials: z.string().optional().describe("Optional credentials")
+});
+var pushNotificationConfigSchema = z.object({
+  url: z.string().describe("URL for sending the push notifications"),
+  id: z.string().optional().describe("Push Notification ID - created by server to support multiple callbacks"),
+  token: z.string().optional().describe("Token unique to this task/session"),
+  authentication: pushNotificationAuthenticationInfoSchema.optional()
+});
+var messageSendConfigurationSchema = z.object({
+  acceptedOutputModes: z.array(z.string()).describe("Accepted output modalities by the client"),
+  blocking: z.boolean().optional().describe("If the server should treat the client as a blocking request"),
+  historyLength: z.number().optional().describe("Number of recent messages to be retrieved"),
+  pushNotificationConfig: pushNotificationConfigSchema.optional()
+});
+var textPartSchema = z.object({
+  kind: z.literal("text").describe("Part type - text for TextParts"),
+  text: z.string().describe("Text content"),
+  metadata: z.record(z.unknown()).optional().describe("Optional metadata associated with the part")
+});
+var fileWithBytesSchema = z.object({
+  bytes: z.string().describe("base64 encoded content of the file"),
+  mimeType: z.string().optional().describe("Optional mimeType for the file"),
+  name: z.string().optional().describe("Optional name for the file")
+});
+var fileWithUriSchema = z.object({
+  uri: z.string().describe("URL for the File content"),
+  mimeType: z.string().optional().describe("Optional mimeType for the file"),
+  name: z.string().optional().describe("Optional name for the file")
+});
+var filePartSchema = z.object({
+  kind: z.literal("file").describe("Part type - file for FileParts"),
+  file: z.union([fileWithBytesSchema, fileWithUriSchema]).describe("File content either as url or bytes"),
+  metadata: z.record(z.unknown()).optional().describe("Optional metadata associated with the part")
+});
+var dataPartSchema = z.object({
+  kind: z.literal("data").describe("Part type - data for DataParts"),
+  data: z.record(z.unknown()).describe("Structured data content"),
+  metadata: z.record(z.unknown()).optional().describe("Optional metadata associated with the part")
+});
+var partSchema = z.union([textPartSchema, filePartSchema, dataPartSchema]);
+var messageSchema = z.object({
+  kind: z.literal("message").describe("Event type"),
+  messageId: z.string().describe("Identifier created by the message creator"),
+  role: z.enum(["user", "agent"]).describe("Message sender's role"),
+  parts: z.array(partSchema).describe("Message content"),
+  contextId: z.string().optional().describe("The context the message is associated with"),
+  taskId: z.string().optional().describe("Identifier of task the message is related to"),
+  referenceTaskIds: z.array(z.string()).optional().describe("List of tasks referenced as context by this message"),
+  extensions: z.array(z.string()).optional().describe("The URIs of extensions that are present or contributed to this Message"),
+  metadata: z.record(z.unknown()).optional().describe("Extension metadata")
+});
+var messageSendParamsSchema = z.object({
+  message: messageSchema,
+  configuration: messageSendConfigurationSchema.optional(),
+  metadata: z.record(z.unknown()).optional().describe("Extension metadata")
+});
+var taskQueryParamsSchema = z.object({
+  id: z.string().describe("Task id"),
+  historyLength: z.number().optional().describe("Number of recent messages to be retrieved"),
+  metadata: z.record(z.unknown()).optional()
+});
+var taskIdParamsSchema = z.object({
+  id: z.string().describe("Task id"),
+  metadata: z.record(z.unknown()).optional()
+});
+z.object({
+  message: messageSchema,
+  metadata: z.record(z.any()).optional()
+});
+z.object({
+  id: z.string()
+});
+var agentExecutionParamsSchema = z.union([messageSendParamsSchema, taskQueryParamsSchema, taskIdParamsSchema]);
+var agentExecutionBodySchema = z.object({
+  jsonrpc: z.literal("2.0"),
+  id: z.union([z.string(), z.number()]),
+  method: z.enum(["message/send", "message/stream", "tasks/get", "tasks/cancel"]),
+  params: agentExecutionParamsSchema
+});
+var agentCardResponseSchema = z.object({
+  name: z.string(),
+  description: z.string(),
+  url: z.string(),
+  provider: z.object({
+    organization: z.string(),
+    url: z.string()
+  }).optional(),
+  version: z.string(),
+  capabilities: z.object({
+    streaming: z.boolean().optional(),
+    pushNotifications: z.boolean().optional(),
+    stateTransitionHistory: z.boolean().optional()
+  }),
+  defaultInputModes: z.array(z.string()),
+  defaultOutputModes: z.array(z.string()),
+  skills: z.array(
+    z.object({
+      id: z.string(),
+      name: z.string(),
+      description: z.string(),
+      tags: z.array(z.string()).optional()
+    })
+  )
+});
+z.unknown();
+var agentExecutionResponseSchema = z.unknown();
+
+// src/server/handlers/a2a.ts
+var messageSendParamsSchema2 = z$1.object({
+  message: z$1.object({
+    role: z$1.enum(["user", "agent"]),
+    parts: z$1.array(
+      z$1.object({
+        kind: z$1.enum(["text"]),
+        text: z$1.string()
+      })
+    ),
+    kind: z$1.literal("message"),
+    messageId: z$1.string(),
+    contextId: z$1.string().optional(),
+    taskId: z$1.string().optional(),
+    referenceTaskIds: z$1.array(z$1.string()).optional(),
+    extensions: z$1.array(z$1.string()).optional(),
+    metadata: z$1.record(z$1.any()).optional()
+  })
+});
+async function getAgentCardByIdHandler({
+  mastra,
+  agentId,
+  executionUrl = `/a2a/${agentId}`,
+  provider = {
+    organization: "Mastra",
+    url: "https://mastra.ai"
+  },
+  version = "1.0",
+  requestContext
+}) {
+  const agent = await getAgentFromSystem({ mastra, agentId });
+  const [instructions, tools] = await Promise.all([agent.getInstructions({ requestContext }), agent.listTools({ requestContext })]);
+  const agentCard = {
+    name: agent.id || agentId,
+    description: convertInstructionsToString(instructions),
+    url: executionUrl,
+    provider,
+    version,
+    capabilities: {
+      streaming: true,
+      // All agents support streaming
+      pushNotifications: false,
+      stateTransitionHistory: false
+    },
+    defaultInputModes: ["text"],
+    defaultOutputModes: ["text"],
+    // Convert agent tools to skills format for A2A protocol
+    skills: Object.entries(tools).map(([toolId, tool]) => ({
+      id: toolId,
+      name: toolId,
+      description: tool.description || `Tool: ${toolId}`,
+      // Optional fields
+      tags: ["tool"]
+    }))
+  };
+  return agentCard;
+}
+function validateMessageSendParams(params) {
+  try {
+    messageSendParamsSchema2.parse(params);
+  } catch (error) {
+    if (error instanceof z$1.ZodError) {
+      throw MastraA2AError.invalidParams(error.errors[0].message);
+    }
+    throw error;
+  }
+}
+async function handleMessageSend({
+  requestId,
+  params,
+  taskStore,
+  agent,
+  agentId,
+  logger,
+  requestContext
+}) {
+  validateMessageSendParams(params);
+  const { message, metadata } = params;
+  const { contextId } = message;
+  const taskId = message.taskId || crypto.randomUUID();
+  let currentData = await loadOrCreateTask({
+    taskId,
+    taskStore,
+    agentId,
+    message,
+    contextId,
+    metadata
+  });
+  createTaskContext({
+    task: currentData,
+    userMessage: message,
+    history: currentData.history || [],
+    activeCancellations: taskStore.activeCancellations
+  });
+  try {
+    const resourceId = metadata?.resourceId ?? message.metadata?.resourceId ?? agentId;
+    const result = await agent.generate([convertToCoreMessage(message)], {
+      runId: taskId,
+      requestContext,
+      ...contextId ? { threadId: contextId, resourceId } : {}
+    });
+    currentData = applyUpdateToTask(currentData, {
+      state: "completed",
+      message: {
+        messageId: crypto.randomUUID(),
+        role: "agent",
+        parts: [
+          {
+            kind: "text",
+            text: result.text
+          }
+        ],
+        kind: "message"
+      }
+    });
+    currentData.metadata = {
+      ...currentData.metadata,
+      execution: {
+        toolCalls: result.toolCalls,
+        toolResults: result.toolResults,
+        usage: result.usage,
+        finishReason: result.finishReason
+      }
+    };
+    await taskStore.save({ agentId, data: currentData });
+  } catch (handlerError) {
+    const failureStatusUpdate = {
+      state: "failed",
+      message: {
+        messageId: crypto.randomUUID(),
+        role: "agent",
+        parts: [
+          {
+            kind: "text",
+            text: `Handler failed: ${handlerError instanceof Error ? handlerError.message : String(handlerError)}`
+          }
+        ],
+        kind: "message"
+      }
+    };
+    currentData = applyUpdateToTask(currentData, failureStatusUpdate);
+    try {
+      await taskStore.save({ agentId, data: currentData });
+    } catch (saveError) {
+      logger?.error(`Failed to save task ${currentData.id} after handler error:`, saveError?.message);
+    }
+    return normalizeError(handlerError, requestId, currentData.id, logger);
+  }
+  return createSuccessResponse(requestId, currentData);
+}
+async function handleTaskGet({
+  requestId,
+  taskStore,
+  agentId,
+  taskId
+}) {
+  const task = await taskStore.load({ agentId, taskId });
+  if (!task) {
+    throw MastraA2AError.taskNotFound(taskId);
+  }
+  return createSuccessResponse(requestId, task);
+}
+async function* handleMessageStream({
+  requestId,
+  params,
+  taskStore,
+  agent,
+  agentId,
+  logger,
+  requestContext
+}) {
+  yield createSuccessResponse(requestId, {
+    state: "working",
+    message: {
+      messageId: crypto.randomUUID(),
+      kind: "message",
+      role: "agent",
+      parts: [{ kind: "text", text: "Generating response..." }]
+    }
+  });
+  let result;
+  try {
+    result = await handleMessageSend({
+      requestId,
+      params,
+      taskStore,
+      agent,
+      agentId,
+      requestContext,
+      logger
+    });
+  } catch (err) {
+    if (!(err instanceof MastraA2AError)) {
+      throw err;
+    }
+    result = createErrorResponse(requestId, err.toJSONRPCError());
+  }
+  yield result;
+}
+async function handleTaskCancel({
+  requestId,
+  taskStore,
+  agentId,
+  taskId,
+  logger
+}) {
+  let data = await taskStore.load({
+    agentId,
+    taskId
+  });
+  if (!data) {
+    throw MastraA2AError.taskNotFound(taskId);
+  }
+  const finalStates = ["completed", "failed", "canceled"];
+  if (finalStates.includes(data.status.state)) {
+    logger?.info(`Task ${taskId} already in final state ${data.status.state}, cannot cancel.`);
+    return createSuccessResponse(requestId, data);
+  }
+  taskStore.activeCancellations.add(taskId);
+  const cancelUpdate = {
+    state: "canceled",
+    message: {
+      role: "agent",
+      parts: [{ kind: "text", text: "Task cancelled by request." }],
+      kind: "message",
+      messageId: crypto.randomUUID()
+    }
+  };
+  data = applyUpdateToTask(data, cancelUpdate);
+  await taskStore.save({ agentId, data });
+  taskStore.activeCancellations.delete(taskId);
+  return createSuccessResponse(requestId, data);
+}
+async function getAgentExecutionHandler({
+  requestId,
+  mastra,
+  agentId,
+  requestContext,
+  method,
+  params,
+  taskStore,
+  logger
+}) {
+  const agent = await getAgentFromSystem({ mastra, agentId });
+  let taskId;
+  try {
+    taskId = "id" in params ? params.id : params.message?.taskId || "No task ID provided";
+    switch (method) {
+      case "message/send": {
+        const result2 = await handleMessageSend({
+          requestId,
+          params,
+          taskStore,
+          agent,
+          agentId,
+          requestContext
+        });
+        return result2;
+      }
+      case "message/stream":
+        const result = await handleMessageStream({
+          requestId,
+          taskStore,
+          params,
+          agent,
+          agentId,
+          requestContext
+        });
+        return result;
+      case "tasks/get": {
+        const result2 = await handleTaskGet({
+          requestId,
+          taskStore,
+          agentId,
+          taskId
+        });
+        return result2;
+      }
+      case "tasks/cancel": {
+        const result2 = await handleTaskCancel({
+          requestId,
+          taskStore,
+          agentId,
+          taskId
+        });
+        return result2;
+      }
+      default:
+        throw MastraA2AError.methodNotFound(method);
+    }
+  } catch (error) {
+    if (error instanceof MastraA2AError && taskId && !error.taskId) {
+      error.taskId = taskId;
+    }
+    return normalizeError(error, requestId, taskId, logger);
+  }
+}
+var GET_AGENT_CARD_ROUTE = createRoute({
+  method: "GET",
+  path: "/.well-known/:agentId/agent-card.json",
+  responseType: "json",
+  pathParamSchema: a2aAgentIdPathParams,
+  responseSchema: agentCardResponseSchema,
+  summary: "Get agent card",
+  description: "Returns the agent card information for A2A protocol discovery",
+  tags: ["Agent-to-Agent"],
+  requiresAuth: true,
+  handler: async ({ mastra, agentId, requestContext }) => {
+    const executionUrl = `/a2a/${agentId}`;
+    const provider = {
+      organization: "Mastra",
+      url: "https://mastra.ai"
+    };
+    const version = "1.0";
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    const [instructions, tools] = await Promise.all([agent.getInstructions({ requestContext }), agent.listTools({ requestContext })]);
+    const agentCard = {
+      name: agent.id || agentId,
+      description: convertInstructionsToString(instructions),
+      url: executionUrl,
+      provider,
+      version,
+      capabilities: {
+        streaming: true,
+        pushNotifications: false,
+        stateTransitionHistory: false
+      },
+      defaultInputModes: ["text"],
+      defaultOutputModes: ["text"],
+      skills: Object.entries(tools).map(([toolId, tool]) => ({
+        id: toolId,
+        name: toolId,
+        description: tool.description || `Tool: ${toolId}`,
+        tags: ["tool"]
+      }))
+    };
+    return agentCard;
+  }
+});
+var AGENT_EXECUTION_ROUTE = createRoute({
+  method: "POST",
+  path: "/a2a/:agentId",
+  responseType: "json",
+  pathParamSchema: a2aAgentIdPathParams,
+  bodySchema: agentExecutionBodySchema,
+  responseSchema: agentExecutionResponseSchema,
+  summary: "Execute agent",
+  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, ...bodyParams }) => {
+    const { id: requestId, method, params } = bodyParams;
+    return await getAgentExecutionHandler({
+      requestId: String(requestId),
+      mastra,
+      agentId,
+      requestContext,
+      method,
+      params,
+      taskStore
+    });
+  }
+});
+
+export { AGENT_EXECUTION_ROUTE, GET_AGENT_CARD_ROUTE, a2a_exports, getAgentCardByIdHandler, getAgentExecutionHandler, handleMessageSend, handleMessageStream, handleTaskCancel, handleTaskGet };
+//# sourceMappingURL=chunk-GUG6KRAJ.js.map
+//# sourceMappingURL=chunk-GUG6KRAJ.js.map