feat: use object info display_name as fallback before node name

This commit is contained in:
Csongor Czezar
2025-12-18 12:50:11 -08:00
parent 552452622d
commit adec2336fd
2 changed files with 152 additions and 14 deletions

View File

@@ -929,20 +929,25 @@ export class ComfyApp {
}
async getNodeDefs(): Promise<Record<string, ComfyNodeDefV1>> {
const translateNodeDef = (def: ComfyNodeDefV1): ComfyNodeDefV1 => ({
...def,
display_name: st(
`nodeDefs.${def.name}.display_name`,
def.display_name ?? def.name
),
description: def.description
? st(`nodeDefs.${def.name}.description`, def.description)
: '',
category: def.category
.split('/')
.map((category: string) => st(`nodeCategories.${category}`, category))
.join('/')
})
const translateNodeDef = (def: ComfyNodeDefV1): ComfyNodeDefV1 => {
// Use object info display_name as fallback before using name
const objectInfoDisplayName = def.display_name || def.name
return {
...def,
display_name: st(
`nodeDefs.${def.name}.display_name`,
objectInfoDisplayName
),
description: def.description
? st(`nodeDefs.${def.name}.description`, def.description)
: '',
category: def.category
.split('/')
.map((category: string) => st(`nodeCategories.${category}`, category))
.join('/')
}
}
return _.mapValues(await api.getNodeDefs(), (def) => translateNodeDef(def))
}