refactor socket types

2026-03-28 13:56:22 -07:00
parent 4368aeb4a0
commit 1b831cda5d
20 changed files with 366 additions and 79 deletions
--- a/backend/nodes/markup.py
+++ b/backend/nodes/markup.py
@@ -21,7 +21,10 @@ class Markup:
    def INPUT_TYPES(cls):
        return {
            "required": {
-                "input": ("ANNOTATION_SOURCE", {"label": "Input"}),
+                "input": ("ANNOTATION_SOURCE", {
+                    "label": "Input",
+                    "accepted_types": ["DATA_FIELD", "IMAGE"],
+                }),
                "shape": (["line", "rectangle", "circle", "arrow"], {"default": "arrow"}),
                "stroke_color": ("STRING", {"default": "#ff0000", "color_picker": True}),
                "stroke_width": ("INT", {"default": 3, "min": 1, "max": 64, "step": 1}),