comfyui-mcp 0.52.58 → 0.52.59

package/dist/orchestrator/panel-tools.js

@@ -1303,6 +1303,8 @@ const RECONNECT_WAIT_MAX_MS = 60_000;
  *  healthy (panel#654): `panel_graph_outline` reported Connected: none and only a
  *  hard refresh restored the tab. 35s covers those recoveries; the 60s ceiling stays. */
 const RECONNECT_WAIT_DEFAULT_S = 35;
+/** Fresh original-tab hello must persist this long before graph tools are ready (#2067). */
+const POST_RESTART_RECONNECT_STABLE_MS = 2000;
 function reconnectWaitTiming() {
     if (reconnectWaitTimingOverride)
         return reconnectWaitTimingOverride;
@@ -8770,6 +8772,11 @@ export function makePanelToolCtx(bridge, tabId, workflowTargets, onRunTicketOpen
     // fresh hello can keep the same tab/socket id or arrive under a new one; in the
     // latter case only rebind after the old target is gone, preserving strict
     // multi-tab routing and never guessing away from a still-live pre-restart tab.
+    //
+    // #2067 — the first newer generation is not durable. A ComfyUI restart can
+    // hello, then drop the socket a moment later; returning true there reports
+    // graph_tools_ready while the next graph call sees Connected: none. The same
+    // identity must still be present after a short stability window.
     const awaitPostRestartReachable = async (before, budgetMs) => {
         // The actual UiBridge exposes a tab-session binding. Preserve historical
         // lightweight/mock-context behavior only when that capability does not exist
@@ -8781,14 +8788,24 @@ export function makePanelToolCtx(bridge, tabId, workflowTargets, onRunTicketOpen
         const timing = reconnectWaitTiming();
         const budget = Math.max(0, budgetMs != null ? Math.min(budgetMs, timing.budgetMs) : timing.budgetMs);
         const deadline = Date.now() + budget;
+        const stableMs = Math.max(0, timing.stableMs ?? POST_RESTART_RECONNECT_STABLE_MS);
         const isOriginalTabReconnected = () => {
             const current = panelConnectionIdentity();
             return (current != null &&
                 current.generation > before.generation &&
                 current.tabSessionId === before.tabSessionId);
         };
+        const originalTabStayedReconnected = async () => {
+            if (!isOriginalTabReconnected())
+                return false;
+            const left = deadline - Date.now();
+            if (left <= 0)
+                return false;
+            await sleep(Math.min(stableMs, left));
+            return isOriginalTabReconnected();
+        };
         for (;;) {
-            if (isOriginalTabReconnected())
+            if (await originalTabStayedReconnected())
                 return true;
             // A new tab id cannot resolve through the retired binding. Once that binding is
             // actually gone, the existing conservative rebind can follow the sole new tab.
@@ -8796,7 +8813,7 @@ export function makePanelToolCtx(bridge, tabId, workflowTargets, onRunTicketOpen
                 const interactive = interactiveTabIds() ?? [];
                 if (interactive.length > 0)
                     ensureReachable();
-                if (isOriginalTabReconnected())
+                if (await originalTabStayedReconnected())
                     return true;
             }
             const left = deadline - Date.now();