fix(runtime): comfyui package not load comfyui-legacy nodes (#68)

Author: Chaoses-Ib

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "comfy-script"
-version = "0.5.0"
+version = "0.5.1"
 description = "A Python frontend and library for ComfyUI"
 readme = "README.md"
 # ComfyUI: >=3.8

src/comfy_script/runtime/__init__.py

@@ -270,8 +270,10 @@ def setup_comfyui_polyfills(main_locals: dict = None):
         #     for name in 'loop', 'server', 'q', 'extra_model_paths_config_path':
         #         setattr(main, name, main_locals[name])
 
+        import comfy.nodes.base_nodes
         import comfy.nodes.common
         nodes = types.ModuleType('nodes')
+        nodes.__dict__.update(comfy.nodes.base_nodes.__dict__)
         exported_nodes = getattr(server, 'nodes', None)
         if exported_nodes is None:
             exported_nodes = comfy.cmd.server.nodes
@@ -383,6 +385,8 @@ def enable_args_parsing_hook():
             else:
                 main.main()
             del main.exit
+
+            main.init_custom_nodes()
 
         asyncio.get_event_loop_policy().set_event_loop(original_loop)