diff --git a/browser_tests/tests/nodeBadge.spec.ts-snapshots/node-badge-Hide-built-in-chromium-linux.png b/browser_tests/tests/nodeBadge.spec.ts-snapshots/node-badge-Hide-built-in-chromium-linux.png index 7a2293ae155..467bce51feb 100644 Binary files a/browser_tests/tests/nodeBadge.spec.ts-snapshots/node-badge-Hide-built-in-chromium-linux.png and b/browser_tests/tests/nodeBadge.spec.ts-snapshots/node-badge-Hide-built-in-chromium-linux.png differ diff --git a/src/stores/nodeDefStore.ts b/src/stores/nodeDefStore.ts index 754c432f49c..df6c4cfa9c1 100644 --- a/src/stores/nodeDefStore.ts +++ b/src/stores/nodeDefStore.ts @@ -23,7 +23,7 @@ import { useSettingStore } from '@/platform/settings/settingStore' import { NodeSearchService } from '@/services/nodeSearchService' import { useSubgraphStore } from '@/stores/subgraphStore' import { - NodeSourceType, + CORE_NODE_MODULES, getEssentialsCategory, getNodeSource } from '@/types/nodeSource' @@ -91,6 +91,7 @@ export class ComfyNodeDefImpl readonly essentials_category?: string /** Whether the blueprint is a global/installed blueprint (not user-created). */ readonly isGlobal?: boolean + readonly isCoreNode: boolean // V2 fields readonly inputs: Record @@ -168,6 +169,9 @@ export class ComfyNodeDefImpl obj.essentials_category ) this.isGlobal = obj.isGlobal + this.isCoreNode = CORE_NODE_MODULES.includes( + this.python_module.split('.')[0] + ) // Initialize V2 fields const defV2 = transformNodeDefV1ToV2(obj) @@ -197,10 +201,6 @@ export class ComfyNodeDefImpl return [scores[0], -nodeFrequency, ...scores.slice(1)] } - get isCoreNode(): boolean { - return this.nodeSource.type === NodeSourceType.Core - } - get nodeLifeCycleBadgeText(): string { if (this.deprecated) return '[DEPR]' if (this.experimental) return '[BETA]' diff --git a/src/types/nodeSource.ts b/src/types/nodeSource.ts index 333fd7bf8cc..77fd7c6dbeb 100644 --- a/src/types/nodeSource.ts +++ b/src/types/nodeSource.ts @@ -5,6 +5,7 @@ export enum NodeSourceType { Essentials = 'essentials', Unknown = 'unknown' } +export const CORE_NODE_MODULES = ['nodes', 'comfy_extras', 'comfy_api_nodes'] export type NodeSource = { type: NodeSourceType @@ -98,9 +99,7 @@ export const getNodeSource = ( displayText: displayName, badgeText: displayName } - } else if ( - ['nodes', 'comfy_extras', 'comfy_api_nodes'].includes(modules[0]) - ) { + } else if (CORE_NODE_MODULES.includes(modules[0])) { return { type: NodeSourceType.Core, className: 'comfy-core',