Merge pull request #221 from Anson2048/patch-1

fix: handle malformed node IDs in ComfyUI WebSocket executing events

Author: Weixuanf

server/routers/comfyui_execution.py

@@ -270,16 +270,19 @@ async def on_executing(self, data):
             if self.current_node:
                 self.remaining_nodes.discard(self.current_node)
                 self.update_overall_progress()
-            self.current_node = data["node"]
-            self.log_node("Executing", data["node"])
+            # Use display_node if available, otherwise use node
+            node_id = data.get("display_node", data.get('node'))
+            
+            self.current_node = node_id
+            self.log_node("Executing", node_id)
             if self.ctx.get("session_id"):
                 await send_to_websocket(
                     self.ctx.get("session_id"),
                     {
                         "type": "tool_call_progress",
                         "tool_call_id": self.ctx.get("tool_call_id"),
                         "session_id": self.ctx.get("session_id"),
-                        "update": f"Executing {self.get_node_title(data['node'])}",
+                        "update": f"Executing {self.get_node_title(node_id)}",
                     },
                 )
         return True