Selaa lähdekoodia

Fix/code transform result (#3203)

Yeuoly 1 vuosi sitten
vanhempi
commit
337899a03d
1 muutettua tiedostoa jossa 3 lisäystä ja 0 poistoa
  1. 3 0
      api/core/workflow/nodes/code/code_node.py

+ 3 - 0
api/core/workflow/nodes/code/code_node.py

@@ -234,6 +234,9 @@ class CodeNode(BaseNode):
         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):