Fix frontend node conflicting with node badge (#895)

This commit is contained in:
Chenlei Hu
2024-09-20 19:45:34 +09:00
committed by GitHub
parent d223f3865b
commit 9ed3545b95

View File

@@ -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
}