From 13213d45231a6fe0c05d19bd681ab3946136097c Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Tue, 25 Feb 2025 11:59:22 -0800 Subject: [PATCH] Add aggregated extracted information in workflow run status response (#1836) Co-authored-by: Muhammed Salih Altun --- skyvern/forge/sdk/workflow/service.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py index c3d8d724..58abd034 100644 --- a/skyvern/forge/sdk/workflow/service.py +++ b/skyvern/forge/sdk/workflow/service.py @@ -1032,6 +1032,12 @@ class WorkflowService: outputs = None if output_parameter_tuples: outputs = {output_parameter.key: output.value for output_parameter, output in output_parameter_tuples} + extracted_information = { + output_parameter.key: output.value["extracted_information"] + for output_parameter, output in output_parameter_tuples + if isinstance(output.value, dict) and output.value["extracted_information"] is not None + } + outputs["extracted_information"] = extracted_information total_steps = None total_cost = None