From ba354ae1192a6677143a313d6e91fe0613e0ec08 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 18 Jun 2024 12:59:10 -0300 Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20(schema.py):=20refactor=20?= =?UTF-8?q?code=20to=20handle=20different=20types=20of=20component=5Finsta?= =?UTF-8?q?nce=20status=20and=20retrieve=20output=20results=20accordingly?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/base/langflow/schema/schema.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/schema/schema.py b/src/backend/base/langflow/schema/schema.py index acd1dbab5..884c35e8e 100644 --- a/src/backend/base/langflow/schema/schema.py +++ b/src/backend/base/langflow/schema/schema.py @@ -75,9 +75,14 @@ def get_message(payload): def build_logs(vertex, result) -> dict: logs = dict() - payload = result[0]._results + component_instance = result[0] for index, output in enumerate(vertex.outputs): - output_result = payload.get(output["name"]) + if component_instance.status is None: + payload = component_instance._results + output_result = payload.get(output["name"]) + else: + payload = component_instance._artifacts + output_result = payload.get(output["name"]).get("raw") message = get_message(output_result) _type = get_type(output_result)