nodes/arcum42-ComfyUI_SageUtils
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix LLM Construct Prompt nodes in v3.

Browse Source
This commit is contained in:
Shanoah Alkire
2025-12-28 14:52:11 -08:00
parent 248d40cdb1
commit 4eafdc64bb
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/llm_v3.py
+14 -2
View File
@@ -60,10 +60,16 @@ class Sage_ConstructLLMPrompt(io.ComfyNode):
prompt_list.append(f"{category}/{key}")
# Dynamic extras: add boolean inputs for llm_prompts["extra"] keys
# Only include extras in (style, quality, content_focus) categories
extra_inputs = []
try:
for ekey in llm_prompts.get("extra", {}).keys():
extra_inputs.append(io.Boolean.Input(ekey, default=False))
cfg = llm_prompts["extra"][ekey]
if cfg.get("category") in ("style", "quality", "content_focus"):
if cfg.get("type") == "boolean":
default_value = cfg.get("default", False)
tooltip = cfg.get("name", ekey)
extra_inputs.append(io.Boolean.Input(ekey, default=default_value, tooltip=tooltip))
except Exception:
extra_inputs = []
return io.Schema(
@@ -127,10 +133,16 @@ class Sage_ConstructLLMPromptExtra(io.ComfyNode):
@classmethod
def define_schema(cls):
# Dynamic extras: add boolean inputs for llm_prompts["extra"] keys
# Only include extras NOT in (style, quality, content_focus) categories
extra_inputs = []
try:
for ekey in llm_prompts.get("extra", {}).keys():
extra_inputs.append(io.Boolean.Input(ekey, default=False))
cfg = llm_prompts["extra"][ekey]
if cfg.get("category") not in ("style", "quality", "content_focus"):
if cfg.get("type") == "boolean":
default_value = cfg.get("default", False)
tooltip = cfg.get("name", ekey)
extra_inputs.append(io.Boolean.Input(ekey, default=default_value, tooltip=tooltip))
except Exception:
extra_inputs = []
return io.Schema(