From 3211170d74fa3093dd9d30dd7464d9c179396f17 Mon Sep 17 00:00:00 2001
From: Kerem Yilmaz <kerem@skyvern.com>
Date: Wed, 31 Jul 2024 15:55:41 -0700
Subject: [PATCH] Skip updating context params if task has errors in workflow
 (#664)

---
 skyvern/forge/sdk/workflow/context_manager.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/skyvern/forge/sdk/workflow/context_manager.py b/skyvern/forge/sdk/workflow/context_manager.py
index 455641cc..5f25a693 100644
--- a/skyvern/forge/sdk/workflow/context_manager.py
+++ b/skyvern/forge/sdk/workflow/context_manager.py
@@ -293,6 +293,14 @@ class WorkflowRunContext:
                 and isinstance(parameter.source, OutputParameter)
                 and parameter.source.key == output_parameter.key
             ):
+                if isinstance(value, dict) and "errors" in value:
+                    LOG.error(
+                        f"Output parameter {output_parameter.key} has errors. Setting ContextParameter {parameter.key} value to None"
+                    )
+                    parameter.value = None
+                    self.parameters[parameter.key] = parameter
+                    self.values[parameter.key] = parameter.value
+                    continue
                 value = (
                     value["extracted_information"]
                     if isinstance(value, dict) and "extracted_information" in value