@mastra/server 0.0.0-mcp-schema-serializer-20250430202337 → 0.0.0-mcp-changeset-20250707162621

package/dist/{chunk-Y3SV5XK4.js → chunk-BK4XT6EG.js}

@@ -1,7 +1,9 @@
-import { stringify, esm_default } from './chunk-OMN3UI6X.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-MEGCYGBU.js';
+import { validateBody } from './chunk-RSEO4XPX.js';
+import { handleError } from './chunk-LF7P5PLR.js';
+import { HTTPException } from './chunk-LCM566I4.js';
+import { __export } from './chunk-MLKGABMK.js';
+import { RuntimeContext } from '@mastra/core/runtime-context';
 
 // src/server/handlers/agents.ts
 var agents_exports = {};
@@ -21,6 +23,8 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
         const instructions = await agent.getInstructions({ runtimeContext });
         const tools = await agent.getTools({ runtimeContext });
         const llm = await agent.getLLM({ runtimeContext });
+        const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext });
+        const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext });
         const serializedAgentTools = Object.entries(tools || {}).reduce((acc, [key, tool]) => {
           const _tool = tool;
           acc[key] = {
@@ -30,13 +34,33 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
           };
           return acc;
         }, {});
+        let serializedAgentWorkflows = {};
+        if ("getWorkflows" in agent) {
+          const logger = mastra.getLogger();
+          try {
+            const workflows = await agent.getWorkflows({ runtimeContext });
+            serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
+              return {
+                ...acc,
+                [key]: {
+                  name: workflow.name
+                }
+              };
+            }, {});
+          } catch (error) {
+            logger.error("Error getting workflows for agent", { agentName: agent.name, error });
+          }
+        }
         return {
           id,
           name: agent.name,
           instructions,
           tools: serializedAgentTools,
+          workflows: serializedAgentWorkflows,
           provider: llm?.getProvider(),
-          modelId: llm?.getModelId()
+          modelId: llm?.getModelId(),
+          defaultGenerateOptions,
+          defaultStreamOptions
         };
       })
     );
@@ -52,7 +76,8 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
 async function getAgentByIdHandler({
   mastra,
   runtimeContext,
-  agentId
+  agentId,
+  isPlayground = false
 }) {
   try {
     const agent = mastra.getAgent(agentId);
@@ -69,14 +94,59 @@ async function getAgentByIdHandler({
       };
       return acc;
     }, {});
-    const instructions = await agent.getInstructions({ runtimeContext });
+    let serializedAgentWorkflows = {};
+    if ("getWorkflows" in agent) {
+      const logger = mastra.getLogger();
+      try {
+        const workflows = await agent.getWorkflows({ runtimeContext });
+        serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
+          return {
+            ...acc,
+            [key]: {
+              name: workflow.name,
+              steps: Object.entries(workflow.steps).reduce((acc2, [key2, step]) => {
+                return {
+                  ...acc2,
+                  [key2]: {
+                    id: step.id,
+                    description: step.description
+                  }
+                };
+              }, {})
+            }
+          };
+        }, {});
+      } catch (error) {
+        logger.error("Error getting workflows for agent", { agentName: agent.name, error });
+      }
+    }
+    let proxyRuntimeContext = runtimeContext;
+    if (isPlayground) {
+      proxyRuntimeContext = new Proxy(runtimeContext, {
+        get(target, prop) {
+          if (prop === "get") {
+            return function(key) {
+              const value = target.get(key);
+              return value ?? `<${key}>`;
+            };
+          }
+          return Reflect.get(target, prop);
+        }
+      });
+    }
+    const instructions = await agent.getInstructions({ runtimeContext: proxyRuntimeContext });
     const llm = await agent.getLLM({ runtimeContext });
+    const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext: proxyRuntimeContext });
+    const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext: proxyRuntimeContext });
     return {
       name: agent.name,
       instructions,
       tools: serializedAgentTools,
+      workflows: serializedAgentWorkflows,
       provider: llm?.getProvider(),
-      modelId: llm?.getModelId()
+      modelId: llm?.getModelId(),
+      defaultGenerateOptions,
+      defaultStreamOptions
     };
   } catch (error) {
     return handleError(error, "Error getting agent");
@@ -124,21 +194,27 @@ async function generateHandler({
   mastra,
   runtimeContext,
   agentId,
-  body
+  body,
+  abortSignal
 }) {
   try {
     const agent = mastra.getAgent(agentId);
     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,
-      runtimeContext
+      runtimeContext: finalRuntimeContext,
+      signal: abortSignal
     });
     return result;
   } catch (error) {
@@ -149,32 +225,45 @@ async function streamGenerateHandler({
   mastra,
   runtimeContext,
   agentId,
-  body
+  body,
+  abortSignal
 }) {
   try {
     const agent = mastra.getAgent(agentId);
     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,
-      runtimeContext
+      runtimeContext: finalRuntimeContext,
+      signal: abortSignal
     });
-    const streamResponse = rest.output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
+    const streamResponse = rest.output ? streamResult.toTextStreamResponse({
+      headers: {
+        "Transfer-Encoding": "chunked"
+      }
+    }) : streamResult.toDataStreamResponse({
       sendUsage: true,
       sendReasoning: true,
       getErrorMessage: (error) => {
         return `An error occurred while processing your request. ${error instanceof Error ? error.message : JSON.stringify(error)}`;
+      },
+      headers: {
+        "Transfer-Encoding": "chunked"
       }
     });
     return streamResponse;
   } catch (error) {
-    throw new HTTPException(error?.status ?? 500, { message: error?.message ?? "Error streaming from agent" });
+    return handleError(error, "error streaming agent response");
   }
 }
 

package/dist/{chunk-QN4KF3BH.cjs → chunk-CCGRCYWJ.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+var chunk2KZFMI6P_cjs = require('./chunk-2KZFMI6P.cjs');
 
 // src/server/handlers/utils.ts
 function validateBody(body) {
@@ -11,7 +11,7 @@ function validateBody(body) {
     return acc;
   }, {});
   if (Object.keys(errorResponse).length > 0) {
-    throw new chunkFV45V6WC_cjs.HTTPException(400, { message: Object.values(errorResponse)[0] });
+    throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: Object.values(errorResponse)[0] });
   }
 }
 

package/dist/{chunk-QJ3AHN64.js → chunk-CRCR3ZUK.js}

@@ -1,6 +1,7 @@
-import { validateBody } from './chunk-L7XE5QTW.js';
-import { handleError } from './chunk-3AHQ5RGN.js';
-import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { validateBody } from './chunk-RSEO4XPX.js';
+import { handleError } from './chunk-LF7P5PLR.js';
+import { HTTPException } from './chunk-LCM566I4.js';
+import { __export } from './chunk-MLKGABMK.js';
 
 // src/server/handlers/network.ts
 var network_exports = {};

package/dist/{chunk-2FJURXCL.cjs → chunk-DN7K6FJK.cjs}

@@ -1,12 +1,13 @@
 'use strict';
 
-var chunkQN4KF3BH_cjs = require('./chunk-QN4KF3BH.cjs');
-var chunkZLBRQFDD_cjs = require('./chunk-ZLBRQFDD.cjs');
-var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+var chunkCCGRCYWJ_cjs = require('./chunk-CCGRCYWJ.cjs');
+var chunkPZQDCRPV_cjs = require('./chunk-PZQDCRPV.cjs');
+var chunk2KZFMI6P_cjs = require('./chunk-2KZFMI6P.cjs');
+var chunk75ZPJI57_cjs = require('./chunk-75ZPJI57.cjs');
 
 // src/server/handlers/memory.ts
 var memory_exports = {};
-chunkFV45V6WC_cjs.__export(memory_exports, {
+chunk75ZPJI57_cjs.__export(memory_exports, {
   createThreadHandler: () => createThreadHandler,
   deleteThreadHandler: () => deleteThreadHandler,
   getMemoryStatusHandler: () => getMemoryStatusHandler,
@@ -16,80 +17,102 @@ chunkFV45V6WC_cjs.__export(memory_exports, {
   saveMessagesHandler: () => saveMessagesHandler,
   updateThreadHandler: () => updateThreadHandler
 });
-function getMemoryFromContext({
+async function getMemoryFromContext({
   mastra,
-  agentId
+  agentId,
+  networkId,
+  runtimeContext
 }) {
   const agent = agentId ? mastra.getAgent(agentId) : null;
   if (agentId && !agent) {
-    throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Agent not found" });
+    throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Agent not found" });
+  }
+  const network = networkId ? mastra.vnext_getNetwork(networkId) : null;
+  if (networkId && !network) {
+    throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Network not found" });
+  }
+  if (agent) {
+    return agent?.getMemory() || mastra.getMemory();
+  }
+  if (network) {
+    return await network?.getMemory({ runtimeContext }) || mastra.getMemory();
   }
-  const memory = agent?.getMemory?.() || mastra.getMemory();
-  return memory;
+  return mastra.getMemory();
 }
-async function getMemoryStatusHandler({ mastra, agentId }) {
+async function getMemoryStatusHandler({
+  mastra,
+  agentId,
+  networkId,
+  runtimeContext
+}) {
   try {
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
       return { result: false };
     }
     return { result: true };
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting memory status");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting memory status");
   }
 }
 async function getThreadsHandler({
   mastra,
   agentId,
-  resourceId
+  resourceId,
+  networkId,
+  runtimeContext
 }) {
   try {
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
-    chunkQN4KF3BH_cjs.validateBody({ resourceId });
+    chunkCCGRCYWJ_cjs.validateBody({ resourceId });
     const threads = await memory.getThreadsByResourceId({ resourceId });
     return threads;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting threads");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting threads");
   }
 }
 async function getThreadByIdHandler({
   mastra,
   agentId,
-  threadId
+  threadId,
+  networkId,
+  runtimeContext
 }) {
   try {
-    chunkQN4KF3BH_cjs.validateBody({ threadId });
-    const memory = getMemoryFromContext({ mastra, agentId });
+    chunkCCGRCYWJ_cjs.validateBody({ threadId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
     const thread = await memory.getThreadById({ threadId });
     if (!thread) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Thread not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Thread not found" });
     }
     return thread;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting thread");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting thread");
   }
 }
 async function saveMessagesHandler({
   mastra,
   agentId,
-  body
+  body,
+  networkId,
+  runtimeContext
 }) {
   try {
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
     if (!body?.messages) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Messages are required" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Messages are required" });
     }
     if (!Array.isArray(body.messages)) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Messages should be an array" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Messages should be an array" });
     }
     const processedMessages = body.messages.map((message) => ({
       ...message,
@@ -99,20 +122,22 @@ async function saveMessagesHandler({
     const result = await memory.saveMessages({ messages: processedMessages, memoryConfig: {} });
     return result;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error saving messages");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error saving messages");
   }
 }
 async function createThreadHandler({
   mastra,
   agentId,
-  body
+  body,
+  networkId,
+  runtimeContext
 }) {
   try {
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
-    chunkQN4KF3BH_cjs.validateBody({ resourceId: body?.resourceId });
+    chunkCCGRCYWJ_cjs.validateBody({ resourceId: body?.resourceId });
     const result = await memory.createThread({
       resourceId: body?.resourceId,
       title: body?.title,
@@ -121,29 +146,31 @@ async function createThreadHandler({
     });
     return result;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error saving thread to memory");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error saving thread to memory");
   }
 }
 async function updateThreadHandler({
   mastra,
   agentId,
   threadId,
-  body
+  body,
+  networkId,
+  runtimeContext
 }) {
   try {
-    const memory = getMemoryFromContext({ mastra, agentId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!body) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Body is required" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Body is required" });
     }
     const { title, metadata, resourceId } = body;
     const updatedAt = /* @__PURE__ */ new Date();
-    chunkQN4KF3BH_cjs.validateBody({ threadId });
+    chunkCCGRCYWJ_cjs.validateBody({ threadId });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
     const thread = await memory.getThreadById({ threadId });
     if (!thread) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Thread not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Thread not found" });
     }
     const updatedThread = {
       ...thread,
@@ -156,49 +183,60 @@ async function updateThreadHandler({
     const result = await memory.saveThread({ thread: updatedThread });
     return result;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error updating thread");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error updating thread");
   }
 }
 async function deleteThreadHandler({
   mastra,
   agentId,
-  threadId
+  threadId,
+  networkId,
+  runtimeContext
 }) {
   try {
-    chunkQN4KF3BH_cjs.validateBody({ threadId });
-    const memory = getMemoryFromContext({ mastra, agentId });
+    chunkCCGRCYWJ_cjs.validateBody({ threadId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
     const thread = await memory.getThreadById({ threadId });
     if (!thread) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Thread not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Thread not found" });
     }
     await memory.deleteThread(threadId);
     return { result: "Thread deleted" };
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error deleting thread");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error deleting thread");
   }
 }
 async function getMessagesHandler({
   mastra,
   agentId,
-  threadId
+  threadId,
+  limit,
+  networkId,
+  runtimeContext
 }) {
+  if (limit !== void 0 && (!Number.isInteger(limit) || limit <= 0)) {
+    throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Invalid limit: must be a positive integer" });
+  }
   try {
-    chunkQN4KF3BH_cjs.validateBody({ threadId });
-    const memory = getMemoryFromContext({ mastra, agentId });
+    chunkCCGRCYWJ_cjs.validateBody({ threadId });
+    const memory = await getMemoryFromContext({ mastra, agentId, networkId, runtimeContext });
     if (!memory) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Memory is not initialized" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Memory is not initialized" });
     }
     const thread = await memory.getThreadById({ threadId });
     if (!thread) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Thread not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Thread not found" });
     }
-    const result = await memory.query({ threadId });
-    return result;
+    const result = await memory.query({
+      threadId,
+      ...limit && { selectBy: { last: limit } }
+    });
+    return { messages: result.messages, uiMessages: result.uiMessages };
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting messages");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting messages");
   }
 }
 

package/dist/{chunk-D3G23FP3.cjs → chunk-FRVBFMO2.cjs}

@@ -1,14 +1,15 @@
 'use strict';
 
-var chunk5SWCVTNL_cjs = require('./chunk-5SWCVTNL.cjs');
-var chunkQN4KF3BH_cjs = require('./chunk-QN4KF3BH.cjs');
-var chunkZLBRQFDD_cjs = require('./chunk-ZLBRQFDD.cjs');
-var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+var chunkLI436ITD_cjs = require('./chunk-LI436ITD.cjs');
+var chunkCCGRCYWJ_cjs = require('./chunk-CCGRCYWJ.cjs');
+var chunkPZQDCRPV_cjs = require('./chunk-PZQDCRPV.cjs');
+var chunk2KZFMI6P_cjs = require('./chunk-2KZFMI6P.cjs');
+var chunk75ZPJI57_cjs = require('./chunk-75ZPJI57.cjs');
 var tools = require('@mastra/core/tools');
 
 // src/server/handlers/tools.ts
 var tools_exports = {};
-chunkFV45V6WC_cjs.__export(tools_exports, {
+chunk75ZPJI57_cjs.__export(tools_exports, {
   executeAgentToolHandler: () => executeAgentToolHandler,
   executeToolHandler: () => executeToolHandler,
   getToolByIdHandler: () => getToolByIdHandler,
@@ -24,8 +25,8 @@ async function getToolsHandler({ tools }) {
         const tool = _tool;
         acc[id] = {
           ...tool,
-          inputSchema: tool.inputSchema ? chunk5SWCVTNL_cjs.stringify(chunk5SWCVTNL_cjs.esm_default(tool.inputSchema)) : void 0,
-          outputSchema: tool.outputSchema ? chunk5SWCVTNL_cjs.stringify(chunk5SWCVTNL_cjs.esm_default(tool.outputSchema)) : void 0
+          inputSchema: tool.inputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(tool.inputSchema)) : void 0,
+          outputSchema: tool.outputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(tool.outputSchema)) : void 0
         };
         return acc;
       },
@@ -33,23 +34,23 @@ async function getToolsHandler({ tools }) {
     );
     return serializedTools;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting tools");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting tools");
   }
 }
 async function getToolByIdHandler({ tools, toolId }) {
   try {
     const tool = Object.values(tools || {}).find((tool2) => tool2.id === toolId);
     if (!tool) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Tool not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Tool not found" });
     }
     const serializedTool = {
       ...tool,
-      inputSchema: tool.inputSchema ? chunk5SWCVTNL_cjs.stringify(chunk5SWCVTNL_cjs.esm_default(tool.inputSchema)) : void 0,
-      outputSchema: tool.outputSchema ? chunk5SWCVTNL_cjs.stringify(chunk5SWCVTNL_cjs.esm_default(tool.outputSchema)) : void 0
+      inputSchema: tool.inputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(tool.inputSchema)) : void 0,
+      outputSchema: tool.outputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(tool.outputSchema)) : void 0
     };
     return serializedTool;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting tool");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting tool");
   }
 }
 function executeToolHandler(tools$1) {
@@ -62,16 +63,16 @@ function executeToolHandler(tools$1) {
   }) => {
     try {
       if (!toolId) {
-        throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Tool ID is required" });
+        throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Tool ID is required" });
       }
       const tool = Object.values(tools$1 || {}).find((tool2) => tool2.id === toolId);
       if (!tool) {
-        throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Tool not found" });
+        throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Tool not found" });
       }
       if (!tool?.execute) {
-        throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Tool is not executable" });
+        throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Tool is not executable" });
       }
-      chunkQN4KF3BH_cjs.validateBody({ data });
+      chunkCCGRCYWJ_cjs.validateBody({ data });
       if (tools.isVercelTool(tool)) {
         const result2 = await tool.execute(data);
         return result2;
@@ -84,7 +85,7 @@ function executeToolHandler(tools$1) {
       });
       return result;
     } catch (error) {
-      return chunkZLBRQFDD_cjs.handleError(error, "Error executing tool");
+      return chunkPZQDCRPV_cjs.handleError(error, "Error executing tool");
     }
   };
 }
@@ -98,14 +99,15 @@ async function executeAgentToolHandler({
   try {
     const agent = agentId ? mastra.getAgent(agentId) : null;
     if (!agent) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Tool not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Tool not found" });
     }
-    const tool = Object.values(agent?.tools || {}).find((tool2) => tool2.id === toolId);
+    const agentTools = await agent.getTools({ runtimeContext });
+    const tool = Object.values(agentTools || {}).find((tool2) => tool2.id === toolId);
     if (!tool) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Tool not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Tool not found" });
     }
     if (!tool?.execute) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Tool is not executable" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Tool is not executable" });
     }
     const result = await tool.execute({
       context: data,
@@ -115,7 +117,7 @@ async function executeAgentToolHandler({
     });
     return result;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error executing tool");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error executing tool");
   }
 }