@mastra/server 0.0.0-fix-memory-xxhash-20250409202110 → 0.0.0-generate-message-id-20250512171942

package/dist/chunk-P6SCPDYW.js

@@ -0,0 +1,500 @@
+import { __export } from './chunk-MLKGABMK.js';
+import { A2AError } from '@mastra/core/a2a';
+import { z } from 'zod';
+
+// src/server/handlers/a2a.ts
+var a2a_exports = {};
+__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 A2AError) {
+    a2aError = error;
+  } else if (error instanceof Error) {
+    a2aError = A2AError.internalError(error.message, { stack: error.stack });
+  } else {
+    a2aError = 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 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 = z.object({
+  id: z.string().min(1, "Invalid or missing task ID (params.id)."),
+  message: z.object({
+    parts: z.array(
+      z.object({
+        type: z.enum(["text"]),
+        text: 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 z.ZodError) {
+      throw A2AError.invalidParams(error.errors[0].message);
+    }
+    throw error;
+  }
+}
+async function handleTaskSend({
+  requestId,
+  params,
+  taskStore,
+  agent,
+  logger,
+  runtimeContext
+}) {
+  validateTaskSendParams(params);
+  const agentId = agent.id;
+  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 A2AError.taskNotFound(taskId);
+  }
+  return createSuccessResponse(requestId, task);
+}
+async function* handleTaskSendSubscribe({
+  requestId,
+  params,
+  taskStore,
+  agent,
+  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,
+      runtimeContext,
+      logger
+    });
+  } catch (err) {
+    if (!(err instanceof 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 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,
+          runtimeContext
+        });
+        return result2;
+      }
+      case "tasks/sendSubscribe":
+        const result = await handleTaskSendSubscribe({
+          requestId,
+          taskStore,
+          params,
+          agent,
+          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 A2AError.methodNotFound(method);
+    }
+  } catch (error) {
+    if (error instanceof A2AError && taskId && !error.taskId) {
+      error.taskId = taskId;
+    }
+    return normalizeError(error, requestId, taskId, logger);
+  }
+}
+
+export { a2a_exports, getAgentCardByIdHandler, getAgentExecutionHandler, handleTaskCancel, handleTaskGet, handleTaskSend, handleTaskSendSubscribe };

package/dist/{chunk-BMA2ORRT.js → chunk-Q6KMBIAN.js}

@@ -1,7 +1,9 @@
-import { stringify, esm_default } from './chunk-VK6FX47H.js';
-import { validateBody } from './chunk-L7XE5QTW.js';
-import { handleError } from './chunk-3AHQ5RGN.js';
-import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { stringify, esm_default } from './chunk-LIVAK2DM.js';
+import { validateBody } from './chunk-H5PTF3Y4.js';
+import { handleError } from './chunk-M5ABIP7D.js';
+import { HTTPException } from './chunk-NYN7KFXL.js';
+import { __export } from './chunk-MLKGABMK.js';
+import { RuntimeContext } from '@mastra/core/runtime-context';
 
 // src/server/handlers/agents.ts
 var agents_exports = {};
@@ -13,27 +15,35 @@ __export(agents_exports, {
   getLiveEvalsByAgentIdHandler: () => getLiveEvalsByAgentIdHandler,
   streamGenerateHandler: () => streamGenerateHandler
 });
-async function getAgentsHandler({ mastra }) {
+async function getAgentsHandler({ mastra, runtimeContext }) {
   try {
     const agents = mastra.getAgents();
-    const serializedAgents = Object.entries(agents).reduce((acc, [_id, _agent]) => {
-      const agent = _agent;
-      const serializedAgentTools = Object.entries(agent?.tools || {}).reduce((acc2, [key, tool]) => {
-        const _tool = tool;
-        acc2[key] = {
-          ..._tool,
-          inputSchema: _tool.inputSchema ? stringify(esm_default(_tool.inputSchema)) : void 0,
-          outputSchema: _tool.outputSchema ? stringify(esm_default(_tool.outputSchema)) : void 0
+    const serializedAgentsMap = await Promise.all(
+      Object.entries(agents).map(async ([id, agent]) => {
+        const instructions = await agent.getInstructions({ runtimeContext });
+        const tools = await agent.getTools({ runtimeContext });
+        const llm = await agent.getLLM({ runtimeContext });
+        const serializedAgentTools = Object.entries(tools || {}).reduce((acc, [key, tool]) => {
+          const _tool = tool;
+          acc[key] = {
+            ..._tool,
+            inputSchema: _tool.inputSchema ? stringify(esm_default(_tool.inputSchema)) : void 0,
+            outputSchema: _tool.outputSchema ? stringify(esm_default(_tool.outputSchema)) : void 0
+          };
+          return acc;
+        }, {});
+        return {
+          id,
+          name: agent.name,
+          instructions,
+          tools: serializedAgentTools,
+          provider: llm?.getProvider(),
+          modelId: llm?.getModelId()
         };
-        return acc2;
-      }, {});
-      acc[_id] = {
-        name: agent.name,
-        instructions: agent.instructions,
-        tools: serializedAgentTools,
-        provider: agent.llm?.getProvider(),
-        modelId: agent.llm?.getModelId()
-      };
+      })
+    );
+    const serializedAgents = serializedAgentsMap.reduce((acc, { id, ...rest }) => {
+      acc[id] = rest;
       return acc;
     }, {});
     return serializedAgents;
@@ -41,13 +51,18 @@ async function getAgentsHandler({ mastra }) {
     return handleError(error, "Error getting agents");
   }
 }
-async function getAgentByIdHandler({ mastra, agentId }) {
+async function getAgentByIdHandler({
+  mastra,
+  runtimeContext,
+  agentId
+}) {
   try {
     const agent = mastra.getAgent(agentId);
     if (!agent) {
       throw new HTTPException(404, { message: "Agent not found" });
     }
-    const serializedAgentTools = Object.entries(agent?.tools || {}).reduce((acc, [key, tool]) => {
+    const tools = await agent.getTools({ runtimeContext });
+    const serializedAgentTools = Object.entries(tools || {}).reduce((acc, [key, tool]) => {
       const _tool = tool;
       acc[key] = {
         ..._tool,
@@ -56,39 +71,51 @@ async function getAgentByIdHandler({ mastra, agentId }) {
       };
       return acc;
     }, {});
+    const instructions = await agent.getInstructions({ runtimeContext });
+    const llm = await agent.getLLM({ runtimeContext });
     return {
       name: agent.name,
-      instructions: agent.instructions,
+      instructions,
       tools: serializedAgentTools,
-      provider: agent.llm?.getProvider(),
-      modelId: agent.llm?.getModelId()
+      provider: llm?.getProvider(),
+      modelId: llm?.getModelId()
     };
   } catch (error) {
     return handleError(error, "Error getting agent");
   }
 }
-async function getEvalsByAgentIdHandler({ mastra, agentId }) {
+async function getEvalsByAgentIdHandler({
+  mastra,
+  runtimeContext,
+  agentId
+}) {
   try {
     const agent = mastra.getAgent(agentId);
-    const evals = await mastra.getStorage()?.__getEvalsByAgentName?.(agent.name, "test") || [];
+    const evals = await mastra.getStorage()?.getEvalsByAgentName?.(agent.name, "test") || [];
+    const instructions = await agent.getInstructions({ runtimeContext });
     return {
       id: agentId,
       name: agent.name,
-      instructions: agent.instructions,
+      instructions,
       evals
     };
   } catch (error) {
     return handleError(error, "Error getting test evals");
   }
 }
-async function getLiveEvalsByAgentIdHandler({ mastra, agentId }) {
+async function getLiveEvalsByAgentIdHandler({
+  mastra,
+  runtimeContext,
+  agentId
+}) {
   try {
     const agent = mastra.getAgent(agentId);
-    const evals = await mastra.getStorage()?.__getEvalsByAgentName?.(agent.name, "live") || [];
+    const evals = await mastra.getStorage()?.getEvalsByAgentName?.(agent.name, "live") || [];
+    const instructions = await agent.getInstructions({ runtimeContext });
     return {
       id: agentId,
       name: agent.name,
-      instructions: agent.instructions,
+      instructions,
       evals
     };
   } catch (error) {
@@ -97,6 +124,7 @@ async function getLiveEvalsByAgentIdHandler({ mastra, agentId }) {
 }
 async function generateHandler({
   mastra,
+  runtimeContext,
   agentId,
   body
 }) {
@@ -105,13 +133,18 @@ async function generateHandler({
     if (!agent) {
       throw new HTTPException(404, { message: "Agent not found" });
     }
-    const { messages, resourceId, resourceid, ...rest } = body;
+    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
+    const finalRuntimeContext = new RuntimeContext([
+      ...Array.from(runtimeContext.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
     validateBody({ messages });
     const result = await agent.generate(messages, {
       ...rest,
       // @ts-expect-error TODO fix types
-      resourceId: finalResourceId
+      resourceId: finalResourceId,
+      runtimeContext: finalRuntimeContext
     });
     return result;
   } catch (error) {
@@ -120,6 +153,7 @@ async function generateHandler({
 }
 async function streamGenerateHandler({
   mastra,
+  runtimeContext,
   agentId,
   body
 }) {
@@ -128,13 +162,18 @@ async function streamGenerateHandler({
     if (!agent) {
       throw new HTTPException(404, { message: "Agent not found" });
     }
-    const { messages, resourceId, resourceid, ...rest } = body;
+    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
+    const finalRuntimeContext = new RuntimeContext([
+      ...Array.from(runtimeContext.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
     validateBody({ messages });
     const streamResult = await agent.stream(messages, {
       ...rest,
       // @ts-expect-error TODO fix types
-      resourceId: finalResourceId
+      resourceId: finalResourceId,
+      runtimeContext: finalRuntimeContext
     });
     const streamResponse = rest.output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
       sendUsage: true,