jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

♻️ (schema.py): refactor code to handle different types of component_instance status and retrieve output results accordingly

Browse source
This commit is contained in:
ogabrielluiz 2024-06-18 12:59:10 -03:00
commit ba354ae119
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/base/langflow/schema/schema.py
View file

@ -75,9 +75,14 @@ def get_message(payload):
def build_logs(vertex, result) -> dict: def build_logs(vertex, result) -> dict:
logs = dict() logs = dict()
payload = result[0]._results component_instance = result[0]
for index, output in enumerate(vertex.outputs): 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) message = get_message(output_result)
_type = get_type(output_result) _type = get_type(output_result)