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

[Bug] Identify api nodes as core nodes (#3683)

Browse Source
This commit is contained in:
Chenlei Hu
2025-04-29 11:45:52 -04:00
committed by GitHub
parent 1531025070
commit a2f04844e0
2 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/types/nodeSource.ts
View File

@@ -29,7 +29,7 @@ export const getNodeSource = (python_module?: string): NodeSource => {
return UNKNOWN_NODE_SOURCE return UNKNOWN_NODE_SOURCE
} }
const modules = python_module.split('.') const modules = python_module.split('.')
if (['nodes', 'comfy_extras'].includes(modules[0])) { if (['nodes', 'comfy_extras', 'comfy_api_nodes'].includes(modules[0])) {
return { return {
type: NodeSourceType.Core, type: NodeSourceType.Core,
className: 'comfy-core', className: 'comfy-core',

10
tests-ui/tests/nodeSource.test.ts
View File

@@ -33,6 +33,16 @@ describe('getNodeSource', () => {
}) })
}) })
it('should identify core nodes from comfy_api_nodes module', () => {
const result = getNodeSource('comfy_api_nodes.some_module')
expect(result).toEqual({
type: NodeSourceType.Core,
className: 'comfy-core',
displayText: 'Comfy Core',
badgeText: '🦊'
})
})
it('should identify custom nodes and format their names', () => { it('should identify custom nodes and format their names', () => {
const result = getNodeSource('custom_nodes.ComfyUI-Example') const result = getNodeSource('custom_nodes.ComfyUI-Example')
expect(result).toEqual({ expect(result).toEqual({