Fix/code transform result (#3203)

langgenius · Apr 9, 2024 · 337899a · 337899a
1 parent bae0c07
commit 337899a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/core/workflow/nodes/code/code_node.py b/api/core/workflow/nodes/code/code_node.py
@@ -234,6 +234,9 @@ def _transform_result(self, result: dict, output_schema: Optional[dict[str, Code
         parameters_validated = {}
         for output_name, output_config in output_schema.items():
             dot = '.' if prefix else ''
+            if output_name not in result:
+                raise ValueError(f'Output {prefix}{dot}{output_name} is missing.')
+
             if output_config.type == 'object':
                 # check if output is object
                 if not isinstance(result.get(output_name), dict):