Update __init__.py

This commit is contained in:
rikanrino
2026-05-13 21:16:03 +08:00
committed by GitHub
parent 0241c25c94
commit af53f1122d
-20
View File
@@ -1,27 +1,7 @@
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
from .nodes import RikanPromptRelayEncodeTimeline
from comfy_api.latest import ComfyExtension, io
from typing_extensions import override
class PromptRelay(ComfyExtension):
@override
async def get_node_list(self) -> list[type[io.ComfyNode]]:
return [
RikanPromptRelayEncodeTimeline,
]
async def comfy_entrypoint() -> PromptRelay:
return PromptRelay()
NODE_CLASS_MAPPINGS = {
"RikanPromptRelayEncodeTimeline": RikanPromptRelayEncodeTimeline,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"RikanPromptRelayEncodeTimeline": "Rikan Prompt Relay Encode (Timeline)",
}
WEB_DIRECTORY = "./web"
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]