fix: fix iterator upstream check

instill-ai · Oct 31, 2024 · 9c6462f · 9c6462f
1 parent 24153e2
commit 9c6462f
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/pkg/recipe/dag.go b/pkg/recipe/dag.go
@@ -560,6 +560,18 @@ func GenerateDAG(componentMap datamodel.ComponentMap) (*dag, error) {
 			if component.Input != nil {
 				parents = append(parents, FindReferenceParent(component.Input.(string))...)
 			}
+			if component.Range != nil {
+				switch rangeVal := component.Range.(type) {
+				case []any:
+					for _, v := range rangeVal {
+						parents = append(parents, FindReferenceParent(fmt.Sprintf("%v", v))...)
+					}
+				case map[string]any:
+					for _, v := range rangeVal {
+						parents = append(parents, FindReferenceParent(fmt.Sprintf("%v", v))...)
+					}
+				}
+			}
 			nestedComponentIDs := []string{id}
 			for nestedID := range component.Component {
 				nestedComponentIDs = append(nestedComponentIDs, nestedID)