From dd05570dfc83aa654a8df4b0bb99c96acdb47591 Mon Sep 17 00:00:00 2001
From: malventano <malventano@users.noreply.github.com>
Date: Wed, 20 May 2026 23:24:59 -0400
Subject: [PATCH] fix: allow resetTimeoutOnProgress without onprogress callback

---
 src/shared/protocol.ts | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/shared/protocol.ts b/src/shared/protocol.ts
index 2637be65bc..1cad645030 100644
--- a/src/shared/protocol.ts
+++ b/src/shared/protocol.ts
@@ -857,12 +857,6 @@ export abstract class Protocol<SendRequestT extends Request, SendNotificationT e
         const { progressToken, ...params } = notification.params;
         const messageId = Number(progressToken);
 
-        const handler = this._progressHandlers.get(messageId);
-        if (!handler) {
-            this._onerror(new Error(`Received a progress notification for an unknown token: ${JSON.stringify(notification)}`));
-            return;
-        }
-
         const responseHandler = this._responseHandlers.get(messageId);
         const timeoutInfo = this._timeoutInfo.get(messageId);
 
@@ -874,11 +868,21 @@ export abstract class Protocol<SendRequestT extends Request, SendNotificationT e
                 this._responseHandlers.delete(messageId);
                 this._progressHandlers.delete(messageId);
                 this._cleanupTimeout(messageId);
+                const handler = this._progressHandlers.get(messageId);
+                if (handler) {
+                    handler(params);
+                }
                 responseHandler(error as Error);
                 return;
             }
         }
 
+        const handler = this._progressHandlers.get(messageId);
+        if (!handler) {
+            this._onerror(new Error(`Received a progress notification for an unknown token: ${JSON.stringify(notification)}`));
+            return;
+        }
+
         handler(params);
     }