add type array to artifacts

This commit is contained in:
italojohnny 2024-06-04 16:25:03 -03:00
commit 4f20da7228

View file

@ -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