From be979744f5eb4aa7e8951330fb9bb6e83e302588 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Wed, 26 Feb 2025 16:19:56 -0800 Subject: [PATCH] fix workflow run get bug (#1845) --- skyvern/forge/sdk/workflow/service.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py index 58abd034..c84a9c55 100644 --- a/skyvern/forge/sdk/workflow/service.py +++ b/skyvern/forge/sdk/workflow/service.py @@ -1030,14 +1030,17 @@ class WorkflowService: ) outputs = None + EXTRACTED_INFORMATION_KEY = "extracted_information" 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"] + output_parameter.key: output.value[EXTRACTED_INFORMATION_KEY] for output_parameter, output in output_parameter_tuples - if isinstance(output.value, dict) and output.value["extracted_information"] is not None + if output.value is not None + and isinstance(output.value, dict) + and output.value[EXTRACTED_INFORMATION_KEY] is not None } - outputs["extracted_information"] = extracted_information + outputs[EXTRACTED_INFORMATION_KEY] = extracted_information total_steps = None total_cost = None