Skip to content

Commit 2663361

Browse files
authored
Fix frontend only node category (#499)
1 parent 8c40f83 commit 2663361

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/scripts/app.ts

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1955,16 +1955,17 @@ export class ComfyApp {
19551955
this.canvas?.draw(true, true)
19561956
}
19571957

1958-
updateVueAppNodeDefs(defs: Record<string, ComfyNodeDef>) {
1958+
private updateVueAppNodeDefs(defs: Record<string, ComfyNodeDef>) {
19591959
// Frontend only nodes registered by custom nodes.
19601960
// Example: https://github.com/rgthree/rgthree-comfy/blob/dd534e5384be8cf0c0fa35865afe2126ba75ac55/src_web/comfyui/fast_groups_bypasser.ts#L10
19611961
const rawDefs = Object.fromEntries(
1962-
Object.entries(LiteGraph.registered_node_types).map(([name, _]) => [
1962+
Object.entries(LiteGraph.registered_node_types).map(([name, node]) => [
19631963
name,
19641964
{
19651965
name,
19661966
display_name: name,
1967-
category: '__frontend_only__',
1967+
// @ts-expect-error
1968+
category: node.category || '__frontend_only__',
19681969
input: { required: {}, optional: {} },
19691970
output: [],
19701971
output_name: [],

0 commit comments

Comments
 (0)