from comfy_api.latest import ComfyExtension, io
from typing_extensions import override

from .minimax_h3_audio_preview import MiniMaxH3AudioPreview


class MiniMaxH3PreviewExtension(ComfyExtension):
    @override
    async def get_node_list(self) -> list[type[io.ComfyNode]]:
        return [MiniMaxH3AudioPreview]


async def comfy_entrypoint() -> MiniMaxH3PreviewExtension:
    return MiniMaxH3PreviewExtension()


NODE_CLASS_MAPPINGS = {
    "MiniMaxH3AudioPreviewCS": MiniMaxH3AudioPreview,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "MiniMaxH3AudioPreviewCS": "MiniMax H3 Audio Preview",
}

WEB_DIRECTORY = "./js"

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]