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/save_image.py
+++ b/backend/nodes/save_image.py
@@ -17,7 +17,10 @@ class SaveImage:
            "directory": ("DIRECTORY", {"label": "directory"}),
        }
        for i in range(_MAX_SAVE_FIELDS):
-            optional[f"field_{i}"] = ("SAVE_LAYER", {"label": f"layer {i + 1}"})
+            optional[f"field_{i}"] = ("DATA_FIELD", {
+                "label": f"layer {i + 1}",
+                "accepted_types": ["IMAGE", "ANNOTATION_SOURCE"],
+            })
            optional[f"layer_name_{i}"] = ("STRING", {
                "default": "",
                "placeholder": "name",