From 4f1c925d9a932c4b4d13062ddde1836006003075 Mon Sep 17 00:00:00 2001 From: italojohnny Date: Tue, 4 Jun 2024 16:17:53 -0300 Subject: [PATCH] fix artifact type --- src/backend/base/langflow/graph/utils.py | 2 +- src/backend/base/langflow/interface/initialize/loading.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/graph/utils.py b/src/backend/base/langflow/graph/utils.py index 86fbebc4a..bf29e3ab5 100644 --- a/src/backend/base/langflow/graph/utils.py +++ b/src/backend/base/langflow/graph/utils.py @@ -72,7 +72,7 @@ def get_artifact_type(custom_component, build_result) -> str: case str(): result = ArtifactType.TEXT - case dict(): + case dict() | list(): result = ArtifactType.OBJECT if result == ArtifactType.UNKNOWN: diff --git a/src/backend/base/langflow/interface/initialize/loading.py b/src/backend/base/langflow/interface/initialize/loading.py index 8eb2a4d9a..f9e360fb1 100644 --- a/src/backend/base/langflow/interface/initialize/loading.py +++ b/src/backend/base/langflow/interface/initialize/loading.py @@ -134,5 +134,5 @@ async def instantiate_custom_component(params, user_id, vertex, fallback_to_env_ elif hasattr(raw, "model_dump"): raw = raw.model_dump() - artifact = {"repr": custom_repr, "raw": raw, "type": get_artifact_type(custom_component.repr_value)} + artifact = {"repr": custom_repr, "raw": raw, "type": get_artifact_type(custom_component, build_result)} return custom_component, build_result, artifact