@bian-womp/spark-graph 0.1.16 → 0.1.17

package/lib/cjs/index.cjs

@@ -378,8 +378,7 @@ function typed(typeId, value) {
 function isTypedOutput(v) {
     return (!!v &&
         typeof v === "object" &&
-        Object.prototype.hasOwnProperty.call(v, "__spark_type") &&
-        Object.prototype.hasOwnProperty.call(v, "__spark_value"));
+        Object.prototype.hasOwnProperty.call(v, "__spark_type"));
 }
 
 class GraphRuntime {
@@ -1250,15 +1249,30 @@ class GraphBuilder {
                     message: `Edge ${e.id} target node missing`,
                     data: { edgeId: e.id },
                 });
-            // infer edge type from source output if missing
+            // Infer edge type when missing. For union sources, prefer target input type if available.
             let effectiveTypeId = e.typeId;
-            if (!effectiveTypeId && srcNode) {
+            let _srcDeclared;
+            let _dstDeclared;
+            if (srcNode) {
                 const srcType = this.registry.nodes.get(srcNode.typeId);
-                if (srcType) {
-                    const declared = srcType.outputs[e.source.handle];
-                    effectiveTypeId = Array.isArray(declared)
-                        ? declared[0]
-                        : declared;
+                if (srcType)
+                    _srcDeclared = srcType.outputs[e.source.handle];
+            }
+            if (dstNode) {
+                const dstType = this.registry.nodes.get(dstNode.typeId);
+                if (dstType)
+                    _dstDeclared = dstType.inputs[e.target.handle];
+            }
+            if (!effectiveTypeId) {
+                if (Array.isArray(_srcDeclared) && _dstDeclared) {
+                    // When source is a union and target input type is known, adopt the input type
+                    // so validation checks are performed against the target, not an arbitrary variant.
+                    effectiveTypeId = _dstDeclared;
+                }
+                else if (_srcDeclared) {
+                    effectiveTypeId = Array.isArray(_srcDeclared)
+                        ? _srcDeclared[0]
+                        : _srcDeclared;
                 }
             }
             const type = effectiveTypeId
@@ -1295,7 +1309,8 @@ class GraphBuilder {
                             : [];
                     if (declaredArr.length > 0 && effectiveTypeId) {
                         for (const s of declaredArr) {
-                            if (s !== effectiveTypeId && !this.registry.canCoerce(s, effectiveTypeId)) {
+                            if (s !== effectiveTypeId &&
+                                !this.registry.canCoerce(s, effectiveTypeId)) {
                                 issues.push({
                                     level: "error",
                                     code: "TYPE_MISMATCH_OUTPUT",
@@ -1342,7 +1357,8 @@ class GraphBuilder {
                                         ? [effectiveTypeId]
                                         : [];
                             for (const s of srcArr) {
-                                if (s !== declaredIn && !this.registry.canCoerce(s, declaredIn)) {
+                                if (s !== declaredIn &&
+                                    !this.registry.canCoerce(s, declaredIn)) {
                                     issues.push({
                                         level: "error",
                                         code: "TYPE_MISMATCH_INPUT",
@@ -1358,7 +1374,8 @@ class GraphBuilder {
                                 }
                             }
                         }
-                        else if (declaredIn !== effectiveTypeId && !this.registry.canCoerce(effectiveTypeId, declaredIn)) {
+                        else if (declaredIn !== effectiveTypeId &&
+                            !this.registry.canCoerce(effectiveTypeId, declaredIn)) {
                             issues.push({
                                 level: "error",
                                 code: "TYPE_MISMATCH_INPUT",