diff --git a/src/backend/base/langflow/graph/utils.py b/src/backend/base/langflow/graph/utils.py index bf29e3ab5..bdb2be5c1 100644 --- a/src/backend/base/langflow/graph/utils.py +++ b/src/backend/base/langflow/graph/utils.py @@ -20,6 +20,7 @@ class ArtifactType(str, Enum): TEXT = "text" RECORD = "record" OBJECT = "object" + ARRAY = "array" STREAM = "stream" UNKNOWN = "unknown" @@ -72,9 +73,13 @@ def get_artifact_type(custom_component, build_result) -> str: case str(): result = ArtifactType.TEXT - case dict() | list(): + case dict(): result = ArtifactType.OBJECT + case list(): + result = ArtifactType.ARRAY + + if result == ArtifactType.UNKNOWN: if isinstance(build_result, Generator): result = ArtifactType.STREAM