from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS

try:
    from .style_nodes import (
        NODE_CLASS_MAPPINGS as STYLE_CLASS_MAPPINGS,
        NODE_DISPLAY_NAME_MAPPINGS as STYLE_DISPLAY_NAME_MAPPINGS,
    )

    NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **STYLE_CLASS_MAPPINGS}
    NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **STYLE_DISPLAY_NAME_MAPPINGS}
except ImportError:  # style stream needs full ComfyUI (folder_paths)
    pass

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]