support upstream STRUCT downstream JSON case

Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Aug 5, 2024 · 108350f · 108350f
1 parent f3682da
commit 108350f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flytepropeller/pkg/compiler/validators/typing.go b/flytepropeller/pkg/compiler/validators/typing.go
@@ -44,6 +44,10 @@ func (t trivialChecker) CastsFrom(upstreamType *flyte.LiteralType) bool {
 	downstreamTypeCopy.Metadata = &structpb.Struct{}
 	upstreamTypeCopy.Annotation = &flyte.TypeAnnotation{}
 	downstreamTypeCopy.Annotation = &flyte.TypeAnnotation{}
+	// Handle upstream: STRUCT, downstream JSON
+	if upstreamTypeCopy.GetSimple() == flyte.SimpleType_STRUCT && downstreamTypeCopy.GetSimple() == flyte.SimpleType_JSON {
+		return true
+	}
 	return upstreamTypeCopy.String() == downstreamTypeCopy.String()
 }