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

[Refactor] Use node def v2 in registerNodeDef (#2856)

Browse Source
This commit is contained in:
Chenlei Hu
2025-03-04 12:07:13 -05:00
committed by GitHub
parent fe5964ceb6
commit 2d179ad632
5 changed files with 84 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/scripts/app.ts
View File

@@ -19,9 +19,11 @@ import {
type NodeId,
validateComfyWorkflow
} from '@/schemas/comfyWorkflowSchema'
import { transformNodeDefV1ToV2 } from '@/schemas/nodeDef/migration'
import {
type ComfyNodeDef as ComfyNodeDefV2,
isComboInputSpec
isComboInputSpec,
isComfyNodeDef as isComfyNodeDefV2
} from '@/schemas/nodeDef/nodeDefSchemaV2'
import type { ComfyNodeDef as ComfyNodeDefV1 } from '@/schemas/nodeDefSchema'
import { getFromWebmFile } from '@/scripts/metadata/ebml'
@@ -944,16 +946,19 @@ export class ComfyApp {
}
}
async registerNodeDef(
nodeId: string,
nodeDef: ComfyNodeDefV1 & ComfyNodeDefV2
) {
return await useLitegraphService().registerNodeDef(nodeId, nodeDef)
async registerNodeDef(nodeId: string, nodeDef: ComfyNodeDefV1) {
return await useLitegraphService().registerNodeDef(
nodeId,
isComfyNodeDefV2(nodeDef)
? nodeDef
: {
...(nodeDef as ComfyNodeDefV1),
...transformNodeDefV1ToV2(nodeDef)
}
)
}
async registerNodesFromDefs(
defs: Record<string, ComfyNodeDefV1 & ComfyNodeDefV2>
) {
async registerNodesFromDefs(defs: Record<string, ComfyNodeDefV1>) {
await useExtensionService().invokeExtensionsAsync('addCustomNodeDefs', defs)
// Register a node for each definition