Standard CATEGORY system for api nodes (#35)

This commit is contained in:
Christian Byrne 2025-04-29 15:32:44 -07:00 committed by Jedrzej Kosinski
parent ee0cdb3d26
commit 6276d754bf
4 changed files with 17 additions and 17 deletions

View File

@ -384,7 +384,7 @@ class OpenAIDalle2(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call"
CATEGORY = "api node"
CATEGORY = "api node/image/openai"
DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True
@ -525,7 +525,7 @@ class OpenAIDalle3(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call"
CATEGORY = "api node"
CATEGORY = "api node/image/openai"
DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True
@ -657,7 +657,7 @@ class OpenAIGPTImage1(ComfyNodeABC):
RETURN_TYPES = (IO.IMAGE,)
FUNCTION = "api_call"
CATEGORY = "api node"
CATEGORY = "api node/image/openai"
DESCRIPTION = cleandoc(__doc__ or "")
API_NODE = True
@ -889,7 +889,7 @@ class IdeogramTextToImage(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "Example"
CATEGORY = "api node/image/ideogram"
def api_call(
self,
@ -1063,7 +1063,7 @@ class FluxProUltraImageNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node"
CATEGORY = "api node/image/bfl"
def api_call(
self,
@ -1187,7 +1187,7 @@ class LumaReferenceNode(ComfyNodeABC):
RETURN_NAMES = ("luma_ref",)
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "create_luma_reference"
CATEGORY = "api node/Luma"
CATEGORY = "api node/image/Luma"
@classmethod
def INPUT_TYPES(s):
@ -1232,7 +1232,7 @@ class LumaConceptsNode(ComfyNodeABC):
RETURN_NAMES = ("luma_concepts",)
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "create_concepts"
CATEGORY = "api node/Luma"
CATEGORY = "api node/image/Luma"
@classmethod
def INPUT_TYPES(s):
@ -1265,7 +1265,7 @@ class LumaImageGenerationNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node"
CATEGORY = "api node/image/Luma"
@classmethod
def INPUT_TYPES(s):
@ -1434,7 +1434,7 @@ class LumaImageModifyNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node"
CATEGORY = "api node/image/Luma"
@classmethod
def INPUT_TYPES(s):
@ -1542,7 +1542,7 @@ class LumaTextToVideoGenerationNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node"
CATEGORY = "api node/image/Luma"
@classmethod
def INPUT_TYPES(s):
@ -1644,7 +1644,7 @@ class LumaImageToVideoGenerationNode(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node"
CATEGORY = "api node/image/Luma"
@classmethod
def INPUT_TYPES(s):
@ -1780,7 +1780,7 @@ class RecraftStyleV3RealisticImageNode:
RETURN_TYPES = (RecraftIO.STYLEV3,)
RETURN_NAMES = ("recraft_style",)
FUNCTION = "create_style"
CATEGORY = "api node/Recraft"
CATEGORY = "api node/image/Recraft"
RECRAFT_STYLE = RecraftStyleV3.realistic_image
@ -1831,7 +1831,7 @@ class RecraftTextToImageNode:
DESCRIPTION = cleandoc(__doc__ or "") # Handle potential None value
FUNCTION = "api_call"
API_NODE = True
CATEGORY = "api node"
CATEGORY = "api node/image/Recraft"
@classmethod
def INPUT_TYPES(s):
@ -1996,7 +1996,7 @@ class MinimaxTextToVideoNode:
RETURN_TYPES = ("VIDEO",)
DESCRIPTION = "Generates videos from prompts using Minimax's API"
FUNCTION = "generate_video"
CATEGORY = "video"
CATEGORY = "api node/video/Minimax"
API_NODE = True
OUTPUT_NODE = True

View File

@ -174,7 +174,7 @@ class KlingNodeABC(ComfyNodeABC):
DESCRIPTION = cleandoc(__doc__ or "")
FUNCTION = "api_call"
CATEGORY = "api node/video/kling"
CATEGORY = "api node/video/Kling"
API_NODE = True

View File

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

View File

@ -123,7 +123,7 @@ class VeoVideoGenerationNode(ComfyNodeABC):
RETURN_TYPES = (IO.VIDEO,)
FUNCTION = "generate_video"
CATEGORY = "api node/video"
CATEGORY = "api node/video/Veo"
DESCRIPTION = "Generates videos from text prompts using Google's Veo API"
API_NODE = True