Add resolution parameter support for video models

2025-12-04 20:34:10 -05:00
parent 9e5d8e3d3d
commit 86914d0fee
9 changed files with 394 additions and 4 deletions
@@ -0,0 +1,67 @@
+"""
+Runware Alibaba Provider Settings Node
+Provides Alibaba-specific settings for image/video generation
+"""
+
+from typing import Optional, Dict, Any
+
+
+class RunwareAlibabaProviderSettings:
+    """Runware Alibaba Provider Settings Node"""
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {},
+            "optional": {
+                "usePromptEnhancer": ("BOOLEAN", {
+                    "tooltip": "Enable to include promptEnhancer parameter in provider settings",
+                    "default": False,
+                }),
+                "promptEnhancer": ("BOOLEAN", {
+                    "tooltip": "Enable prompt enhancement for Alibaba generation. Only used when 'Use Prompt Enhancer' is enabled.",
+                    "default": False,
+                    "label_on": "Enabled",
+                    "label_off": "Disabled",
+                }),
+            }
+        }
+
+    RETURN_TYPES = ("RUNWAREPROVIDERSETTINGS",)
+    RETURN_NAMES = ("Provider Settings",)
+    FUNCTION = "createProviderSettings"
+    CATEGORY = "Runware/Provider Settings"
+    DESCRIPTION = "Configure Alibaba-specific provider settings for image/video generation including prompt enhancement."
+
+    def createProviderSettings(self, **kwargs) -> tuple[Dict[str, Any]]:
+        """Create Alibaba provider settings"""
+
+        # Get control parameters
+        usePromptEnhancer = kwargs.get("usePromptEnhancer", False)
+
+        # Get value parameters
+        promptEnhancer = kwargs.get("promptEnhancer", False)
+
+        # Build settings dictionary - only include what is enabled
+        alibabaSettings: Dict[str, Any] = {}
+
+        # Add promptEnhancer only if usePromptEnhancer is enabled
+        if usePromptEnhancer:
+            alibabaSettings["promptEnhancer"] = promptEnhancer
+
+        # Clean up None values
+        alibabaSettings = {k: v for k, v in alibabaSettings.items() if v is not None}
+
+        # Return flat dictionary - will be wrapped by inference node with provider name
+        return (alibabaSettings,)
+
+
+# Node class mappings
+NODE_CLASS_MAPPINGS = {
+    "RunwareAlibabaProviderSettings": RunwareAlibabaProviderSettings,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "RunwareAlibabaProviderSettings": "Runware Alibaba Provider Settings",
+}
+