@mastra/mcp 1.3.0 → 1.3.1-alpha.1

package/dist/index.js

@@ -409,6 +409,7 @@ var InternalMastraMCPClient = class extends MastraBase {
   operationContextStore = new AsyncLocalStorage();
   exitHookUnsubscribe;
   sigTermHandler;
+  sigHupHandler;
   _roots;
   /** Provides access to resource operations (list, read, subscribe, etc.) */
   resources;
@@ -667,6 +668,10 @@ var InternalMastraMCPClient = class extends MastraBase {
       this.sigTermHandler = () => gracefulExit();
       process.on("SIGTERM", this.sigTermHandler);
     }
+    if (!this.sigHupHandler) {
+      this.sigHupHandler = () => gracefulExit();
+      process.on("SIGHUP", this.sigHupHandler);
+    }
     this.log("debug", `Successfully connected to MCP server`);
     return this.isConnected;
   }
@@ -723,6 +728,10 @@ var InternalMastraMCPClient = class extends MastraBase {
         process.off("SIGTERM", this.sigTermHandler);
         this.sigTermHandler = void 0;
       }
+      if (this.sigHupHandler) {
+        process.off("SIGHUP", this.sigHupHandler);
+        this.sigHupHandler = void 0;
+      }
     }
   }
   /**
@@ -887,34 +896,6 @@ var InternalMastraMCPClient = class extends MastraBase {
       });
     }
   }
-  async convertOutputSchema(outputSchema) {
-    if (!outputSchema) return;
-    try {
-      await $RefParser.dereference(outputSchema);
-      return "jsonSchema" in outputSchema ? outputSchema.jsonSchema : outputSchema;
-    } catch (error) {
-      let errorDetails;
-      if (error instanceof Error) {
-        errorDetails = error.stack;
-      } else {
-        try {
-          errorDetails = JSON.stringify(error);
-        } catch {
-          errorDetails = String(error);
-        }
-      }
-      this.log("error", "Failed to dereference JSON schema", {
-        error: errorDetails,
-        originalJsonSchema: outputSchema
-      });
-      throw new MastraError({
-        id: "MCP_TOOL_OUTPUT_SCHEMA_CONVERSION_FAILED",
-        domain: ErrorDomain.MCP,
-        category: ErrorCategory.USER,
-        details: { error: errorDetails ?? "Unknown error" }
-      });
-    }
-  }
   async tools() {
     this.log("debug", `Requesting tools from MCP server`);
     const { tools } = await this.client.listTools({}, { timeout: this.timeout });
@@ -926,7 +907,10 @@ var InternalMastraMCPClient = class extends MastraBase {
           id: `${this.name}_${tool.name}`,
           description: tool.description || "",
           inputSchema: await this.convertInputSchema(tool.inputSchema),
-          outputSchema: await this.convertOutputSchema(tool.outputSchema),
+          // Don't pass outputSchema to createTool — the MCP SDK's Client.callTool()
+          // already validates structuredContent against the tool's outputSchema using AJV.
+          // Passing it here causes Zod to strip unrecognized keys from the CallToolResult
+          // envelope, returning {} for tools without structuredContent.
           mcpMetadata: {
             serverName: this.name,
             serverVersion: this.client.getServerVersion()?.version
@@ -953,16 +937,6 @@ var InternalMastraMCPClient = class extends MastraBase {
                 if (res.structuredContent !== void 0) {
                   return res.structuredContent;
                 }
-                if (tool.outputSchema && !res.isError) {
-                  const content = res.content;
-                  if (content && content.length === 1 && content[0].type === "text" && content[0].text !== void 0) {
-                    try {
-                      return JSON.parse(content[0].text);
-                    } catch {
-                      return content[0].text;
-                    }
-                  }
-                }
                 return res;
               };
               try {
@@ -2814,9 +2788,16 @@ Provided arguments: ${JSON.stringify(request.params.arguments, null, 2)}`
             return this.handleElicitationRequest(request2, serverInstance, options);
           }
         };
+        const proxiedContext = new RequestContext();
+        if (extra) {
+          Object.entries(extra).forEach(([key, value]) => {
+            proxiedContext.set(key, value);
+          });
+        }
         const mcpOptions = {
           messages: [],
           toolCallId: "",
+          requestContext: proxiedContext,
           // Pass MCP-specific context through the mcp property
           mcp: {
             elicitation: sessionElicitation,