mcp-use 1.5.0-canary.0 → 1.5.0-canary.2

package/dist/{chunk-GRLCLVAK.js → chunk-UT7O4SIJ.js}

@@ -371,7 +371,11 @@ var HttpConnector = class extends BaseConnector {
           maxRetries: 2
         }
       });
-      const transport = await this.connectionManager.start();
+      let transport = await this.connectionManager.start();
+      if (this.opts.wrapTransport) {
+        const serverId = this.baseUrl;
+        transport = this.opts.wrapTransport(transport, serverId);
+      }
       this.client = new Client(this.clientInfo, this.opts.clientOptions);
       try {
         await this.client.connect(transport);
@@ -405,7 +409,11 @@ var HttpConnector = class extends BaseConnector {
           headers: this.headers
         }
       });
-      const transport = await this.connectionManager.start();
+      let transport = await this.connectionManager.start();
+      if (this.opts.wrapTransport) {
+        const serverId = this.baseUrl;
+        transport = this.opts.wrapTransport(transport, serverId);
+      }
       this.client = new Client(this.clientInfo, this.opts.clientOptions);
       await this.client.connect(transport);
       this.connected = true;
@@ -1140,15 +1148,17 @@ var BrowserMCPClient = class _BrowserMCPClient extends BaseMCPClient {
    * Supports HTTP and WebSocket connectors only
    */
   createConnectorFromConfig(serverConfig) {
-    const { url, transport, headers, authToken, authProvider } = serverConfig;
+    const { url, transport, headers, authToken, authProvider, wrapTransport } = serverConfig;
     if (!url) {
       throw new Error("Server URL is required");
     }
     const connectorOptions = {
       headers,
       authToken,
-      authProvider
+      authProvider,
       // ← Pass OAuth provider to connector
+      wrapTransport
+      // ← Pass transport wrapper if provided
     };
     if (transport === "websocket" || url.startsWith("ws://") || url.startsWith("wss://")) {
       return new WebSocketConnector(url, connectorOptions);