From 9ed3545b9549210a74aae900718e020e7ebc0592 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Fri, 20 Sep 2024 19:45:34 +0900
Subject: [PATCH] Fix frontend node conflicting with node badge (#895)

---
 src/extensions/core/nodeBadge.ts | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/extensions/core/nodeBadge.ts b/src/extensions/core/nodeBadge.ts
index a1b2a5b71..9a3ae5e1e 100644
--- a/src/extensions/core/nodeBadge.ts
+++ b/src/extensions/core/nodeBadge.ts
@@ -15,6 +15,10 @@ import { useNodeDefStore } from '@/stores/nodeDefStore'
 function getNodeSource(node: ComfyLGraphNode): NodeSource | null {
   const nodeDef = (node.constructor as typeof ComfyLGraphNode)
     .nodeData as ComfyNodeDef
+  // Frontend-only nodes don't have nodeDef
+  if (!nodeDef) {
+    return null
+  }
   const nodeDefStore = useNodeDefStore()
   return nodeDefStore.nodeDefsByName[nodeDef.name]?.nodeSource ?? null
 }