@mastra/mcp 1.0.0-beta.0 → 1.0.0-beta.2

package/dist/index.js

@@ -366,6 +366,7 @@ var ResourceClientActions = class {
 };
 
 // src/client/client.ts
+var DEFAULT_SERVER_CONNECT_TIMEOUT_MSEC = 3e3;
 function convertLogLevelToLoggerMethod(level) {
   switch (level) {
     case "debug":
@@ -483,7 +484,7 @@ var InternalMastraMCPClient = class extends MastraBase {
     }
   }
   async connectHttp(url) {
-    const { requestInit, eventSourceInit, authProvider } = this.serverConfig;
+    const { requestInit, eventSourceInit, authProvider, connectTimeout } = this.serverConfig;
     this.log("debug", `Attempting to connect to URL: ${url}`);
     let shouldTrySSE = url.pathname.endsWith(`/sse`);
     if (!shouldTrySSE) {
@@ -495,10 +496,7 @@ var InternalMastraMCPClient = class extends MastraBase {
           authProvider
         });
         await this.client.connect(streamableTransport, {
-          timeout: (
-            // this is hardcoded to 3s because the long default timeout would be extremely slow for sse backwards compat (60s)
-            3e3
-          )
+          timeout: connectTimeout ?? DEFAULT_SERVER_CONNECT_TIMEOUT_MSEC
         });
         this.transport = streamableTransport;
         this.log("debug", "Successfully connected using Streamable HTTP transport.");
@@ -1580,7 +1578,7 @@ To fix this you have three different options:
   }
 };
 
-// ../../node_modules/.pnpm/hono@4.10.3/node_modules/hono/dist/utils/stream.js
+// ../../node_modules/.pnpm/hono@4.10.6/node_modules/hono/dist/utils/stream.js
 var StreamingApi = class {
   writer;
   encoder;
@@ -1647,7 +1645,7 @@ var StreamingApi = class {
   }
 };
 
-// ../../node_modules/.pnpm/hono@4.10.3/node_modules/hono/dist/helper/streaming/utils.js
+// ../../node_modules/.pnpm/hono@4.10.6/node_modules/hono/dist/helper/streaming/utils.js
 var isOldBunVersion = () => {
   const version = typeof Bun !== "undefined" ? Bun.version : void 0;
   if (version === void 0) {
@@ -1658,7 +1656,7 @@ var isOldBunVersion = () => {
   return result;
 };
 
-// ../../node_modules/.pnpm/hono@4.10.3/node_modules/hono/dist/utils/html.js
+// ../../node_modules/.pnpm/hono@4.10.6/node_modules/hono/dist/utils/html.js
 var HtmlEscapedCallbackPhase = {
   Stringify: 1};
 var resolveCallback = async (str, phase, preserveCallbacks, context, buffer) => {
@@ -1689,7 +1687,7 @@ var resolveCallback = async (str, phase, preserveCallbacks, context, buffer) =>
   }
 };
 
-// ../../node_modules/.pnpm/hono@4.10.3/node_modules/hono/dist/helper/streaming/sse.js
+// ../../node_modules/.pnpm/hono@4.10.6/node_modules/hono/dist/helper/streaming/sse.js
 var SSEStreamingApi = class extends StreamingApi {
   constructor(writable, readable) {
     super(writable, readable);
@@ -2568,7 +2566,14 @@ Provided arguments: ${JSON.stringify(request.params.arguments, null, 2)}`
             `Executing agent tool '${agentToolName}' for agent '${agent.name}' with message: "${inputData.message}"`
           );
           try {
-            const response = await agent.generate(inputData.message, context);
+            const proxiedContext = context?.requestContext || new RequestContext();
+            if (context?.mcp?.extra) {
+              proxiedContext.set("mcp.extra", context.mcp.extra);
+            }
+            const response = await agent.generate(inputData.message, {
+              ...context ?? {},
+              requestContext: proxiedContext
+            });
             return response;
           } catch (error) {
             this.logger.error(`Error executing agent tool '${agentToolName}' for agent '${agent.name}':`, error);