Skip to content

Commit 034fac7

Browse files
authored
chore(api-nodes): auto-discover all nodes_*.py files to avoid merge conflicts when adding new API nodes (#11943)
1 parent a498556 commit 034fac7

File tree

1 file changed

+4
-26
lines changed

1 file changed

+4
-26
lines changed

nodes.py

Lines changed: 4 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import os
66
import sys
77
import json
8+
import glob
89
import hashlib
910
import inspect
1011
import traceback
@@ -2384,35 +2385,12 @@ async def init_builtin_extra_nodes():
23842385

23852386
async def init_builtin_api_nodes():
23862387
api_nodes_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), "comfy_api_nodes")
2387-
api_nodes_files = [
2388-
"nodes_ideogram.py",
2389-
"nodes_openai.py",
2390-
"nodes_minimax.py",
2391-
"nodes_veo2.py",
2392-
"nodes_kling.py",
2393-
"nodes_bfl.py",
2394-
"nodes_bytedance.py",
2395-
"nodes_ltxv.py",
2396-
"nodes_luma.py",
2397-
"nodes_recraft.py",
2398-
"nodes_pixverse.py",
2399-
"nodes_stability.py",
2400-
"nodes_runway.py",
2401-
"nodes_sora.py",
2402-
"nodes_topaz.py",
2403-
"nodes_tripo.py",
2404-
"nodes_meshy.py",
2405-
"nodes_moonvalley.py",
2406-
"nodes_rodin.py",
2407-
"nodes_gemini.py",
2408-
"nodes_vidu.py",
2409-
"nodes_wan.py",
2410-
]
2388+
api_nodes_files = sorted(glob.glob(os.path.join(api_nodes_dir, "nodes_*.py")))
24112389

24122390
import_failed = []
24132391
for node_file in api_nodes_files:
2414-
if not await load_custom_node(os.path.join(api_nodes_dir, node_file), module_parent="comfy_api_nodes"):
2415-
import_failed.append(node_file)
2392+
if not await load_custom_node(node_file, module_parent="comfy_api_nodes"):
2393+
import_failed.append(os.path.basename(node_file))
24162394

24172395
return import_failed
24182396

0 commit comments

Comments
 (0)