@mcp-use/client 2.2.1-canary.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.1-canary.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.
    *
@@ -4404,7 +4419,7 @@ var BaseMCPClient = class {
     if (previous && previous !== session) {
       try {
         logger.debug(`Disconnecting replaced session for server ${serverName}`);
-        await previous.disconnect();
+        await this.disconnectSession(previous);
       } catch (e) {
         logger.error(
           `Error disconnecting replaced session for server '${serverName}': ${e}`
@@ -4589,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 {