input output nodes v1

2024-09-25 09:49:23 +08:00
parent 6627185e28
commit 89e275252d
7 changed files with 548 additions and 0 deletions
@@ -0,0 +1,44 @@
+import folder_paths
+from PIL import Image, ImageOps
+import numpy as np
+import torch
+
+class ShellAgentPluginInputText:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "input_name": (
+                    "STRING",
+                    {"multiline": False, "default": "input_text"},
+                ),
+            },
+            "optional": {
+                "default_value": (
+                    "STRING",
+                    {"multiline": True, "default": ""},
+                ),
+                "description": (
+                    "STRING",
+                    {"multiline": True, "default": ""},
+                ),
+                "choices": (
+                    "STRING",
+                    {"multiline": False, "default": ""},
+                ),
+            }
+        }
+
+    RETURN_TYPES = ("STRING",)
+    RETURN_NAMES = ("text",)
+
+    FUNCTION = "run"
+
+    CATEGORY = "shellagent"
+
+    def run(self, input_name, default_value=None, display_name=None, description=None, choices=None):
+        return [default_value]
+
+
+NODE_CLASS_MAPPINGS = {"ShellAgentPluginInputText": ShellAgentPluginInputText}
+NODE_DISPLAY_NAME_MAPPINGS = {"ShellAgentPluginInputText": "Input Text (ShellAgent Plugin)"}