38 lines
835 B
Python
38 lines
835 B
Python
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
|
|
import importlib
|
|
|
|
modules = [
|
|
"flux_kontext_replicate",
|
|
"gemini_node",
|
|
"gemini_diarisation",
|
|
"gpt_image_edit",
|
|
"imagen",
|
|
"imagen_edit",
|
|
"veo",
|
|
"veo_api",
|
|
"nano_banana",
|
|
"gemini_tts",
|
|
"elevenlabs_tts",
|
|
"flux2_replicate",
|
|
"openai_tts",
|
|
"openai_node",
|
|
"sora",
|
|
"cleanup",
|
|
"tripoTextToModel",
|
|
"tripoImageToModel",
|
|
"grok",
|
|
"groq_node",
|
|
"groq_orpheus",
|
|
]
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
for module_name in modules:
|
|
module = importlib.import_module(f".{module_name}", package=__name__)
|
|
NODE_CLASS_MAPPINGS.update(module.NODE_CLASS_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS)
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] |