Merge pull request #164 from Runware/Ernie-quick-fix

Added settings.promptEnhance.enabled

clientlibs/utils.js

@@ -5687,6 +5687,8 @@ function imageInferenceSettingsPromptEnhanceToggleHandler(node) {
     const temperatureWidget = node.widgets.find((w) => w && w.name === "temperature");
     const useTopPWidget = node.widgets.find((w) => w && w.name === "useTopP");
     const topPWidget = node.widgets.find((w) => w && w.name === "topP");
+    const useEnabledWidget = node.widgets.find((w) => w && w.name === "useEnabled");
+    const enabledWidget = node.widgets.find((w) => w && w.name === "enabled");
 
     function toggleWidgetState(useWidget, paramWidget) {
         if (!useWidget || !paramWidget) return;
@@ -5704,6 +5706,9 @@ function imageInferenceSettingsPromptEnhanceToggleHandler(node) {
     if (useTopPWidget && topPWidget) {
         toggleWidgetState(useTopPWidget, topPWidget);
     }
+    if (useEnabledWidget && enabledWidget) {
+        toggleWidgetState(useEnabledWidget, enabledWidget);
+    }
 }
 
 function imageInferenceSettingsMoodboardsToggleHandler(moodboardsNode) {

modules/imageInferenceSettingsPromptEnhance.py

@@ -8,7 +8,7 @@ from typing import Any, Dict, Tuple
 
 
 class RunwareImageInferenceSettingsPromptEnhance:
-    """Build settings.promptEnhance object (temperature, topP)."""
+    """Build settings.promptEnhance object (enabled, temperature, topP)."""
 
     @classmethod
     def INPUT_TYPES(cls):
@@ -47,6 +47,21 @@ class RunwareImageInferenceSettingsPromptEnhance:
                         "Ernie-specific API default when omitted: 0.95."
                     ),
                 }),
+                "useEnabled": ("BOOLEAN", {
+                    "default": False,
+                    "label_on": "Enabled",
+                    "label_off": "Disabled",
+                    "tooltip": "Include enabled in settings.promptEnhance.",
+                }),
+                "enabled": ("BOOLEAN", {
+                    "default": True,
+                    "label_on": "true",
+                    "label_off": "false",
+                    "tooltip": (
+                        "settings.promptEnhance.enabled. Required when using temperature or topP. "
+                        "Only used when 'Use Enabled' is on."
+                    ),
+                }),
             },
         }
 
@@ -55,19 +70,27 @@ class RunwareImageInferenceSettingsPromptEnhance:
     FUNCTION = "build_prompt_enhance"
     CATEGORY = "Runware"
     DESCRIPTION = (
-        "Build settings.promptEnhance (temperature, topP) for image inference. "
+        "Build settings.promptEnhance (enabled, temperature, topP) for image inference. "
         "Connect to Runware Image Inference Settings."
     )
 
     def build_prompt_enhance(self, **kwargs) -> Tuple[Dict[str, Any]]:
         prompt_enhance: Dict[str, Any] = {}
 
+        if kwargs.get("useEnabled", False):
+            prompt_enhance["enabled"] = bool(kwargs.get("enabled", True))
+
         if kwargs.get("useTemperature", False):
             prompt_enhance["temperature"] = float(kwargs.get("temperature", 1.2))
 
         if kwargs.get("useTopP", False):
             prompt_enhance["topP"] = float(kwargs.get("topP", 0.95))
 
+        if prompt_enhance and "enabled" not in prompt_enhance and (
+            "temperature" in prompt_enhance or "topP" in prompt_enhance
+        ):
+            prompt_enhance["enabled"] = True
+
         return (prompt_enhance,)