File size: 1,522 Bytes
96d97a7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
from .nodes import PromptRelayEncode, PromptRelayEncodeTimeline, PromptRelayAdvancedOptions
from .smart_nodes import PromptRelaySmartEncode, PromptRelaySmartEncodeTest
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 [
PromptRelayEncode,
PromptRelayEncodeTimeline,
PromptRelaySmartEncode,
PromptRelaySmartEncodeTest,
PromptRelayAdvancedOptions
]
async def comfy_entrypoint() -> PromptRelay:
return PromptRelay()
NODE_CLASS_MAPPINGS = {
"PromptRelayEncode": PromptRelayEncode,
"PromptRelayEncodeTimeline": PromptRelayEncodeTimeline,
"PromptRelaySmartEncode": PromptRelaySmartEncode,
"PromptRelaySmartEncodeTest": PromptRelaySmartEncodeTest,
"PromptRelayAdvancedOptions": PromptRelayAdvancedOptions
}
NODE_DISPLAY_NAME_MAPPINGS = {
"PromptRelayEncode": "Prompt Relay Encode",
"PromptRelayEncodeTimeline": "Prompt Relay Encode (Timeline)",
"PromptRelaySmartEncode": "Prompt Relay Encode (Smart)",
"PromptRelaySmartEncodeTest": "Prompt Relay Smart Encode Test",
"PromptRelayAdvancedOptions": "Prompt Relay Advanced Options"
}
WEB_DIRECTORY = "./web"
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|