Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-02-01 05:49:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Refactor] Use V2 node def in ComfyApp (#2854)

Browse Source
This commit is contained in:
Chenlei Hu
2025-03-04 11:14:14 -05:00
committed by GitHub
parent f434610979
commit fe5964ceb6
3 changed files with 101 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/stores/nodeDefStore.ts
View File

@@ -228,17 +228,17 @@ export const useNodeDefStore = defineStore('nodeDef', () => {
function updateNodeDefs(nodeDefs: ComfyNodeDefV1[]) {
const newNodeDefsByName: Record<string, ComfyNodeDefImpl> = {}
const newNodeDefsByDisplayName: Record<string, ComfyNodeDefImpl> = {}
for (const nodeDef of nodeDefs) {
try {
const nodeDefImpl = new ComfyNodeDefImpl(nodeDef)
newNodeDefsByName[nodeDef.name] = nodeDefImpl
newNodeDefsByDisplayName[nodeDef.display_name] = nodeDefImpl
} catch (e) {
// Avoid breaking the app for invalid nodeDefs
// NodeDef validation is now optional for performance reasons
console.error('Error adding nodeDef:', e)
}
const nodeDefImpl =
nodeDef instanceof ComfyNodeDefImpl
? nodeDef
: new ComfyNodeDefImpl(nodeDef)
newNodeDefsByName[nodeDef.name] = nodeDefImpl
newNodeDefsByDisplayName[nodeDef.display_name] = nodeDefImpl
}
nodeDefsByName.value = newNodeDefsByName
nodeDefsByDisplayName.value = newNodeDefsByDisplayName
}