From 8151f22366c2c27d98c74611c0355058bf58fcd9 Mon Sep 17 00:00:00 2001
From: rgthree <regis.gaughan@gmail.com>
Date: Thu, 11 Apr 2024 20:13:45 -0400
Subject: [PATCH] Return original call to app.loadGraphData (so awaiters can
 await). Fixes #193

---
 src_web/comfyui/rgthree.ts | 2 +-
 web/comfyui/rgthree.js     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src_web/comfyui/rgthree.ts b/src_web/comfyui/rgthree.ts
index f756935..9034f23 100644
--- a/src_web/comfyui/rgthree.ts
+++ b/src_web/comfyui/rgthree.ts
@@ -655,7 +655,7 @@ class Rgthree extends EventTarget {
           }, 5000);
         }
       }, 100);
-      loadGraphData && loadGraphData.call(app, ...arguments);
+      return loadGraphData && loadGraphData.call(app, ...arguments);
     };
   }
 
diff --git a/web/comfyui/rgthree.js b/web/comfyui/rgthree.js
index 8be1d78..64a093d 100644
--- a/web/comfyui/rgthree.js
+++ b/web/comfyui/rgthree.js
@@ -473,7 +473,7 @@ class Rgthree extends EventTarget {
                     }, 5000);
                 }
             }, 100);
-            loadGraphData && loadGraphData.call(app, ...arguments);
+            return loadGraphData && loadGraphData.call(app, ...arguments);
         };
     }
     getNodeFromInitialGraphToPromptSerializedWorkflowBecauseComfyUIBrokeStuff(node) {