@mcp-use/client 2.1.2 → 2.1.3-canary.1

package/dist/index.js

@@ -1525,7 +1525,7 @@ init_connector_telemetry();
 init_logging();
 
 // src/utils/version.ts
-var VERSION = "2.1.2";
+var VERSION = "2.1.3-canary.1";
 function getPackageVersion() {
   return VERSION;
 }
@@ -2115,7 +2115,6 @@ var FileKVStore = class {
   constructor(serverUrlHash, baseDir) {
     const root = baseDir ?? join(homedir(), ".mcp-use", "oauth");
     this.dir = join(root, serverUrlHash);
-    this.ensureDir();
   }
   ensureDir() {
     if (!existsSync(this.dir)) {
@@ -2570,6 +2569,8 @@ var NodeOAuthClientProvider = class _NodeOAuthClientProvider {
   kv;
   authTimeoutMs;
   openBrowserOverride;
+  /** Whether the selected callback port still needs to be written to storage. */
+  shouldPersistSelectedPort;
   server = null;
   /** Provider authorization URL, exposed only through the local redirect route. */
   authorizationUrl = null;
@@ -2578,13 +2579,14 @@ var NodeOAuthClientProvider = class _NodeOAuthClientProvider {
   /** Latest deferred (settled or in-flight) for the loopback response. */
   lastFlow = null;
   pendingTimer = null;
-  constructor(serverUrl, port, session, kv, options) {
+  constructor(serverUrl, port, session, kv, options, shouldPersistSelectedPort) {
     this.serverUrl = serverUrl;
     this.port = port;
     this.session = session;
     this.kv = kv;
     this.authTimeoutMs = options.authTimeoutMs ?? DEFAULT_AUTH_TIMEOUT_MS2;
     this.openBrowserOverride = options.openBrowser;
+    this.shouldPersistSelectedPort = shouldPersistSelectedPort;
   }
   /**
    * Creates a Node OAuth provider and reserves a localhost callback port.
@@ -2613,16 +2615,20 @@ var NodeOAuthClientProvider = class _NodeOAuthClientProvider {
         probe.listen(0, "127.0.0.1");
       });
     }
-    if (port !== persistedPort) {
-      await kv.set("port", String(port));
-    }
     const callbackUrl = `http://127.0.0.1:${port}/callback`;
     const session = new OAuthSessionStore(
       serverUrl,
       { ...options, callbackUrl },
       kv
     );
-    return new _NodeOAuthClientProvider(serverUrl, port, session, kv, options);
+    return new _NodeOAuthClientProvider(
+      serverUrl,
+      port,
+      session,
+      kv,
+      options,
+      port !== persistedPort
+    );
   }
   // --- Identity passthroughs (parallel to BrowserOAuthClientProvider) ---
   /** Prefix used for persisted OAuth session keys. */
@@ -2737,6 +2743,10 @@ var NodeOAuthClientProvider = class _NodeOAuthClientProvider {
         "NodeOAuthClientProvider: an authorization is already in progress"
       );
     }
+    if (this.shouldPersistSelectedPort) {
+      await this.kv.set("port", String(this.port));
+      this.shouldPersistSelectedPort = false;
+    }
     const sanitizedUrl = await this.session.storeAuthorizationState(
       authorizationUrl,
       { flowType: "redirect" }
@@ -3349,23 +3359,15 @@ var HttpConnector = class extends BaseConnector {
         authProviderUrl: this.opts.authProvider && "serverUrl" in this.opts.authProvider && typeof this.opts.authProvider.serverUrl === "string" ? this.opts.authProvider.serverUrl : "none",
         headers: this.headers
       });
-      let markPushStreamReady;
-      const pushStreamReady = new Promise((resolve) => {
-        markPushStreamReady = resolve;
-      });
       const baseFetch = this.customFetch ?? globalThis.fetch.bind(globalThis);
       const observedFetch = async (input, init) => {
         const response = await baseFetch(input, init);
-        const method = init?.method ?? (input instanceof Request ? input.method : "GET");
         const requestHeaders = new Headers(
           input instanceof Request ? input.headers : void 0
         );
         new Headers(init?.headers).forEach((value, key) => {
           requestHeaders.set(key, value);
         });
-        if (method.toUpperCase() === "GET" && response.ok && response.headers.get("content-type")?.includes("text/event-stream")) {
-          markPushStreamReady?.();
-        }
         return requestHeaders.get("mcp-method") === "subscriptions/listen" ? response : this.observeSseProgress(response);
       };
       const streamableTransport = new StreamableHTTPClientTransport(
@@ -3404,8 +3406,9 @@ var HttpConnector = class extends BaseConnector {
       this.setupRootsHandler();
       this.setupSamplingHandler();
       this.setupElicitationHandler();
+      this.setupNotificationHandler();
       logger.debug(
-        "Roots/sampling/elicitation handlers registered before connect"
+        "Roots/sampling/elicitation/notification handlers registered before connect"
       );
       try {
         let connectTimeout;
@@ -3422,24 +3425,6 @@ var HttpConnector = class extends BaseConnector {
         ]).finally(() => {
           if (connectTimeout !== void 0) clearTimeout(connectTimeout);
         });
-        if ((this.client.getProtocolEra?.() ?? "legacy") === "legacy" && streamableTransport.sessionId) {
-          let readinessTimeout;
-          const attached = await Promise.race([
-            pushStreamReady.then(() => true),
-            new Promise(
-              (resolve) => readinessTimeout = setTimeout(
-                () => resolve(false),
-                Math.min(this.timeout, 5e3)
-              )
-            )
-          ]);
-          if (readinessTimeout) clearTimeout(readinessTimeout);
-          if (!attached) {
-            logger.warn(
-              "Legacy server push stream did not attach before connect completed"
-            );
-          }
-        }
         const sessionId2 = streamableTransport.sessionId;
         if (sessionId2) {
           logger.debug(`Session ID obtained: ${sessionId2}`);
@@ -3473,7 +3458,6 @@ var HttpConnector = class extends BaseConnector {
       };
       this.connected = true;
       this.transportType = "streamable-http";
-      this.setupNotificationHandler();
       logger.debug(
         `Successfully connected to MCP implementation via streamable HTTP: ${baseUrl}`
       );