xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-04 05:37:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Standard CATEGORY system for api nodes (#35)

Browse Source
This commit is contained in:
Christian Byrne 2025-04-29 15:32:44 -07:00 committed by Robin Huang
parent ff5fd5d507
commit a55f29662d
2 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
comfy_api_nodes/nodes_api.py
View File

@ -381,7 +381,7 @@ class OpenAIDalle2(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,) RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call" FUNCTION = "api_call"
CATEGORY = "api node" CATEGORY = "api node/image/openai"
DESCRIPTION = cleandoc(__doc__ or "") DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True API_NODE = True
@ -522,7 +522,7 @@ class OpenAIDalle3(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,) RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call" FUNCTION = "api_call"
CATEGORY = "api node" CATEGORY = "api node/image/openai"
DESCRIPTION = cleandoc(__doc__ or "") DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True API_NODE = True
@ -654,7 +654,7 @@ class OpenAIGPTImage1(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,) RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call" FUNCTION = "api_call"
CATEGORY = "api node" CATEGORY = "api node/image/openai"
DESCRIPTION = cleandoc(__doc__ or "") DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True API_NODE = True
@ -885,7 +885,7 @@ class IdeogramTextToImage(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call" FUNCTION = "api_call"
API_NODE = True API_NODE = True
CATEGORY = "Example" CATEGORY = "api node/image/ideogram"
def api_call( def api_call(
self, self,
@ -1059,7 +1059,7 @@ class FluxProUltraImageNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call" FUNCTION = "api_call"
API_NODE = True API_NODE = True
CATEGORY = "api node" CATEGORY = "api node/image/bfl"
def api_call( def api_call(
self, self,
@ -1183,7 +1183,7 @@ class LumaReferenceNode(ComfyNodeABC):
RETURN_NAMES = ("luma_ref",) RETURN_NAMES = ("luma_ref",)
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "create_luma_reference" FUNCTION = "create_luma_reference"
CATEGORY = "api node/Luma" CATEGORY = "api node/image/Luma"
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -1228,7 +1228,7 @@ class LumaConceptsNode(ComfyNodeABC):
RETURN_NAMES = ("luma_concepts",) RETURN_NAMES = ("luma_concepts",)
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "create_concepts" FUNCTION = "create_concepts"
CATEGORY = "api node/Luma" CATEGORY = "api node/image/Luma"
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -1261,7 +1261,7 @@ class LumaImageGenerationNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call" FUNCTION = "api_call"
API_NODE = True API_NODE = True
CATEGORY = "api node" CATEGORY = "api node/image/Luma"
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -1430,7 +1430,7 @@ class LumaImageModifyNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call" FUNCTION = "api_call"
API_NODE = True API_NODE = True
CATEGORY = "api node" CATEGORY = "api node/image/Luma"
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -1538,7 +1538,7 @@ class LumaTextToVideoGenerationNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call" FUNCTION = "api_call"
API_NODE = True API_NODE = True
CATEGORY = "api node" CATEGORY = "api node/image/Luma"
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -1640,7 +1640,7 @@ class LumaImageToVideoGenerationNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call" FUNCTION = "api_call"
API_NODE = True API_NODE = True
CATEGORY = "api node" CATEGORY = "api node/image/Luma"
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -1776,7 +1776,7 @@ class RecraftStyleV3RealisticImageNode:
RETURN_TYPES = (RecraftIO.STYLEV3,) RETURN_TYPES = (RecraftIO.STYLEV3,)
RETURN_NAMES = ("recraft_style",) RETURN_NAMES = ("recraft_style",)
FUNCTION = "create_style" FUNCTION = "create_style"
CATEGORY = "api node/Recraft" CATEGORY = "api node/image/Recraft"
RECRAFT_STYLE = RecraftStyleV3.realistic_image RECRAFT_STYLE = RecraftStyleV3.realistic_image
@ -1827,7 +1827,7 @@ class RecraftTextToImageNode:
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call" FUNCTION = "api_call"
API_NODE = True API_NODE = True
CATEGORY = "api node" CATEGORY = "api node/image/Recraft"
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -1992,7 +1992,7 @@ class MinimaxTextToVideoNode:
RETURN_TYPES = ("VIDEO",) RETURN_TYPES = ("VIDEO",)
DESCRIPTION = "Generates videos from prompts using Minimax's API" DESCRIPTION = "Generates videos from prompts using Minimax's API"
FUNCTION = "generate_video" FUNCTION = "generate_video"
CATEGORY = "video" CATEGORY = "api node/video/Minimax"
API_NODE = True API_NODE = True
OUTPUT_NODE = True OUTPUT_NODE = True

2
comfy_api_nodes/nodes_runway.py
View File

@ -141,7 +141,7 @@ class RunwayImageToVideoNode(ComfyNodeABC):
RETURN_TYPES = ("VIDEO",) RETURN_TYPES = ("VIDEO",)
FUNCTION = "api_call" FUNCTION = "api_call"
CATEGORY = "api node/video/runway" CATEGORY = "api node/video/Runway"
DESCRIPTION = cleandoc(__doc__ or "") DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True API_NODE = True