@mcp-use/client 2.2.0 → 2.2.1

package/dist/index.js

@@ -1526,7 +1526,7 @@ init_connector_telemetry();
 init_logging();
 
 // src/utils/version.ts
-var VERSION = "2.2.0";
+var VERSION = "2.2.1";
 function getPackageVersion() {
   return VERSION;
 }
@@ -4143,6 +4143,12 @@ var BaseMCPClient = class {
    * Map of server names to their active sessions.
    */
   sessions = {};
+  /**
+   * Tracks teardown by session identity so overlapping cleanup paths share the
+   * same disconnect. In particular, createSession() can replace a session while
+   * closeSession() is already disconnecting it.
+   */
+  sessionDisconnects = /* @__PURE__ */ new WeakMap();
   /**
    * List of server names that have active sessions.
    * This array is kept in sync with the sessions map and can be used
@@ -4176,6 +4182,15 @@ var BaseMCPClient = class {
       this.config = config;
     }
   }
+  disconnectSession(session) {
+    const existingDisconnect = this.sessionDisconnects.get(session);
+    if (existingDisconnect) {
+      return existingDisconnect;
+    }
+    const disconnect = Promise.resolve().then(() => session.disconnect());
+    this.sessionDisconnects.set(session, disconnect);
+    return disconnect;
+  }
   /**
    * Creates a client instance from a configuration dictionary.
    *
@@ -4321,7 +4336,9 @@ var BaseMCPClient = class {
    * lifecycle of connections and provide methods for calling tools, listing
    * resources, and more.
    *
-   * If a session already exists for the server, it will be replaced with a new one.
+   * If a session already exists for the server, it will be replaced with a new
+   * one and the previous session is disconnected; any previously returned
+   * reference to it becomes unusable.
    *
    * @param serverName - The name of the server as defined in the client configuration
    * @param autoInitialize - Whether to automatically initialize the session (default: true)
@@ -4394,10 +4411,21 @@ var BaseMCPClient = class {
       await completeOAuthFlow(oauthProvider, httpConfig.url);
       session = await openSession();
     }
+    const previous = this.sessions[serverName];
     this.sessions[serverName] = session;
     if (!this.activeSessions.includes(serverName)) {
       this.activeSessions.push(serverName);
     }
+    if (previous && previous !== session) {
+      try {
+        logger.debug(`Disconnecting replaced session for server ${serverName}`);
+        await this.disconnectSession(previous);
+      } catch (e) {
+        logger.error(
+          `Error disconnecting replaced session for server '${serverName}': ${e}`
+        );
+      }
+    }
     return session;
   }
   /**
@@ -4576,7 +4604,7 @@ var BaseMCPClient = class {
     }
     try {
       logger.debug(`Closing session for server ${serverName}`);
-      await session.disconnect();
+      await this.disconnectSession(session);
     } catch (e) {
       logger.error(`Error closing session for server '${serverName}': ${e}`);
     } finally {