Fix set Data for NodeConnectionBase types

1 year ago · 241c5f1475
1 changed files with 9 additions and 1 deletions
--- a/StabilityMatrix.Core/Models/Api/Comfy/Nodes/ComfyTypedNodeBase.cs
+++ b/StabilityMatrix.Core/Models/Api/Comfy/Nodes/ComfyTypedNodeBase.cs
@ -32,7 +32,15 @@ public abstract record ComfyTypedNodeBase
                property.GetCustomAttribute<JsonPropertyNameAttribute>()?.Name
                ?? namingPolicy.ConvertName(property.Name);

-            inputs.Add(key, value);
+            // For connection types, use data property
+            if (value is NodeConnectionBase connection)
+            {
+                inputs.Add(key, connection.Data);
+            }
+            else
+            {
+                inputs.Add(key, value);
+            }
        }

        return new NamedComfyNode(Name) { ClassType = ClassType, Inputs = inputs };