Merge branch 'feat/workflow-backend' into deploy/dev

langgenius · Mar 14, 2024 · 0798d59 · 0798d59
2 parents 59bd908 + 95ee725
commit 0798d59
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/api/core/workflow/nodes/tool/tool_node.py b/api/core/workflow/nodes/tool/tool_node.py
@@ -5,7 +5,6 @@
 from core.tools.entities.tool_entities import ToolInvokeMessage
 from core.tools.tool_manager import ToolManager
 from core.tools.utils.message_transformer import ToolFileMessageTransformer
-from core.workflow.entities.base_node_data_entities import BaseNodeData
 from core.workflow.entities.node_entities import NodeRunResult, NodeType
 from core.workflow.entities.variable_pool import VariablePool
 from core.workflow.nodes.base_node import BaseNode
@@ -136,12 +135,12 @@ def _extract_tool_response_text(self, tool_response: list[ToolInvokeMessage]) ->
 
 
     @classmethod
-    def _extract_variable_selector_to_variable_mapping(cls, node_data: BaseNodeData) -> dict[str, list[str]]:
+    def _extract_variable_selector_to_variable_mapping(cls, node_data: ToolNodeData) -> dict[str, list[str]]:
         """
         Extract variable selector to variable mapping
         """
         return {
             k.variable: k.value_selector
-            for k in cast(ToolNodeData, node_data).tool_parameters
+            for k in node_data.tool_parameters
             if k.variable_type == 'selector'
         }
diff --git a/api/core/workflow/workflow_engine_manager.py b/api/core/workflow/workflow_engine_manager.py
@@ -54,10 +54,7 @@ def get_default_configs(self) -> list[dict]:
         for node_type, node_class in node_classes.items():
             default_config = node_class.get_default_config()
             if default_config:
-                default_block_configs.append({
-                    'type': node_type.value,
-                    'config': default_config
-                })
+                default_block_configs.append(default_config)
 
         return default_block_configs