@mastra/server 0.0.0-message-ordering-20250415215612 → 0.0.0-new-scorer-api-20250801075530

package/dist/chunk-G7NVCO5M.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/server/handlers/vector.ts"],"names":[],"mappings":";;;;;AAAA,IAAA,cAAA,GAAA;AAAA,QAAA,CAAA,cAAA,EAAA;AAAA,EAAA,WAAA,EAAA,MAAA,WAAA;AAAA,EAAA,WAAA,EAAA,MAAA,WAAA;AAAA,EAAA,aAAA,EAAA,MAAA,aAAA;AAAA,EAAA,WAAA,EAAA,MAAA,WAAA;AAAA,EAAA,YAAA,EAAA,MAAA,YAAA;AAAA,EAAA,aAAA,EAAA,MAAA;AAAA,CAAA,CAAA;AA+BA,SAAS,SAAA,CAAU,QAA2B,UAAA,EAAmC;AAC/E,EAAA,IAAI,CAAC,UAAA,EAAY;AACf,IAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,2BAA2B,CAAA;AAAA,EACrE;AAEA,EAAA,MAAM,MAAA,GAAS,MAAA,CAAO,SAAA,CAAU,UAAU,CAAA;AAC1C,EAAA,IAAI,CAAC,MAAA,EAAQ;AACX,IAAA,MAAM,IAAI,cAAc,GAAA,EAAK,EAAE,SAAS,CAAA,aAAA,EAAgB,UAAU,cAAc,CAAA;AAAA,EAClF;AAEA,EAAA,OAAO,MAAA;AACT;AAGA,eAAsB,aAAA,CAAc,EAAE,MAAA,EAAQ,UAAA,EAAY,OAAM,EAA6C;AAC3G,EAAA,IAAI;AACF,IAAA,IAAI,CAAC,KAAA,EAAO,SAAA,IAAa,CAAC,KAAA,EAAO,OAAA,IAAW,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAA,CAAM,OAAO,CAAA,EAAG;AACzE,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,oEAAoE,CAAA;AAAA,IAC9G;AAEA,IAAA,MAAM,MAAA,GAAS,SAAA,CAAU,MAAA,EAAQ,UAAU,CAAA;AAC3C,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,MAAA,CAAO,KAAK,CAAA;AACxC,IAAA,OAAO,EAAE,KAAK,MAAA,EAAO;AAAA,EACvB,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,yBAAyB,CAAA;AAAA,EACrD;AACF;AAGA,eAAsB,WAAA,CAAY;AAAA,EAChC,MAAA;AAAA,EACA,UAAA;AAAA,EACA;AACF,CAAA,EAAiF;AAC/E,EAAA,IAAI;AACF,IAAA,MAAM,EAAE,SAAA,EAAW,SAAA,EAAW,MAAA,EAAO,GAAI,KAAA;AAEzC,IAAA,IAAI,CAAC,SAAA,IAAa,OAAO,SAAA,KAAc,QAAA,IAAY,aAAa,CAAA,EAAG;AACjE,MAAA,MAAM,IAAI,cAAc,GAAA,EAAK;AAAA,QAC3B,OAAA,EAAS;AAAA,OACV,CAAA;AAAA,IACH;AAEA,IAAA,IAAI,MAAA,IAAU,CAAC,CAAC,QAAA,EAAU,aAAa,YAAY,CAAA,CAAE,QAAA,CAAS,MAAM,CAAA,EAAG;AACrE,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,iEAAiE,CAAA;AAAA,IAC3G;AAEA,IAAA,MAAM,MAAA,GAAS,SAAA,CAAU,MAAA,EAAQ,UAAU,CAAA;AAC3C,IAAA,MAAM,OAAO,WAAA,CAAY,EAAE,SAAA,EAAW,SAAA,EAAW,QAAQ,CAAA;AACzD,IAAA,OAAO,EAAE,SAAS,IAAA,EAAK;AAAA,EACzB,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,sBAAsB,CAAA;AAAA,EAClD;AACF;AAGA,eAAsB,YAAA,CAAa;AAAA,EACjC,MAAA;AAAA,EACA,UAAA;AAAA,EACA;AACF,CAAA,EAA2E;AACzE,EAAA,IAAI;AACF,IAAA,IAAI,CAAC,KAAA,EAAO,SAAA,IAAa,CAAC,KAAA,EAAO,WAAA,IAAe,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAA,CAAM,WAAW,CAAA,EAAG;AACjF,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,wEAAwE,CAAA;AAAA,IAClH;AAEA,IAAA,MAAM,MAAA,GAAS,SAAA,CAAU,MAAA,EAAQ,UAAU,CAAA;AAC3C,IAAA,MAAM,OAAA,GAAyB,MAAM,MAAA,CAAO,KAAA,CAAM,KAAK,CAAA;AACvD,IAAA,OAAO,OAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,wBAAwB,CAAA;AAAA,EACpD;AACF;AAGA,eAAsB,WAAA,CAAY,EAAE,MAAA,EAAQ,UAAA,EAAW,EAAiD;AACtG,EAAA,IAAI;AACF,IAAA,MAAM,MAAA,GAAS,SAAA,CAAU,MAAA,EAAQ,UAAU,CAAA;AAE3C,IAAA,MAAM,OAAA,GAAU,MAAM,MAAA,CAAO,WAAA,EAAY;AACzC,IAAA,OAAO,OAAA,CAAQ,OAAO,OAAO,CAAA;AAAA,EAC/B,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,uBAAuB,CAAA;AAAA,EACnD;AACF;AAGA,eAAsB,aAAA,CAAc;AAAA,EAClC,MAAA;AAAA,EACA,UAAA;AAAA,EACA;AACF,CAAA,EAA0E;AACxE,EAAA,IAAI;AACF,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,0BAA0B,CAAA;AAAA,IACpE;AAEA,IAAA,MAAM,MAAA,GAAS,SAAA,CAAU,MAAA,EAAQ,UAAU,CAAA;AAC3C,IAAA,MAAM,QAAoB,MAAM,MAAA,CAAO,aAAA,CAAc,EAAE,WAAW,CAAA;AAElE,IAAA,OAAO;AAAA,MACL,WAAW,KAAA,CAAM,SAAA;AAAA,MACjB,OAAO,KAAA,CAAM,KAAA;AAAA,MACb,MAAA,EAAQ,KAAA,CAAM,MAAA,EAAQ,WAAA;AAAY,KACpC;AAAA,EACF,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,wBAAwB,CAAA;AAAA,EACpD;AACF;AAGA,eAAsB,WAAA,CAAY;AAAA,EAChC,MAAA;AAAA,EACA,UAAA;AAAA,EACA;AACF,CAAA,EAA0E;AACxE,EAAA,IAAI;AACF,IAAA,IAAI,CAAC,SAAA,EAAW;AACd,MAAA,MAAM,IAAI,aAAA,CAAc,GAAA,EAAK,EAAE,OAAA,EAAS,0BAA0B,CAAA;AAAA,IACpE;AAEA,IAAA,MAAM,MAAA,GAAS,SAAA,CAAU,MAAA,EAAQ,UAAU,CAAA;AAC3C,IAAA,MAAM,MAAA,CAAO,WAAA,CAAY,EAAE,SAAA,EAAW,CAAA;AACtC,IAAA,OAAO,EAAE,SAAS,IAAA,EAAK;AAAA,EACzB,SAAS,KAAA,EAAO;AACd,IAAA,OAAO,WAAA,CAAY,OAAO,sBAAsB,CAAA;AAAA,EAClD;AACF","file":"chunk-G7NVCO5M.js","sourcesContent":["import type { MastraVector, QueryResult, IndexStats } from '@mastra/core/vector';\nimport { HTTPException } from '../http-exception';\nimport type { Context } from '../types';\n\nimport { handleError } from './error';\n\ninterface VectorContext extends Context {\n  vectorName?: string;\n}\n\ninterface UpsertRequest {\n  indexName: string;\n  vectors: number[][];\n  metadata?: Record<string, any>[];\n  ids?: string[];\n}\n\ninterface CreateIndexRequest {\n  indexName: string;\n  dimension: number;\n  metric?: 'cosine' | 'euclidean' | 'dotproduct';\n}\n\ninterface QueryRequest {\n  indexName: string;\n  queryVector: number[];\n  topK?: number;\n  filter?: Record<string, any>;\n  includeVector?: boolean;\n}\n\nfunction getVector(mastra: Context['mastra'], vectorName?: string): MastraVector {\n  if (!vectorName) {\n    throw new HTTPException(400, { message: 'Vector name is required' });\n  }\n\n  const vector = mastra.getVector(vectorName);\n  if (!vector) {\n    throw new HTTPException(404, { message: `Vector store ${vectorName} not found` });\n  }\n\n  return vector;\n}\n\n// Upsert vectors\nexport async function upsertVectors({ mastra, vectorName, index }: VectorContext & { index: UpsertRequest }) {\n  try {\n    if (!index?.indexName || !index?.vectors || !Array.isArray(index.vectors)) {\n      throw new HTTPException(400, { message: 'Invalid request index. indexName and vectors array are required.' });\n    }\n\n    const vector = getVector(mastra, vectorName);\n    const result = await vector.upsert(index);\n    return { ids: result };\n  } catch (error) {\n    return handleError(error, 'Error upserting vectors');\n  }\n}\n\n// Create index\nexport async function createIndex({\n  mastra,\n  vectorName,\n  index,\n}: Pick<VectorContext, 'mastra' | 'vectorName'> & { index: CreateIndexRequest }) {\n  try {\n    const { indexName, dimension, metric } = index;\n\n    if (!indexName || typeof dimension !== 'number' || dimension <= 0) {\n      throw new HTTPException(400, {\n        message: 'Invalid request index, indexName and positive dimension number are required.',\n      });\n    }\n\n    if (metric && !['cosine', 'euclidean', 'dotproduct'].includes(metric)) {\n      throw new HTTPException(400, { message: 'Invalid metric. Must be one of: cosine, euclidean, dotproduct' });\n    }\n\n    const vector = getVector(mastra, vectorName);\n    await vector.createIndex({ indexName, dimension, metric });\n    return { success: true };\n  } catch (error) {\n    return handleError(error, 'Error creating index');\n  }\n}\n\n// Query vectors\nexport async function queryVectors({\n  mastra,\n  vectorName,\n  query,\n}: Pick<VectorContext, 'mastra' | 'vectorName'> & { query: QueryRequest }) {\n  try {\n    if (!query?.indexName || !query?.queryVector || !Array.isArray(query.queryVector)) {\n      throw new HTTPException(400, { message: 'Invalid request query. indexName and queryVector array are required.' });\n    }\n\n    const vector = getVector(mastra, vectorName);\n    const results: QueryResult[] = await vector.query(query);\n    return results;\n  } catch (error) {\n    return handleError(error, 'Error querying vectors');\n  }\n}\n\n// List indexes\nexport async function listIndexes({ mastra, vectorName }: Pick<VectorContext, 'mastra' | 'vectorName'>) {\n  try {\n    const vector = getVector(mastra, vectorName);\n\n    const indexes = await vector.listIndexes();\n    return indexes.filter(Boolean);\n  } catch (error) {\n    return handleError(error, 'Error listing indexes');\n  }\n}\n\n// Describe index\nexport async function describeIndex({\n  mastra,\n  vectorName,\n  indexName,\n}: Pick<VectorContext, 'mastra' | 'vectorName'> & { indexName?: string }) {\n  try {\n    if (!indexName) {\n      throw new HTTPException(400, { message: 'Index name is required' });\n    }\n\n    const vector = getVector(mastra, vectorName);\n    const stats: IndexStats = await vector.describeIndex({ indexName });\n\n    return {\n      dimension: stats.dimension,\n      count: stats.count,\n      metric: stats.metric?.toLowerCase(),\n    };\n  } catch (error) {\n    return handleError(error, 'Error describing index');\n  }\n}\n\n// Delete index\nexport async function deleteIndex({\n  mastra,\n  vectorName,\n  indexName,\n}: Pick<VectorContext, 'mastra' | 'vectorName'> & { indexName?: string }) {\n  try {\n    if (!indexName) {\n      throw new HTTPException(400, { message: 'Index name is required' });\n    }\n\n    const vector = getVector(mastra, vectorName);\n    await vector.deleteIndex({ indexName });\n    return { success: true };\n  } catch (error) {\n    return handleError(error, 'Error deleting index');\n  }\n}\n"]}

package/dist/chunk-GASWM5HJ.cjs

@@ -0,0 +1,514 @@
+'use strict';
+
+var chunkQ7SFCCGT_cjs = require('./chunk-Q7SFCCGT.cjs');
+var a2a = require('@mastra/core/a2a');
+var zod = require('zod');
+
+// src/server/handlers/a2a.ts
+var a2a_exports = {};
+chunkQ7SFCCGT_cjs.__export(a2a_exports, {
+  getAgentCardByIdHandler: () => getAgentCardByIdHandler,
+  getAgentExecutionHandler: () => getAgentExecutionHandler,
+  handleTaskCancel: () => handleTaskCancel,
+  handleTaskGet: () => handleTaskGet,
+  handleTaskSend: () => handleTaskSend,
+  handleTaskSendSubscribe: () => handleTaskSendSubscribe
+});
+function normalizeError(error, reqId, taskId, logger) {
+  let a2aError;
+  if (error instanceof a2a.A2AError) {
+    a2aError = error;
+  } else if (error instanceof Error) {
+    a2aError = a2a.A2AError.internalError(error.message, { stack: error.stack });
+  } else {
+    a2aError = a2a.A2AError.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 a2a.A2AError.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.type) {
+    case "text":
+      return {
+        type: "text",
+        text: part.text
+      };
+    case "file":
+      return {
+        type: "file",
+        data: new URL(part.file.uri),
+        mimeType: part.file.mimeType
+      };
+    case "data":
+      throw new Error("Data parts are not supported in core messages");
+  }
+}
+
+// src/server/a2a/store.ts
+var InMemoryTaskStore = class {
+  store = /* @__PURE__ */ new Map();
+  activeCancellations = /* @__PURE__ */ new Set();
+  async load({ agentId, taskId }) {
+    const entry = this.store.get(`${agentId}-${taskId}`);
+    if (!entry) {
+      return null;
+    }
+    return { task: { ...entry.task }, history: [...entry.history] };
+  }
+  async save({ agentId, data }) {
+    const key = `${agentId}-${data.task.id}`;
+    if (!data.task.id) {
+      throw new Error("Task ID is required");
+    }
+    this.store.set(key, {
+      task: { ...data.task },
+      history: [...data.history]
+    });
+  }
+};
+
+// src/server/a2a/tasks.ts
+function isTaskStatusUpdate(update) {
+  return "state" in update && !("parts" in update);
+}
+function isArtifactUpdate(update) {
+  return "parts" in update;
+}
+function applyUpdateToTaskAndHistory(current, update) {
+  let newTask = structuredClone(current.task);
+  let newHistory = structuredClone(current.history);
+  if (isTaskStatusUpdate(update)) {
+    newTask.status = {
+      ...newTask.status,
+      // Keep existing properties if not overwritten
+      ...update,
+      // Apply updates
+      timestamp: (/* @__PURE__ */ new Date()).toISOString()
+    };
+    if (update.message?.role === "agent") {
+      newHistory.push(update.message);
+    }
+  } else if (isArtifactUpdate(update)) {
+    if (!newTask.artifacts) {
+      newTask.artifacts = [];
+    } else {
+      newTask.artifacts = [...newTask.artifacts];
+    }
+    const existingIndex = update.index ?? -1;
+    let replaced = false;
+    if (existingIndex >= 0 && existingIndex < newTask.artifacts.length) {
+      const existingArtifact = newTask.artifacts[existingIndex];
+      if (update.append) {
+        const appendedArtifact = JSON.parse(JSON.stringify(existingArtifact));
+        appendedArtifact.parts.push(...update.parts);
+        if (update.metadata) {
+          appendedArtifact.metadata = {
+            ...appendedArtifact.metadata || {},
+            ...update.metadata
+          };
+        }
+        if (update.lastChunk !== void 0) appendedArtifact.lastChunk = update.lastChunk;
+        if (update.description) appendedArtifact.description = update.description;
+        newTask.artifacts[existingIndex] = appendedArtifact;
+        replaced = true;
+      } else {
+        newTask.artifacts[existingIndex] = { ...update };
+        replaced = true;
+      }
+    } else if (update.name) {
+      const namedIndex = newTask.artifacts.findIndex((a) => a.name === update.name);
+      if (namedIndex >= 0) {
+        newTask.artifacts[namedIndex] = { ...update };
+        replaced = true;
+      }
+    }
+    if (!replaced) {
+      newTask.artifacts.push({ ...update });
+      if (newTask.artifacts.some((a) => a.index !== void 0)) {
+        newTask.artifacts.sort((a, b) => (a.index ?? 0) - (b.index ?? 0));
+      }
+    }
+  }
+  return { task: newTask, history: newHistory };
+}
+async function loadOrCreateTaskAndHistory({
+  agentId,
+  taskId,
+  taskStore,
+  message,
+  sessionId,
+  metadata,
+  logger
+}) {
+  const data = await taskStore.load({ agentId, taskId });
+  if (!data) {
+    const initialTask = {
+      id: taskId,
+      sessionId,
+      status: {
+        state: "submitted",
+        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+        message: null
+      },
+      artifacts: [],
+      metadata
+    };
+    const initialData = {
+      task: initialTask,
+      history: [message]
+    };
+    logger?.info(`[Task ${taskId}] Created new task and history.`);
+    await taskStore.save({ agentId, data: initialData });
+    return initialData;
+  }
+  logger?.info(`[Task ${taskId}] Loaded existing task and history.`);
+  let updatedData = {
+    task: data.task,
+    history: [...data.history, message]
+  };
+  const { status } = data.task;
+  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 = applyUpdateToTaskAndHistory(updatedData, {
+      state: "submitted",
+      message: null
+    });
+  } else if (status.state === "input-required") {
+    logger?.info(`[Task ${taskId}] Changing state from 'input-required' to 'working'.`);
+    updatedData = applyUpdateToTaskAndHistory(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 {
+    task: { ...updatedData.task },
+    history: [...updatedData.history]
+  };
+}
+function createTaskContext({
+  task,
+  userMessage,
+  history,
+  activeCancellations
+}) {
+  return {
+    task: structuredClone(task),
+    userMessage,
+    history: structuredClone(history),
+    isCancelled: () => activeCancellations.has(task.id)
+  };
+}
+
+// src/server/handlers/a2a.ts
+var taskSendParamsSchema = zod.z.object({
+  id: zod.z.string().min(1, "Invalid or missing task ID (params.id)."),
+  message: zod.z.object({
+    parts: zod.z.array(
+      zod.z.object({
+        type: zod.z.enum(["text"]),
+        text: zod.z.string()
+      })
+    )
+  })
+});
+async function getAgentCardByIdHandler({
+  mastra,
+  agentId,
+  executionUrl = `/a2a/${agentId}`,
+  provider = {
+    organization: "Mastra",
+    url: "https://mastra.ai"
+  },
+  version = "1.0",
+  runtimeContext
+}) {
+  const agent = mastra.getAgent(agentId);
+  if (!agent) {
+    throw new Error(`Agent with ID ${agentId} not found`);
+  }
+  const [instructions, tools] = await Promise.all([
+    agent.getInstructions({ runtimeContext }),
+    agent.getTools({ runtimeContext })
+  ]);
+  const agentCard = {
+    name: agent.id || agentId,
+    description: 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 validateTaskSendParams(params) {
+  try {
+    taskSendParamsSchema.parse(params);
+  } catch (error) {
+    if (error instanceof zod.z.ZodError) {
+      throw a2a.A2AError.invalidParams(error.errors[0].message);
+    }
+    throw error;
+  }
+}
+async function handleTaskSend({
+  requestId,
+  params,
+  taskStore,
+  agent,
+  agentId,
+  logger,
+  runtimeContext
+}) {
+  validateTaskSendParams(params);
+  const { id: taskId, message, sessionId, metadata } = params;
+  let currentData = await loadOrCreateTaskAndHistory({
+    taskId,
+    taskStore,
+    agentId,
+    message,
+    sessionId,
+    metadata
+  });
+  createTaskContext({
+    task: currentData.task,
+    userMessage: message,
+    history: currentData.history,
+    activeCancellations: taskStore.activeCancellations
+  });
+  try {
+    const { text } = await agent.generate([convertToCoreMessage(message)], {
+      runId: taskId,
+      runtimeContext
+    });
+    currentData = applyUpdateToTaskAndHistory(currentData, {
+      state: "completed",
+      message: {
+        role: "agent",
+        parts: [
+          {
+            type: "text",
+            text
+          }
+        ]
+      }
+    });
+    await taskStore.save({ agentId, data: currentData });
+  } catch (handlerError) {
+    const failureStatusUpdate = {
+      state: "failed",
+      message: {
+        role: "agent",
+        parts: [
+          {
+            type: "text",
+            text: `Handler failed: ${handlerError instanceof Error ? handlerError.message : String(handlerError)}`
+          }
+        ]
+      }
+    };
+    currentData = applyUpdateToTaskAndHistory(currentData, failureStatusUpdate);
+    try {
+      await taskStore.save({ agentId, data: currentData });
+    } catch (saveError) {
+      logger?.error(`Failed to save task ${taskId} after handler error:`, saveError?.message);
+    }
+    return normalizeError(handlerError, requestId, taskId, logger);
+  }
+  return createSuccessResponse(requestId, currentData.task);
+}
+async function handleTaskGet({
+  requestId,
+  taskStore,
+  agentId,
+  taskId
+}) {
+  const task = await taskStore.load({ agentId, taskId });
+  if (!task) {
+    throw a2a.A2AError.taskNotFound(taskId);
+  }
+  return createSuccessResponse(requestId, task);
+}
+async function* handleTaskSendSubscribe({
+  requestId,
+  params,
+  taskStore,
+  agent,
+  agentId,
+  logger,
+  runtimeContext
+}) {
+  yield createSuccessResponse(requestId, {
+    state: "working",
+    message: {
+      role: "agent",
+      parts: [{ type: "text", text: "Generating response..." }]
+    }
+  });
+  let result;
+  try {
+    result = await handleTaskSend({
+      requestId,
+      params,
+      taskStore,
+      agent,
+      agentId,
+      runtimeContext,
+      logger
+    });
+  } catch (err) {
+    if (!(err instanceof a2a.A2AError)) {
+      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 a2a.A2AError.taskNotFound(taskId);
+  }
+  const finalStates = ["completed", "failed", "canceled"];
+  if (finalStates.includes(data.task.status.state)) {
+    logger?.info(`Task ${taskId} already in final state ${data.task.status.state}, cannot cancel.`);
+    return createSuccessResponse(requestId, data.task);
+  }
+  taskStore.activeCancellations.add(taskId);
+  const cancelUpdate = {
+    state: "canceled",
+    message: {
+      role: "agent",
+      parts: [{ type: "text", text: "Task cancelled by request." }]
+    }
+  };
+  data = applyUpdateToTaskAndHistory(data, cancelUpdate);
+  await taskStore.save({ agentId, data });
+  taskStore.activeCancellations.delete(taskId);
+  return createSuccessResponse(requestId, data.task);
+}
+async function getAgentExecutionHandler({
+  requestId,
+  mastra,
+  agentId,
+  runtimeContext,
+  method,
+  params,
+  taskStore = new InMemoryTaskStore(),
+  logger
+}) {
+  const agent = mastra.getAgent(agentId);
+  let taskId;
+  try {
+    taskId = params.id;
+    switch (method) {
+      case "tasks/send": {
+        const result2 = await handleTaskSend({
+          requestId,
+          params,
+          taskStore,
+          agent,
+          agentId,
+          runtimeContext
+        });
+        return result2;
+      }
+      case "tasks/sendSubscribe":
+        const result = await handleTaskSendSubscribe({
+          requestId,
+          taskStore,
+          params,
+          agent,
+          agentId,
+          runtimeContext
+        });
+        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 a2a.A2AError.methodNotFound(method);
+    }
+  } catch (error) {
+    if (error instanceof a2a.A2AError && taskId && !error.taskId) {
+      error.taskId = taskId;
+    }
+    return normalizeError(error, requestId, taskId, logger);
+  }
+}
+
+exports.a2a_exports = a2a_exports;
+exports.getAgentCardByIdHandler = getAgentCardByIdHandler;
+exports.getAgentExecutionHandler = getAgentExecutionHandler;
+exports.handleTaskCancel = handleTaskCancel;
+exports.handleTaskGet = handleTaskGet;
+exports.handleTaskSend = handleTaskSend;
+exports.handleTaskSendSubscribe = handleTaskSendSubscribe;
+//# sourceMappingURL=chunk-GASWM5HJ.cjs.map
+//# sourceMappingURL=chunk-GASWM5HJ.cjs.map

package/dist/chunk-GASWM5HJ.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/server/handlers/a2a.ts","../src/server/a2a/protocol.ts","../src/server/a2a/store.ts","../src/server/a2a/tasks.ts"],"names":["__export","A2AError","z","result"],"mappings":";;;;;;;AAAA,IAAA,WAAA,GAAA;AAAAA,0BAAA,CAAA,WAAA,EAAA;AAAA,EAAA,uBAAA,EAAA,MAAA,uBAAA;AAAA,EAAA,wBAAA,EAAA,MAAA,wBAAA;AAAA,EAAA,gBAAA,EAAA,MAAA,gBAAA;AAAA,EAAA,aAAA,EAAA,MAAA,aAAA;AAAA,EAAA,cAAA,EAAA,MAAA,cAAA;AAAA,EAAA,uBAAA,EAAA,MAAA;AAAA,CAAA,CAAA;ACMO,SAAS,cAAA,CACd,KAAA,EACA,KAAA,EACA,MAAA,EACA,MAAA,EACgC;AAChC,EAAA,IAAI,QAAA;AACJ,EAAA,IAAI,iBAAiBC,YAAA,EAAU;AAC7B,IAAA,QAAA,GAAW,KAAA;AAAA,EACb,CAAA,MAAA,IAAW,iBAAiB,KAAA,EAAO;AAEjC,IAAA,QAAA,GAAWA,YAAA,CAAS,cAAc,KAAA,CAAM,OAAA,EAAS,EAAE,KAAA,EAAO,KAAA,CAAM,OAAO,CAAA;AAAA,EACzE,CAAA,MAAO;AAEL,IAAA,QAAA,GAAWA,YAAA,CAAS,aAAA,CAAc,4BAAA,EAA8B,KAAK,CAAA;AAAA,EACvE;AAGA,EAAA,IAAI,MAAA,IAAU,CAAC,QAAA,CAAS,MAAA,EAAQ;AAC9B,IAAA,QAAA,CAAS,MAAA,GAAS,MAAA;AAAA,EACpB;AAEA,EAAA,MAAA,EAAQ,KAAA,CAAM,mCAAmC,QAAA,CAAS,MAAA,IAAU,KAAK,CAAA,SAAA,EAAY,KAAA,IAAS,KAAK,CAAA,EAAA,CAAA,EAAM,QAAQ,CAAA;AAEjH,EAAA,OAAO,mBAAA,CAAoB,KAAA,EAAO,QAAA,CAAS,cAAA,EAAgB,CAAA;AAC7D;AAEO,SAAS,mBAAA,CACd,IACA,KAAA,EACgC;AAEhC,EAAA,OAAO;AAAA,IACL,OAAA,EAAS,KAAA;AAAA,IACT,EAAA;AAAA;AAAA,IACA;AAAA,GACF;AACF;AAEO,SAAS,qBAAA,CAAyB,IAA4B,MAAA,EAA+B;AAClG,EAAA,IAAI,CAAC,EAAA,EAAI;AAEP,IAAA,MAAMA,YAAA,CAAS,cAAc,6CAA6C,CAAA;AAAA,EAC5E;AAEA,EAAA,OAAO;AAAA,IACL,OAAA,EAAS,KAAA;AAAA,IACT,EAAA;AAAA,IACA;AAAA,GACF;AACF;AAEO,SAAS,qBAAqB,OAAA,EAA+B;AAClE,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,OAAA,CAAQ,IAAA,KAAS,MAAA,GAAS,MAAA,GAAS,WAAA;AAAA,IACzC,SAAS,OAAA,CAAQ,KAAA,CAAM,IAAI,CAAA,GAAA,KAAO,wBAAA,CAAyB,GAAG,CAAC;AAAA,GACjE;AACF;AAEA,SAAS,yBAAyB,IAAA,EAAY;AAC5C,EAAA,QAAQ,KAAK,IAAA;AAAM,IACjB,KAAK,MAAA;AACH,MAAA,OAAO;AAAA,QACL,IAAA,EAAM,MAAA;AAAA,QACN,MAAM,IAAA,CAAK;AAAA,OACb;AAAA,IACF,KAAK,MAAA;AACH,MAAA,OAAO;AAAA,QACL,IAAA,EAAM,MAAA;AAAA,QACN,IAAA,EAAM,IAAI,GAAA,CAAI,IAAA,CAAK,KAAK,GAAI,CAAA;AAAA,QAC5B,QAAA,EAAU,KAAK,IAAA,CAAK;AAAA,OACtB;AAAA,IACF,KAAK,MAAA;AACH,MAAA,MAAM,IAAI,MAAM,+CAA+C,CAAA;AAAA;AAErE;;;AC/EO,IAAM,oBAAN,MAAwB;AAAA,EACrB,KAAA,uBAAyC,GAAA,EAAI;AAAA,EAC9C,mBAAA,uBAA0B,GAAA,EAAY;AAAA,EAE7C,MAAM,IAAA,CAAK,EAAE,OAAA,EAAS,QAAO,EAAwE;AACnG,IAAA,MAAM,KAAA,GAAQ,KAAK,KAAA,CAAM,GAAA,CAAI,GAAG,OAAO,CAAA,CAAA,EAAI,MAAM,CAAA,CAAE,CAAA;AAEnD,IAAA,IAAI,CAAC,KAAA,EAAO;AACV,MAAA,OAAO,IAAA;AAAA,IACT;AAGA,IAAA,OAAO,EAAE,IAAA,EAAM,EAAE,GAAG,KAAA,CAAM,IAAA,EAAK,EAAG,OAAA,EAAS,CAAC,GAAG,KAAA,CAAM,OAAO,CAAA,EAAE;AAAA,EAChE;AAAA,EAEA,MAAM,IAAA,CAAK,EAAE,OAAA,EAAS,MAAK,EAA6D;AAEtF,IAAA,MAAM,MAAM,CAAA,EAAG,OAAO,CAAA,CAAA,EAAI,IAAA,CAAK,KAAK,EAAE,CAAA,CAAA;AACtC,IAAA,IAAI,CAAC,IAAA,CAAK,IAAA,CAAK,EAAA,EAAI;AACjB,MAAA,MAAM,IAAI,MAAM,qBAAqB,CAAA;AAAA,IACvC;AACA,IAAA,IAAA,CAAK,KAAA,CAAM,IAAI,GAAA,EAAK;AAAA,MAClB,IAAA,EAAM,EAAE,GAAG,IAAA,CAAK,IAAA,EAAK;AAAA,MACrB,OAAA,EAAS,CAAC,GAAG,IAAA,CAAK,OAAO;AAAA,KAC1B,CAAA;AAAA,EACH;AACF,CAAA;;;ACxBA,SAAS,mBAAmB,MAAA,EAAwE;AAClG,EAAA,OAAO,OAAA,IAAW,MAAA,IAAU,EAAE,OAAA,IAAW,MAAA,CAAA;AAC3C;AAEA,SAAS,iBAAiB,MAAA,EAAmD;AAC3E,EAAA,OAAO,OAAA,IAAW,MAAA;AACpB;AAEO,SAAS,2BAAA,CACd,SACA,MAAA,EACgB;AAChB,EAAA,IAAI,OAAA,GAAU,eAAA,CAAgB,OAAA,CAAQ,IAAI,CAAA;AAC1C,EAAA,IAAI,UAAA,GAAa,eAAA,CAAgB,OAAA,CAAQ,OAAO,CAAA;AAEhD,EAAA,IAAI,kBAAA,CAAmB,MAAM,CAAA,EAAG;AAE9B,IAAA,OAAA,CAAQ,MAAA,GAAS;AAAA,MACf,GAAG,OAAA,CAAQ,MAAA;AAAA;AAAA,MACX,GAAG,MAAA;AAAA;AAAA,MACH,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA;AAAY,KACpC;AAGA,IAAA,IAAI,MAAA,CAAO,OAAA,EAAS,IAAA,KAAS,OAAA,EAAS;AACpC,MAAA,UAAA,CAAW,IAAA,CAAK,OAAO,OAAO,CAAA;AAAA,IAChC;AAAA,EACF,CAAA,MAAA,IAAW,gBAAA,CAAiB,MAAM,CAAA,EAAG;AAEnC,IAAA,IAAI,CAAC,QAAQ,SAAA,EAAW;AACtB,MAAA,OAAA,CAAQ,YAAY,EAAC;AAAA,IACvB,CAAA,MAAO;AAEL,MAAA,OAAA,CAAQ,SAAA,GAAY,CAAC,GAAG,OAAA,CAAQ,SAAS,CAAA;AAAA,IAC3C;AAEA,IAAA,MAAM,aAAA,GAAgB,OAAO,KAAA,IAAS,EAAA;AACtC,IAAA,IAAI,QAAA,GAAW,KAAA;AAEf,IAAA,IAAI,aAAA,IAAiB,CAAA,IAAK,aAAA,GAAgB,OAAA,CAAQ,UAAU,MAAA,EAAQ;AAClE,MAAA,MAAM,gBAAA,GAAmB,OAAA,CAAQ,SAAA,CAAU,aAAa,CAAA;AACxD,MAAA,IAAI,OAAO,MAAA,EAAQ;AAEjB,QAAA,MAAM,mBAAmB,IAAA,CAAK,KAAA,CAAM,IAAA,CAAK,SAAA,CAAU,gBAAgB,CAAC,CAAA;AACpE,QAAA,gBAAA,CAAiB,KAAA,CAAM,IAAA,CAAK,GAAG,MAAA,CAAO,KAAK,CAAA;AAC3C,QAAA,IAAI,OAAO,QAAA,EAAU;AACnB,UAAA,gBAAA,CAAiB,QAAA,GAAW;AAAA,YAC1B,GAAI,gBAAA,CAAiB,QAAA,IAAY,EAAC;AAAA,YAClC,GAAG,MAAA,CAAO;AAAA,WACZ;AAAA,QACF;AACA,QAAA,IAAI,MAAA,CAAO,SAAA,KAAc,MAAA,EAAW,gBAAA,CAAiB,YAAY,MAAA,CAAO,SAAA;AACxE,QAAA,IAAI,MAAA,CAAO,WAAA,EAAa,gBAAA,CAAiB,WAAA,GAAc,MAAA,CAAO,WAAA;AAC9D,QAAA,OAAA,CAAQ,SAAA,CAAU,aAAa,CAAA,GAAI,gBAAA;AACnC,QAAA,QAAA,GAAW,IAAA;AAAA,MACb,CAAA,MAAO;AAEL,QAAA,OAAA,CAAQ,SAAA,CAAU,aAAa,CAAA,GAAI,EAAE,GAAG,MAAA,EAAO;AAC/C,QAAA,QAAA,GAAW,IAAA;AAAA,MACb;AAAA,IACF,CAAA,MAAA,IAAW,OAAO,IAAA,EAAM;AACtB,MAAA,MAAM,UAAA,GAAa,QAAQ,SAAA,CAAU,SAAA,CAAU,OAAK,CAAA,CAAE,IAAA,KAAS,OAAO,IAAI,CAAA;AAC1E,MAAA,IAAI,cAAc,CAAA,EAAG;AACnB,QAAA,OAAA,CAAQ,SAAA,CAAU,UAAU,CAAA,GAAI,EAAE,GAAG,MAAA,EAAO;AAC5C,QAAA,QAAA,GAAW,IAAA;AAAA,MACb;AAAA,IACF;AAEA,IAAA,IAAI,CAAC,QAAA,EAAU;AACb,MAAA,OAAA,CAAQ,SAAA,CAAU,IAAA,CAAK,EAAE,GAAG,QAAQ,CAAA;AAEpC,MAAA,IAAI,QAAQ,SAAA,CAAU,IAAA,CAAK,OAAK,CAAA,CAAE,KAAA,KAAU,MAAS,CAAA,EAAG;AACtD,QAAA,OAAA,CAAQ,SAAA,CAAU,IAAA,CAAK,CAAC,CAAA,EAAG,CAAA,KAAA,CAAO,EAAE,KAAA,IAAS,CAAA,KAAM,CAAA,CAAE,KAAA,IAAS,CAAA,CAAE,CAAA;AAAA,MAClE;AAAA,IACF;AAAA,EACF;AAEA,EAAA,OAAO,EAAE,IAAA,EAAM,OAAA,EAAS,OAAA,EAAS,UAAA,EAAW;AAC9C;AAEA,eAAsB,0BAAA,CAA2B;AAAA,EAC/C,OAAA;AAAA,EACA,MAAA;AAAA,EACA,SAAA;AAAA,EACA,OAAA;AAAA,EACA,SAAA;AAAA,EACA,QAAA;AAAA,EACA;AACF,CAAA,EAQ4B;AAC1B,EAAA,MAAM,OAAO,MAAM,SAAA,CAAU,KAAK,EAAE,OAAA,EAAS,QAAQ,CAAA;AAGrD,EAAA,IAAI,CAAC,IAAA,EAAM;AACT,IAAA,MAAM,WAAA,GAAoB;AAAA,MACxB,EAAA,EAAI,MAAA;AAAA,MACJ,SAAA;AAAA,MACA,MAAA,EAAQ;AAAA,QACN,KAAA,EAAO,WAAA;AAAA,QACP,SAAA,EAAA,iBAAW,IAAI,IAAA,EAAK,EAAE,WAAA,EAAY;AAAA,QAClC,OAAA,EAAS;AAAA,OACX;AAAA,MACA,WAAW,EAAC;AAAA,MACZ;AAAA,KACF;AAEA,IAAA,MAAM,WAAA,GAAc;AAAA,MAClB,IAAA,EAAM,WAAA;AAAA,MACN,OAAA,EAAS,CAAC,OAAO;AAAA,KACnB;AAEA,IAAA,MAAA,EAAQ,IAAA,CAAK,CAAA,MAAA,EAAS,MAAM,CAAA,+BAAA,CAAiC,CAAA;AAC7D,IAAA,MAAM,UAAU,IAAA,CAAK,EAAE,OAAA,EAAS,IAAA,EAAM,aAAa,CAAA;AAEnD,IAAA,OAAO,WAAA;AAAA,EACT;AAGA,EAAA,MAAA,EAAQ,IAAA,CAAK,CAAA,MAAA,EAAS,MAAM,CAAA,mCAAA,CAAqC,CAAA;AAGjE,EAAA,IAAI,WAAA,GAAc;AAAA,IAChB,MAAM,IAAA,CAAK,IAAA;AAAA,IACX,OAAA,EAAS,CAAC,GAAG,IAAA,CAAK,SAAS,OAAO;AAAA,GACpC;AAGA,EAAA,MAAM,EAAE,MAAA,EAAO,GAAI,IAAA,CAAK,IAAA;AACxB,EAAA,MAAM,WAAA,GAA2B,CAAC,WAAA,EAAa,QAAA,EAAU,UAAU,CAAA;AAEnE,EAAA,IAAI,WAAA,CAAY,QAAA,CAAS,MAAA,CAAO,KAAK,CAAA,EAAG;AACtC,IAAA,MAAA,EAAQ,KAAK,CAAA,MAAA,EAAS,MAAM,CAAA,2CAAA,EAA8C,MAAA,CAAO,KAAK,CAAA,aAAA,CAAe,CAAA;AACrG,IAAA,WAAA,GAAc,4BAA4B,WAAA,EAAa;AAAA,MACrD,KAAA,EAAO,WAAA;AAAA,MACP,OAAA,EAAS;AAAA,KACV,CAAA;AAAA,EACH,CAAA,MAAA,IAAW,MAAA,CAAO,KAAA,KAAU,gBAAA,EAAkB;AAC5C,IAAA,MAAA,EAAQ,IAAA,CAAK,CAAA,MAAA,EAAS,MAAM,CAAA,oDAAA,CAAsD,CAAA;AAClF,IAAA,WAAA,GAAc,2BAAA,CAA4B,WAAA,EAAa,EAAE,KAAA,EAAO,WAAW,CAAA;AAAA,EAC7E,CAAA,MAAA,IAAW,MAAA,CAAO,KAAA,KAAU,SAAA,EAAW;AACrC,IAAA,MAAA,EAAQ,IAAA,CAAK,CAAA,MAAA,EAAS,MAAM,CAAA,uDAAA,CAAyD,CAAA;AAAA,EACvF;AAEA,EAAA,MAAM,UAAU,IAAA,CAAK,EAAE,OAAA,EAAS,IAAA,EAAM,aAAa,CAAA;AAEnD,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,EAAE,GAAG,WAAA,CAAY,IAAA,EAAK;AAAA,IAC5B,OAAA,EAAS,CAAC,GAAG,WAAA,CAAY,OAAO;AAAA,GAClC;AACF;AAEO,SAAS,iBAAA,CAAkB;AAAA,EAChC,IAAA;AAAA,EACA,WAAA;AAAA,EACA,OAAA;AAAA,EACA;AACF,CAAA,EAKgB;AACd,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,gBAAgB,IAAI,CAAA;AAAA,IAC1B,WAAA;AAAA,IACA,OAAA,EAAS,gBAAgB,OAAO,CAAA;AAAA,IAChC,WAAA,EAAa,MAAM,mBAAA,CAAoB,GAAA,CAAI,KAAK,EAAE;AAAA,GACpD;AACF;;;AHxKA,IAAM,oBAAA,GAAuBC,MAAE,MAAA,CAAO;AAAA,EACpC,IAAIA,KAAA,CAAE,MAAA,EAAO,CAAE,GAAA,CAAI,GAAG,yCAAyC,CAAA;AAAA,EAC/D,OAAA,EAASA,MAAE,MAAA,CAAO;AAAA,IAChB,OAAOA,KAAA,CAAE,KAAA;AAAA,MACPA,MAAE,MAAA,CAAO;AAAA,QACP,IAAA,EAAMA,KAAA,CAAE,IAAA,CAAK,CAAC,MAAM,CAAC,CAAA;AAAA,QACrB,IAAA,EAAMA,MAAE,MAAA;AAAO,OAChB;AAAA;AACH,GACD;AACH,CAAC,CAAA;AAED,eAAsB,uBAAA,CAAwB;AAAA,EAC5C,MAAA;AAAA,EACA,OAAA;AAAA,EACA,YAAA,GAAe,QAAQ,OAAO,CAAA,CAAA;AAAA,EAC9B,QAAA,GAAW;AAAA,IACT,YAAA,EAAc,QAAA;AAAA,IACd,GAAA,EAAK;AAAA,GACP;AAAA,EACA,OAAA,GAAU,KAAA;AAAA,EACV;AACF,CAAA,EASuB;AACrB,EAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AAErC,EAAA,IAAI,CAAC,KAAA,EAAO;AACV,IAAA,MAAM,IAAI,KAAA,CAAM,CAAA,cAAA,EAAiB,OAAO,CAAA,UAAA,CAAY,CAAA;AAAA,EACtD;AAEA,EAAA,MAAM,CAAC,YAAA,EAAc,KAAK,CAAA,GAAI,MAAM,QAAQ,GAAA,CAAI;AAAA,IAC9C,KAAA,CAAM,eAAA,CAAgB,EAAE,cAAA,EAAgB,CAAA;AAAA,IACxC,KAAA,CAAM,QAAA,CAAS,EAAE,cAAA,EAAgB;AAAA,GAClC,CAAA;AAGD,EAAA,MAAM,SAAA,GAAuB;AAAA,IAC3B,IAAA,EAAM,MAAM,EAAA,IAAM,OAAA;AAAA,IAClB,WAAA,EAAa,YAAA;AAAA,IACb,GAAA,EAAK,YAAA;AAAA,IACL,QAAA;AAAA,IACA,OAAA;AAAA,IACA,YAAA,EAAc;AAAA,MACZ,SAAA,EAAW,IAAA;AAAA;AAAA,MACX,iBAAA,EAAmB,KAAA;AAAA,MACnB,sBAAA,EAAwB;AAAA,KAC1B;AAAA,IACA,iBAAA,EAAmB,CAAC,MAAM,CAAA;AAAA,IAC1B,kBAAA,EAAoB,CAAC,MAAM,CAAA;AAAA;AAAA,IAE3B,MAAA,EAAQ,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,CAAE,IAAI,CAAC,CAAC,MAAA,EAAQ,IAAI,CAAA,MAAO;AAAA,MACrD,EAAA,EAAI,MAAA;AAAA,MACJ,IAAA,EAAM,MAAA;AAAA,MACN,WAAA,EAAa,IAAA,CAAK,WAAA,IAAe,CAAA,MAAA,EAAS,MAAM,CAAA,CAAA;AAAA;AAAA,MAEhD,IAAA,EAAM,CAAC,MAAM;AAAA,KACf,CAAE;AAAA,GACJ;AAEA,EAAA,OAAO,SAAA;AACT;AAEA,SAAS,uBAAuB,MAAA,EAAwB;AACtD,EAAA,IAAI;AACF,IAAA,oBAAA,CAAqB,MAAM,MAAM,CAAA;AAAA,EACnC,SAAS,KAAA,EAAO;AACd,IAAA,IAAI,KAAA,YAAiBA,MAAE,QAAA,EAAU;AAC/B,MAAA,MAAMD,aAAS,aAAA,CAAe,KAAA,CAAqB,MAAA,CAAO,CAAC,EAAG,OAAO,CAAA;AAAA,IACvE;AAEA,IAAA,MAAM,KAAA;AAAA,EACR;AACF;AAEA,eAAsB,cAAA,CAAe;AAAA,EACnC,SAAA;AAAA,EACA,MAAA;AAAA,EACA,SAAA;AAAA,EACA,KAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAQG;AACD,EAAA,sBAAA,CAAuB,MAAM,CAAA;AAE7B,EAAA,MAAM,EAAE,EAAA,EAAI,MAAA,EAAQ,OAAA,EAAS,SAAA,EAAW,UAAS,GAAI,MAAA;AAGrD,EAAA,IAAI,WAAA,GAAc,MAAM,0BAAA,CAA2B;AAAA,IACjD,MAAA;AAAA,IACA,SAAA;AAAA,IACA,OAAA;AAAA,IACA,OAAA;AAAA,IACA,SAAA;AAAA,IACA;AAAA,GACD,CAAA;AAGD,EAAgB,iBAAA,CAAkB;AAAA,IAChC,MAAM,WAAA,CAAY,IAAA;AAAA,IAClB,WAAA,EAAa,OAAA;AAAA,IACb,SAAS,WAAA,CAAY,OAAA;AAAA,IACrB,qBAAqB,SAAA,CAAU;AAAA,GAChC;AAED,EAAA,IAAI;AACF,IAAA,MAAM,EAAE,IAAA,EAAK,GAAI,MAAM,KAAA,CAAM,SAAS,CAAC,oBAAA,CAAqB,OAAO,CAAC,CAAA,EAAG;AAAA,MACrE,KAAA,EAAO,MAAA;AAAA,MACP;AAAA,KACD,CAAA;AAED,IAAA,WAAA,GAAc,4BAA4B,WAAA,EAAa;AAAA,MACrD,KAAA,EAAO,WAAA;AAAA,MACP,OAAA,EAAS;AAAA,QACP,IAAA,EAAM,OAAA;AAAA,QACN,KAAA,EAAO;AAAA,UACL;AAAA,YACE,IAAA,EAAM,MAAA;AAAA,YACN;AAAA;AACF;AACF;AACF,KACD,CAAA;AAED,IAAA,MAAM,UAAU,IAAA,CAAK,EAAE,OAAA,EAAS,IAAA,EAAM,aAAa,CAAA;AACxB,EAC7B,SAAS,YAAA,EAAc;AAErB,IAAA,MAAM,mBAAA,GAAqD;AAAA,MACzD,KAAA,EAAO,QAAA;AAAA,MACP,OAAA,EAAS;AAAA,QACP,IAAA,EAAM,OAAA;AAAA,QACN,KAAA,EAAO;AAAA,UACL;AAAA,YACE,IAAA,EAAM,MAAA;AAAA,YACN,IAAA,EAAM,mBAAmB,YAAA,YAAwB,KAAA,GAAQ,aAAa,OAAA,GAAU,MAAA,CAAO,YAAY,CAAC,CAAA;AAAA;AACtG;AACF;AACF,KACF;AACA,IAAA,WAAA,GAAc,2BAAA,CAA4B,aAAa,mBAAmB,CAAA;AAE1E,IAAA,IAAI;AACF,MAAA,MAAM,UAAU,IAAA,CAAK,EAAE,OAAA,EAAS,IAAA,EAAM,aAAa,CAAA;AAAA,IACrD,SAAS,SAAA,EAAW;AAElB,MAAA,MAAA,EAAQ,KAAA,CAAM,CAAA,oBAAA,EAAuB,MAAM,CAAA,qBAAA,CAAA,EAAyB,WAAW,OAAO,CAAA;AAAA,IACxF;AAEA,IAAA,OAAO,cAAA,CAAe,YAAA,EAAc,SAAA,EAAW,MAAA,EAAQ,MAAM,CAAA;AAAA,EAC/D;AAGA,EAAA,OAAO,qBAAA,CAAsB,SAAA,EAAW,WAAA,CAAY,IAAI,CAAA;AAC1D;AAEA,eAAsB,aAAA,CAAc;AAAA,EAClC,SAAA;AAAA,EACA,SAAA;AAAA,EACA,OAAA;AAAA,EACA;AACF,CAAA,EAKG;AACD,EAAA,MAAM,OAAO,MAAM,SAAA,CAAU,KAAK,EAAE,OAAA,EAAS,QAAQ,CAAA;AAErD,EAAA,IAAI,CAAC,IAAA,EAAM;AACT,IAAA,MAAMA,YAAAA,CAAS,aAAa,MAAM,CAAA;AAAA,EACpC;AAEA,EAAA,OAAO,qBAAA,CAAsB,WAAW,IAAI,CAAA;AAC9C;AAEA,gBAAuB,uBAAA,CAAwB;AAAA,EAC7C,SAAA;AAAA,EACA,MAAA;AAAA,EACA,SAAA;AAAA,EACA,KAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAQG;AACD,EAAA,MAAM,sBAAsB,SAAA,EAAW;AAAA,IACrC,KAAA,EAAO,SAAA;AAAA,IACP,OAAA,EAAS;AAAA,MACP,IAAA,EAAM,OAAA;AAAA,MACN,OAAO,CAAC,EAAE,MAAM,MAAA,EAAQ,IAAA,EAAM,0BAA0B;AAAA;AAC1D,GACD,CAAA;AAED,EAAA,IAAI,MAAA;AACJ,EAAA,IAAI;AACF,IAAA,MAAA,GAAS,MAAM,cAAA,CAAe;AAAA,MAC5B,SAAA;AAAA,MACA,MAAA;AAAA,MACA,SAAA;AAAA,MACA,KAAA;AAAA,MACA,OAAA;AAAA,MACA,cAAA;AAAA,MACA;AAAA,KACD,CAAA;AAAA,EACH,SAAS,GAAA,EAAK;AACZ,IAAA,IAAI,EAAE,eAAeA,YAAAA,CAAAA,EAAW;AAC9B,MAAA,MAAM,GAAA;AAAA,IACR;AAEA,IAAA,MAAA,GAAS,mBAAA,CAAoB,SAAA,EAAW,GAAA,CAAI,cAAA,EAAgB,CAAA;AAAA,EAC9D;AAEA,EAAA,MAAM,MAAA;AACR;AAEA,eAAsB,gBAAA,CAAiB;AAAA,EACrC,SAAA;AAAA,EACA,SAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAMG;AAED,EAAA,IAAI,IAAA,GAAO,MAAM,SAAA,CAAU,IAAA,CAAK;AAAA,IAC9B,OAAA;AAAA,IACA;AAAA,GACD,CAAA;AAED,EAAA,IAAI,CAAC,IAAA,EAAM;AACT,IAAA,MAAMA,YAAAA,CAAS,aAAa,MAAM,CAAA;AAAA,EACpC;AAGA,EAAA,MAAM,WAAA,GAA2B,CAAC,WAAA,EAAa,QAAA,EAAU,UAAU,CAAA;AAEnE,EAAA,IAAI,YAAY,QAAA,CAAS,IAAA,CAAK,IAAA,CAAK,MAAA,CAAO,KAAK,CAAA,EAAG;AAChD,IAAA,MAAA,EAAQ,IAAA,CAAK,QAAQ,MAAM,CAAA,wBAAA,EAA2B,KAAK,IAAA,CAAK,MAAA,CAAO,KAAK,CAAA,gBAAA,CAAkB,CAAA;AAC9F,IAAA,OAAO,qBAAA,CAAsB,SAAA,EAAW,IAAA,CAAK,IAAI,CAAA;AAAA,EACnD;AAGA,EAAA,SAAA,CAAU,mBAAA,CAAoB,IAAI,MAAM,CAAA;AAGxC,EAAA,MAAM,YAAA,GAA8C;AAAA,IAClD,KAAA,EAAO,UAAA;AAAA,IACP,OAAA,EAAS;AAAA,MACP,IAAA,EAAM,OAAA;AAAA,MACN,OAAO,CAAC,EAAE,MAAM,MAAA,EAAQ,IAAA,EAAM,8BAA8B;AAAA;AAC9D,GACF;AAEA,EAAA,IAAA,GAAO,2BAAA,CAA4B,MAAM,YAAY,CAAA;AAGrD,EAAA,MAAM,SAAA,CAAU,IAAA,CAAK,EAAE,OAAA,EAAS,MAAM,CAAA;AAGtC,EAAA,SAAA,CAAU,mBAAA,CAAoB,OAAO,MAAM,CAAA;AAG3C,EAAA,OAAO,qBAAA,CAAsB,SAAA,EAAW,IAAA,CAAK,IAAI,CAAA;AACnD;AAEA,eAAsB,wBAAA,CAAyB;AAAA,EAC7C,SAAA;AAAA,EACA,MAAA;AAAA,EACA,OAAA;AAAA,EACA,cAAA;AAAA,EACA,MAAA;AAAA,EACA,MAAA;AAAA,EACA,SAAA,GAAY,IAAI,iBAAA,EAAkB;AAAA,EAClC;AACF,CAAA,EAQiB;AACf,EAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,QAAA,CAAS,OAAO,CAAA;AAErC,EAAA,IAAI,MAAA;AAEJ,EAAA,IAAI;AAGF,IAAA,MAAA,GAAS,MAAA,CAAO,EAAA;AAGhB,IAAA,QAAQ,MAAA;AAAQ,MACd,KAAK,YAAA,EAAc;AACjB,QAAA,MAAME,OAAAA,GAAS,MAAM,cAAA,CAAe;AAAA,UAClC,SAAA;AAAA,UACA,MAAA;AAAA,UACA,SAAA;AAAA,UACA,KAAA;AAAA,UACA,OAAA;AAAA,UACA;AAAA,SACD,CAAA;AACD,QAAA,OAAOA,OAAAA;AAAA,MACT;AAAA,MACA,KAAK,qBAAA;AACH,QAAA,MAAM,MAAA,GAAS,MAAM,uBAAA,CAAwB;AAAA,UAC3C,SAAA;AAAA,UACA,SAAA;AAAA,UACA,MAAA;AAAA,UACA,KAAA;AAAA,UACA,OAAA;AAAA,UACA;AAAA,SACD,CAAA;AACD,QAAA,OAAO,MAAA;AAAA,MAET,KAAK,WAAA,EAAa;AAChB,QAAA,MAAMA,OAAAA,GAAS,MAAM,aAAA,CAAc;AAAA,UACjC,SAAA;AAAA,UACA,SAAA;AAAA,UACA,OAAA;AAAA,UACA;AAAA,SACD,CAAA;AAED,QAAA,OAAOA,OAAAA;AAAA,MACT;AAAA,MACA,KAAK,cAAA,EAAgB;AACnB,QAAA,MAAMA,OAAAA,GAAS,MAAM,gBAAA,CAAiB;AAAA,UACpC,SAAA;AAAA,UACA,SAAA;AAAA,UACA,OAAA;AAAA,UACA;AAAA,SACD,CAAA;AAED,QAAA,OAAOA,OAAAA;AAAA,MACT;AAAA,MACA;AACE,QAAA,MAAMF,YAAAA,CAAS,eAAe,MAAM,CAAA;AAAA;AACxC,EACF,SAAS,KAAA,EAAO;AACd,IAAA,IAAI,KAAA,YAAiBA,YAAAA,IAAY,MAAA,IAAU,CAAC,MAAM,MAAA,EAAQ;AACxD,MAAA,KAAA,CAAM,MAAA,GAAS,MAAA;AAAA,IACjB;AAEA,IAAA,OAAO,cAAA,CAAe,KAAA,EAAO,SAAA,EAAW,MAAA,EAAQ,MAAM,CAAA;AAAA,EACxD;AACF","file":"chunk-GASWM5HJ.cjs","sourcesContent":["import { A2AError } from '@mastra/core/a2a';\nimport type { TaskSendParams, TaskQueryParams, TaskIdParams, AgentCard, TaskStatus, TaskState } from '@mastra/core/a2a';\nimport type { Agent } from '@mastra/core/agent';\nimport type { IMastraLogger } from '@mastra/core/logger';\nimport type { RuntimeContext } from '@mastra/core/runtime-context';\nimport { z } from 'zod';\nimport { convertToCoreMessage, normalizeError, createSuccessResponse, createErrorResponse } from '../a2a/protocol';\nimport { InMemoryTaskStore } from '../a2a/store';\nimport { applyUpdateToTaskAndHistory, createTaskContext, loadOrCreateTaskAndHistory } from '../a2a/tasks';\nimport type { Context } from '../types';\n\nconst taskSendParamsSchema = z.object({\n  id: z.string().min(1, 'Invalid or missing task ID (params.id).'),\n  message: z.object({\n    parts: z.array(\n      z.object({\n        type: z.enum(['text']),\n        text: z.string(),\n      }),\n    ),\n  }),\n});\n\nexport async function getAgentCardByIdHandler({\n  mastra,\n  agentId,\n  executionUrl = `/a2a/${agentId}`,\n  provider = {\n    organization: 'Mastra',\n    url: 'https://mastra.ai',\n  },\n  version = '1.0',\n  runtimeContext,\n}: Context & {\n  runtimeContext: RuntimeContext;\n  agentId: keyof ReturnType<typeof mastra.getAgents>;\n  executionUrl?: string;\n  version?: string;\n  provider?: {\n    organization: string;\n    url: string;\n  };\n}): Promise<AgentCard> {\n  const agent = mastra.getAgent(agentId);\n\n  if (!agent) {\n    throw new Error(`Agent with ID ${agentId} not found`);\n  }\n\n  const [instructions, tools] = await Promise.all([\n    agent.getInstructions({ runtimeContext }),\n    agent.getTools({ runtimeContext }),\n  ]);\n\n  // Extract agent information to create the AgentCard\n  const agentCard: AgentCard = {\n    name: agent.id || agentId,\n    description: instructions,\n    url: executionUrl,\n    provider,\n    version,\n    capabilities: {\n      streaming: true, // All agents support streaming\n      pushNotifications: false,\n      stateTransitionHistory: false,\n    },\n    defaultInputModes: ['text'],\n    defaultOutputModes: ['text'],\n    // Convert agent tools to skills format for A2A protocol\n    skills: Object.entries(tools).map(([toolId, tool]) => ({\n      id: toolId,\n      name: toolId,\n      description: tool.description || `Tool: ${toolId}`,\n      // Optional fields\n      tags: ['tool'],\n    })),\n  };\n\n  return agentCard;\n}\n\nfunction validateTaskSendParams(params: TaskSendParams) {\n  try {\n    taskSendParamsSchema.parse(params);\n  } catch (error) {\n    if (error instanceof z.ZodError) {\n      throw A2AError.invalidParams((error as z.ZodError).errors[0]!.message);\n    }\n\n    throw error;\n  }\n}\n\nexport async function handleTaskSend({\n  requestId,\n  params,\n  taskStore,\n  agent,\n  agentId,\n  logger,\n  runtimeContext,\n}: {\n  requestId: string;\n  params: TaskSendParams;\n  taskStore: InMemoryTaskStore;\n  agent: Agent;\n  agentId: string;\n  logger?: IMastraLogger;\n  runtimeContext: RuntimeContext;\n}) {\n  validateTaskSendParams(params);\n\n  const { id: taskId, message, sessionId, metadata } = params;\n\n  // Load or create task AND history\n  let currentData = await loadOrCreateTaskAndHistory({\n    taskId,\n    taskStore,\n    agentId,\n    message,\n    sessionId,\n    metadata,\n  });\n\n  // Use the new TaskContext definition, passing history\n  const context = createTaskContext({\n    task: currentData.task,\n    userMessage: message,\n    history: currentData.history,\n    activeCancellations: taskStore.activeCancellations,\n  });\n\n  try {\n    const { text } = await agent.generate([convertToCoreMessage(message)], {\n      runId: taskId,\n      runtimeContext,\n    });\n\n    currentData = applyUpdateToTaskAndHistory(currentData, {\n      state: 'completed',\n      message: {\n        role: 'agent',\n        parts: [\n          {\n            type: 'text',\n            text: text,\n          },\n        ],\n      },\n    });\n\n    await taskStore.save({ agentId, data: currentData });\n    context.task = currentData.task;\n  } catch (handlerError) {\n    // If handler throws, apply 'failed' status, save, and rethrow\n    const failureStatusUpdate: Omit<TaskStatus, 'timestamp'> = {\n      state: 'failed',\n      message: {\n        role: 'agent',\n        parts: [\n          {\n            type: 'text',\n            text: `Handler failed: ${handlerError instanceof Error ? handlerError.message : String(handlerError)}`,\n          },\n        ],\n      },\n    };\n    currentData = applyUpdateToTaskAndHistory(currentData, failureStatusUpdate);\n\n    try {\n      await taskStore.save({ agentId, data: currentData });\n    } catch (saveError) {\n      // @ts-expect-error saveError is an unknown error\n      logger?.error(`Failed to save task ${taskId} after handler error:`, saveError?.message);\n    }\n\n    return normalizeError(handlerError, requestId, taskId, logger); // Rethrow original error\n  }\n\n  // The loop finished, send the final task state\n  return createSuccessResponse(requestId, currentData.task);\n}\n\nexport async function handleTaskGet({\n  requestId,\n  taskStore,\n  agentId,\n  taskId,\n}: {\n  requestId: string;\n  taskStore: InMemoryTaskStore;\n  agentId: string;\n  taskId: string;\n}) {\n  const task = await taskStore.load({ agentId, taskId });\n\n  if (!task) {\n    throw A2AError.taskNotFound(taskId);\n  }\n\n  return createSuccessResponse(requestId, task);\n}\n\nexport async function* handleTaskSendSubscribe({\n  requestId,\n  params,\n  taskStore,\n  agent,\n  agentId,\n  logger,\n  runtimeContext,\n}: {\n  requestId: string;\n  params: TaskSendParams;\n  taskStore: InMemoryTaskStore;\n  agent: Agent;\n  agentId: string;\n  logger?: IMastraLogger;\n  runtimeContext: RuntimeContext;\n}) {\n  yield createSuccessResponse(requestId, {\n    state: 'working',\n    message: {\n      role: 'agent',\n      parts: [{ type: 'text', text: 'Generating response...' }],\n    },\n  });\n\n  let result;\n  try {\n    result = await handleTaskSend({\n      requestId,\n      params,\n      taskStore,\n      agent,\n      agentId,\n      runtimeContext,\n      logger,\n    });\n  } catch (err) {\n    if (!(err instanceof A2AError)) {\n      throw err;\n    }\n\n    result = createErrorResponse(requestId, err.toJSONRPCError());\n  }\n\n  yield result;\n}\n\nexport async function handleTaskCancel({\n  requestId,\n  taskStore,\n  agentId,\n  taskId,\n  logger,\n}: {\n  requestId: string;\n  taskStore: InMemoryTaskStore;\n  agentId: string;\n  taskId: string;\n  logger?: IMastraLogger;\n}) {\n  // Load task and history\n  let data = await taskStore.load({\n    agentId,\n    taskId,\n  });\n\n  if (!data) {\n    throw A2AError.taskNotFound(taskId);\n  }\n\n  // Check if cancelable (not already in a final state)\n  const finalStates: TaskState[] = ['completed', 'failed', 'canceled'];\n\n  if (finalStates.includes(data.task.status.state)) {\n    logger?.info(`Task ${taskId} already in final state ${data.task.status.state}, cannot cancel.`);\n    return createSuccessResponse(requestId, data.task);\n  }\n\n  // Signal cancellation\n  taskStore.activeCancellations.add(taskId);\n\n  // Apply 'canceled' state update\n  const cancelUpdate: Omit<TaskStatus, 'timestamp'> = {\n    state: 'canceled',\n    message: {\n      role: 'agent',\n      parts: [{ type: 'text', text: 'Task cancelled by request.' }],\n    },\n  };\n\n  data = applyUpdateToTaskAndHistory(data, cancelUpdate);\n\n  // Save the updated state\n  await taskStore.save({ agentId, data });\n\n  // Remove from active cancellations *after* saving\n  taskStore.activeCancellations.delete(taskId);\n\n  // Return the updated task object\n  return createSuccessResponse(requestId, data.task);\n}\n\nexport async function getAgentExecutionHandler({\n  requestId,\n  mastra,\n  agentId,\n  runtimeContext,\n  method,\n  params,\n  taskStore = new InMemoryTaskStore(),\n  logger,\n}: Context & {\n  requestId: string;\n  runtimeContext: RuntimeContext;\n  agentId: string;\n  method: 'tasks/send' | 'tasks/sendSubscribe' | 'tasks/get' | 'tasks/cancel';\n  params: TaskSendParams | TaskQueryParams | TaskIdParams;\n  taskStore?: InMemoryTaskStore;\n  logger?: IMastraLogger;\n}): Promise<any> {\n  const agent = mastra.getAgent(agentId);\n\n  let taskId: string | undefined; // For error context\n\n  try {\n    // Attempt to get task ID early for error context. Cast params to any to access id.\n    // Proper validation happens within specific handlers.\n    taskId = params.id;\n\n    // 2. Route based on method\n    switch (method) {\n      case 'tasks/send': {\n        const result = await handleTaskSend({\n          requestId,\n          params: params as TaskSendParams,\n          taskStore,\n          agent,\n          agentId,\n          runtimeContext,\n        });\n        return result;\n      }\n      case 'tasks/sendSubscribe':\n        const result = await handleTaskSendSubscribe({\n          requestId,\n          taskStore,\n          params: params as TaskSendParams,\n          agent,\n          agentId,\n          runtimeContext,\n        });\n        return result;\n\n      case 'tasks/get': {\n        const result = await handleTaskGet({\n          requestId,\n          taskStore,\n          agentId,\n          taskId,\n        });\n\n        return result;\n      }\n      case 'tasks/cancel': {\n        const result = await handleTaskCancel({\n          requestId,\n          taskStore,\n          agentId,\n          taskId,\n        });\n\n        return result;\n      }\n      default:\n        throw A2AError.methodNotFound(method);\n    }\n  } catch (error) {\n    if (error instanceof A2AError && taskId && !error.taskId) {\n      error.taskId = taskId; // Add task ID context if missing\n    }\n\n    return normalizeError(error, requestId, taskId, logger);\n  }\n}\n","import { A2AError } from '@mastra/core/a2a';\n\nimport type { JSONRPCError, JSONRPCResponse, Message, Part } from '@mastra/core/a2a';\nimport type { CoreMessage } from '@mastra/core/llm';\nimport type { IMastraLogger } from '@mastra/core/logger';\n\nexport function normalizeError(\n  error: any,\n  reqId: number | string | null,\n  taskId?: string,\n  logger?: IMastraLogger,\n): JSONRPCResponse<null, unknown> {\n  let a2aError: A2AError;\n  if (error instanceof A2AError) {\n    a2aError = error;\n  } else if (error instanceof Error) {\n    // Generic JS error\n    a2aError = A2AError.internalError(error.message, { stack: error.stack });\n  } else {\n    // Unknown error type\n    a2aError = A2AError.internalError('An unknown error occurred.', error);\n  }\n\n  // Ensure Task ID context is present if possible\n  if (taskId && !a2aError.taskId) {\n    a2aError.taskId = taskId;\n  }\n\n  logger?.error(`Error processing request (Task: ${a2aError.taskId ?? 'N/A'}, ReqID: ${reqId ?? 'N/A'}):`, a2aError);\n\n  return createErrorResponse(reqId, a2aError.toJSONRPCError());\n}\n\nexport function createErrorResponse(\n  id: number | string | null,\n  error: JSONRPCError<unknown>,\n): JSONRPCResponse<null, unknown> {\n  // For errors, ID should be the same as request ID, or null if that couldn't be determined\n  return {\n    jsonrpc: '2.0',\n    id: id, // Can be null if request ID was invalid/missing\n    error: error,\n  };\n}\n\nexport function createSuccessResponse<T>(id: number | string | null, result: T): JSONRPCResponse<T> {\n  if (!id) {\n    // This shouldn't happen for methods that expect a response, but safeguard\n    throw A2AError.internalError('Cannot create success response for null ID.');\n  }\n\n  return {\n    jsonrpc: '2.0',\n    id: id,\n    result: result,\n  };\n}\n\nexport function convertToCoreMessage(message: Message): CoreMessage {\n  return {\n    role: message.role === 'user' ? 'user' : 'assistant',\n    content: message.parts.map(msg => convertToCoreMessagePart(msg)),\n  };\n}\n\nfunction convertToCoreMessagePart(part: Part) {\n  switch (part.type) {\n    case 'text':\n      return {\n        type: 'text',\n        text: part.text,\n      } as const;\n    case 'file':\n      return {\n        type: 'file',\n        data: new URL(part.file.uri!),\n        mimeType: part.file.mimeType!,\n      } as const;\n    case 'data':\n      throw new Error('Data parts are not supported in core messages');\n  }\n}\n","import type { TaskAndHistory } from '@mastra/core/a2a';\n\nexport class InMemoryTaskStore {\n  private store: Map<string, TaskAndHistory> = new Map();\n  public activeCancellations = new Set<string>();\n\n  async load({ agentId, taskId }: { agentId: string; taskId: string }): Promise<TaskAndHistory | null> {\n    const entry = this.store.get(`${agentId}-${taskId}`);\n\n    if (!entry) {\n      return null;\n    }\n\n    // Return copies to prevent external mutation\n    return { task: { ...entry.task }, history: [...entry.history] };\n  }\n\n  async save({ agentId, data }: { agentId: string; data: TaskAndHistory }): Promise<void> {\n    // Store copies to prevent internal mutation if caller reuses objects\n    const key = `${agentId}-${data.task.id}`;\n    if (!data.task.id) {\n      throw new Error('Task ID is required');\n    }\n    this.store.set(key, {\n      task: { ...data.task },\n      history: [...data.history],\n    });\n  }\n}\n","import type { Message, TaskContext, TaskAndHistory, Task, TaskState, TaskStatus, Artifact } from '@mastra/core/a2a';\nimport type { IMastraLogger } from '@mastra/core/logger';\nimport type { InMemoryTaskStore } from './store';\n\nfunction isTaskStatusUpdate(update: TaskStatus | Artifact): update is Omit<TaskStatus, 'timestamp'> {\n  return 'state' in update && !('parts' in update);\n}\n\nfunction isArtifactUpdate(update: TaskStatus | Artifact): update is Artifact {\n  return 'parts' in update;\n}\n\nexport function applyUpdateToTaskAndHistory(\n  current: TaskAndHistory,\n  update: Omit<TaskStatus, 'timestamp'> | Artifact,\n): TaskAndHistory {\n  let newTask = structuredClone(current.task);\n  let newHistory = structuredClone(current.history);\n\n  if (isTaskStatusUpdate(update)) {\n    // Merge status update\n    newTask.status = {\n      ...newTask.status, // Keep existing properties if not overwritten\n      ...update, // Apply updates\n      timestamp: new Date().toISOString(),\n    };\n\n    // If the update includes an agent message, add it to history\n    if (update.message?.role === 'agent') {\n      newHistory.push(update.message);\n    }\n  } else if (isArtifactUpdate(update)) {\n    // Handle artifact update\n    if (!newTask.artifacts) {\n      newTask.artifacts = [];\n    } else {\n      // Ensure we're working with a copy of the artifacts array\n      newTask.artifacts = [...newTask.artifacts];\n    }\n\n    const existingIndex = update.index ?? -1; // Use index if provided\n    let replaced = false;\n\n    if (existingIndex >= 0 && existingIndex < newTask.artifacts.length) {\n      const existingArtifact = newTask.artifacts[existingIndex];\n      if (update.append) {\n        // Create a deep copy for modification to avoid mutating original\n        const appendedArtifact = JSON.parse(JSON.stringify(existingArtifact));\n        appendedArtifact.parts.push(...update.parts);\n        if (update.metadata) {\n          appendedArtifact.metadata = {\n            ...(appendedArtifact.metadata || {}),\n            ...update.metadata,\n          };\n        }\n        if (update.lastChunk !== undefined) appendedArtifact.lastChunk = update.lastChunk;\n        if (update.description) appendedArtifact.description = update.description;\n        newTask.artifacts[existingIndex] = appendedArtifact; // Replace with appended version\n        replaced = true;\n      } else {\n        // Overwrite artifact at index (with a copy of the update)\n        newTask.artifacts[existingIndex] = { ...update };\n        replaced = true;\n      }\n    } else if (update.name) {\n      const namedIndex = newTask.artifacts.findIndex(a => a.name === update.name);\n      if (namedIndex >= 0) {\n        newTask.artifacts[namedIndex] = { ...update }; // Replace by name (with copy)\n        replaced = true;\n      }\n    }\n\n    if (!replaced) {\n      newTask.artifacts.push({ ...update }); // Add as a new artifact (copy)\n      // Sort if indices are present\n      if (newTask.artifacts.some(a => a.index !== undefined)) {\n        newTask.artifacts.sort((a, b) => (a.index ?? 0) - (b.index ?? 0));\n      }\n    }\n  }\n\n  return { task: newTask, history: newHistory };\n}\n\nexport async function loadOrCreateTaskAndHistory({\n  agentId,\n  taskId,\n  taskStore,\n  message,\n  sessionId,\n  metadata,\n  logger,\n}: {\n  agentId: string;\n  taskId: string;\n  taskStore: InMemoryTaskStore;\n  message: Message;\n  sessionId?: string | null;\n  metadata?: Record<string, unknown> | null;\n  logger?: IMastraLogger;\n}): Promise<TaskAndHistory> {\n  const data = await taskStore.load({ agentId, taskId });\n\n  // Create new task if none exists\n  if (!data) {\n    const initialTask: Task = {\n      id: taskId,\n      sessionId: sessionId,\n      status: {\n        state: 'submitted',\n        timestamp: new Date().toISOString(),\n        message: null,\n      },\n      artifacts: [],\n      metadata: metadata,\n    };\n\n    const initialData = {\n      task: initialTask,\n      history: [message],\n    };\n\n    logger?.info(`[Task ${taskId}] Created new task and history.`);\n    await taskStore.save({ agentId, data: initialData });\n\n    return initialData;\n  }\n\n  // Handle existing task\n  logger?.info(`[Task ${taskId}] Loaded existing task and history.`);\n\n  // Add message to history and prepare updated data\n  let updatedData = {\n    task: data.task,\n    history: [...data.history, message],\n  };\n\n  // Handle state transitions\n  const { status } = data.task;\n  const finalStates: TaskState[] = ['completed', 'failed', 'canceled'];\n\n  if (finalStates.includes(status.state)) {\n    logger?.warn(`[Task ${taskId}] Received message for task in final state ${status.state}. Restarting.`);\n    updatedData = applyUpdateToTaskAndHistory(updatedData, {\n      state: 'submitted',\n      message: null,\n    });\n  } else if (status.state === 'input-required') {\n    logger?.info(`[Task ${taskId}] Changing state from 'input-required' to 'working'.`);\n    updatedData = applyUpdateToTaskAndHistory(updatedData, { state: 'working' });\n  } else if (status.state === 'working') {\n    logger?.warn(`[Task ${taskId}] Received message while already 'working'. Proceeding.`);\n  }\n\n  await taskStore.save({ agentId, data: updatedData });\n\n  return {\n    task: { ...updatedData.task },\n    history: [...updatedData.history],\n  };\n}\n\nexport function createTaskContext({\n  task,\n  userMessage,\n  history,\n  activeCancellations,\n}: {\n  task: Task;\n  userMessage: Message;\n  history: Message[];\n  activeCancellations: Set<string>;\n}): TaskContext {\n  return {\n    task: structuredClone(task),\n    userMessage,\n    history: structuredClone(history),\n    isCancelled: () => activeCancellations.has(task.id),\n  };\n}\n"]}