| 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"]
|
|
|
