@bian-womp/spark-workbench 0.3.36 → 0.3.38

package/lib/cjs/index.cjs

@@ -652,15 +652,27 @@ class InMemoryWorkbench extends AbstractWorkbench {
             if (!outputTypeId || outputValue === undefined)
                 return undefined;
             const unwrap = (v) => sparkGraph.isTypedOutput(v) ? sparkGraph.getTypedOutputValue(v) : v;
-            const coerceIfNeeded = async (fromType, toType, value) => {
-                if (!toType || toType === fromType || !runner?.coerce)
+            const coerceIfNeeded = async (fromType, toTypes, value) => {
+                if (!toTypes)
                     return value;
-                try {
-                    return await runner.coerce(fromType, toType, value);
-                }
-                catch {
+                const typesArray = Array.isArray(toTypes) ? toTypes : [toTypes];
+                // If output type matches any input type exactly, no coercion needed
+                if (typesArray.includes(fromType))
+                    return value;
+                // If no coercion function available, return as-is
+                if (!runner?.coerce)
                     return value;
+                // Try coercing to each type until one succeeds
+                for (const toType of typesArray) {
+                    try {
+                        return await runner.coerce(fromType, toType, value);
+                    }
+                    catch {
+                        // Continue to next type
+                    }
                 }
+                // If all coercion attempts failed, return value as-is
+                return value;
             };
             const pos = nodePosition;
             const isArray = outputTypeId.endsWith("[]");
@@ -673,8 +685,8 @@ class InMemoryWorkbench extends AbstractWorkbench {
             let newNodeId;
             if (singleTarget) {
                 const nodeDesc = registry.nodes.get(singleTarget.nodeTypeId);
-                const inType = sparkGraph.getInputTypeId(nodeDesc?.inputs, singleTarget.inputHandle);
-                const coerced = await coerceIfNeeded(outputTypeId, inType, unwrap(outputValue));
+                const inTypes = sparkGraph.getInputDeclaredTypes(nodeDesc?.inputs, singleTarget.inputHandle);
+                const coerced = await coerceIfNeeded(outputTypeId, inTypes, unwrap(outputValue));
                 newNodeId = this.addNode({
                     typeId: singleTarget.nodeTypeId,
                 }, {
@@ -684,8 +696,8 @@ class InMemoryWorkbench extends AbstractWorkbench {
             }
             else if (isArray && arrTarget) {
                 const nodeDesc = registry.nodes.get(arrTarget.nodeTypeId);
-                const inType = sparkGraph.getInputTypeId(nodeDesc?.inputs, arrTarget.inputHandle);
-                const coerced = await coerceIfNeeded(outputTypeId, inType, unwrap(outputValue));
+                const inTypes = sparkGraph.getInputDeclaredTypes(nodeDesc?.inputs, arrTarget.inputHandle);
+                const coerced = await coerceIfNeeded(outputTypeId, inTypes, unwrap(outputValue));
                 newNodeId = this.addNode({
                     typeId: arrTarget.nodeTypeId,
                 }, {
@@ -695,10 +707,10 @@ class InMemoryWorkbench extends AbstractWorkbench {
             }
             else if (isArray && elemTarget) {
                 const nodeDesc = registry.nodes.get(elemTarget.nodeTypeId);
-                const inType = sparkGraph.getInputTypeId(nodeDesc?.inputs, elemTarget.inputHandle);
+                const inTypes = sparkGraph.getInputDeclaredTypes(nodeDesc?.inputs, elemTarget.inputHandle);
                 const src = unwrap(outputValue);
                 const items = Array.isArray(src) ? src : [src];
-                const coercedItems = await Promise.all(items.map((v) => coerceIfNeeded(baseTypeId, inType, v)));
+                const coercedItems = await Promise.all(items.map((v) => coerceIfNeeded(baseTypeId, inTypes, v)));
                 const COLS = 4;
                 const DX = 180;
                 const DY = 160;