Feat/environment variables in workflow (#6515)

Co-authored-by: JzoNg <jzongcode@gmail.com>
2024-07-22 15:29:39 +08:00 · 2024-07-22 15:29:39 +08:00 · 5e6fc58db3
commit 5e6fc58db3
parent 87d583f454
146 changed files with 2486 additions and 746 deletions
--- a/api/core/workflow/nodes/code/code_node.py
+++ b/api/core/workflow/nodes/code/code_node.py
@ -59,11 +59,8 @@ class CodeNode(BaseNode):
        variables = {}
        for variable_selector in node_data.variables:
            variable = variable_selector.variable
-            value = variable_pool.get_variable_value(
-                variable_selector=variable_selector.value_selector
-            )
-
-            variables[variable] = value
+            value = variable_pool.get(variable_selector.value_selector)
+            variables[variable] = value.value if value else None
        # Run code
        try:
            result = CodeExecutor.execute_workflow_code_template(