32 lines
1.7 KiB
Python
32 lines
1.7 KiB
Python
from .py.llm_loader import NODE_CLASS_MAPPINGS as LLM_LOADER_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as LLM_LOADER_DISPLAY_MAPPINGS
|
|
from .py.image_generation import NODE_CLASS_MAPPINGS as IMAGE_GENERATION_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as IMAGE_GENERATION_DISPLAY_MAPPINGS
|
|
from .py.image_understanding import NODE_CLASS_MAPPINGS as IMAGE_UNDERSTANDING_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as IMAGE_UNDERSTANDING_DISPLAY_MAPPINGS
|
|
from .py.model_loader import NODE_CLASS_MAPPINGS as MODEL_LOADER_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as MODEL_LOADER_DISPLAY_MAPPINGS
|
|
from .py.openai_compatible_loader import NODE_CLASS_MAPPINGS as OPENAI_COMPATIBLE_LOADER_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as OPENAI_COMPATIBLE_LOADER_DISPLAY_MAPPINGS
|
|
from .py.image_preprocessor import NODE_CLASS_MAPPINGS as IMAGE_PREPROCESSOR_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as IMAGE_PREPROCESSOR_DISPLAY_MAPPINGS
|
|
from .py.VideoFileUploader import NODE_CLASS_MAPPINGS as VIDEO_FILE_UPLOADER_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as VIDEO_FILE_UPLOADER_DISPLAY_MAPPINGS
|
|
|
|
# 合并所有节点映射
|
|
NODE_CLASS_MAPPINGS = {
|
|
**LLM_LOADER_MAPPINGS,
|
|
**IMAGE_GENERATION_MAPPINGS,
|
|
**IMAGE_UNDERSTANDING_MAPPINGS,
|
|
**MODEL_LOADER_MAPPINGS,
|
|
**OPENAI_COMPATIBLE_LOADER_MAPPINGS,
|
|
**IMAGE_PREPROCESSOR_MAPPINGS,
|
|
**VIDEO_FILE_UPLOADER_MAPPINGS,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**LLM_LOADER_DISPLAY_MAPPINGS,
|
|
**IMAGE_GENERATION_DISPLAY_MAPPINGS,
|
|
**IMAGE_UNDERSTANDING_DISPLAY_MAPPINGS,
|
|
**MODEL_LOADER_DISPLAY_MAPPINGS,
|
|
**OPENAI_COMPATIBLE_LOADER_DISPLAY_MAPPINGS,
|
|
**IMAGE_PREPROCESSOR_DISPLAY_MAPPINGS,
|
|
**VIDEO_FILE_UPLOADER_DISPLAY_MAPPINGS,
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|