feat: Support variable splitting nodes

Author: shaohuzhang1

apps/application/flow/step_node/__init__.py

@@ -31,6 +31,7 @@
 from .tool_lib_node import *
 from .tool_node import *
 from .variable_assign_node import BaseVariableAssignNode
+from .variable_splitting_node import BaseVariableSplittingNode
 
 node_list = [BaseStartStepNode, BaseChatNode, BaseSearchKnowledgeNode, BaseQuestionNode,
              BaseConditionNode, BaseReplyNode,
@@ -40,7 +41,7 @@
              BaseImageGenerateNode, BaseVariableAssignNode, BaseMcpNode, BaseTextToVideoNode, BaseImageToVideoNode,
              BaseIntentNode, BaseLoopNode, BaseLoopStartStepNode,
              BaseLoopContinueNode,
-             BaseLoopBreakNode]
+             BaseLoopBreakNode, BaseVariableSplittingNode]
 
 
 def get_node(node_type):

apps/application/flow/step_node/variable_splitting_node/__init__.py

@@ -0,0 +1,9 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎虎
+    @file： __init__.py.py
+    @date：2025/10/13 14:56
+    @desc:
+"""
+from .impl import *

apps/application/flow/step_node/variable_splitting_node/i_variable_splitting_node.py

@@ -0,0 +1,32 @@
+# coding=utf-8
+
+from typing import Type
+
+from django.utils.translation import gettext_lazy as _
+from rest_framework import serializers
+
+from application.flow.i_step_node import INode, NodeResult
+
+
+class VariableSplittingNodeParamsSerializer(serializers.Serializer):
+    input_variable = serializers.ListField(required=True,
+                                           label=_("input variable"))
+
+    variable_list = serializers.ListField(required=True,
+                                          label=_("Split variables"))
+
+
+class IVariableSplittingNode(INode):
+    type = 'variable-splitting-node'
+
+    def get_node_params_serializer_class(self) -> Type[serializers.Serializer]:
+        return VariableSplittingNodeParamsSerializer
+
+    def _run(self):
+        input_variable = self.workflow_manage.get_reference_field(
+            self.node_params_serializer.data.get('input_variable')[0],
+            self.node_params_serializer.data.get('input_variable')[1:])
+        return self.execute(input_variable, self.node_params_serializer.data['variable_list'])
+
+    def execute(self, input_variable, variable_list, **kwargs) -> NodeResult:
+        pass

apps/application/flow/step_node/variable_splitting_node/impl/__init__.py

@@ -0,0 +1,9 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎虎
+    @file： __init__.py.py
+    @date：2025/10/13 15:01
+    @desc:
+"""
+from .base_variable_splitting_node import *

apps/application/flow/step_node/variable_splitting_node/impl/base_variable_splitting_node.py

@@ -0,0 +1,53 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎虎
+    @file： base_variable_splitting_node.py
+    @date：2025/10/13 15:02
+    @desc:
+"""
+from jsonpath_ng import parse
+
+from application.flow.i_step_node import NodeResult
+from application.flow.step_node.variable_splitting_node.i_variable_splitting_node import IVariableSplittingNode
+
+
+def smart_jsonpath_search(data: dict, path: str):
+    """
+    智能JSON Path搜索
+    返回:
+    - 单个匹配: 直接返回值
+    - 多个匹配: 返回值的列表
+    - 无匹配: 返回None
+    """
+    jsonpath_expr = parse(path)
+    matches = jsonpath_expr.find(data)
+
+    if not matches:
+        return None
+    elif len(matches) == 1:
+        return matches[0].value
+    else:
+        return [match.value for match in matches]
+
+
+class BaseVariableSplittingNode(IVariableSplittingNode):
+    def save_context(self, details, workflow_manage):
+        for key, value in details.get('result').items():
+            self.context['key'] = value
+        self.context['result'] = details.get('result')
+
+    def execute(self, input_variable, variable_list, **kwargs) -> NodeResult:
+        response = {v['field']: smart_jsonpath_search(input_variable, v['expression']) for v in variable_list}
+        return NodeResult({'result': response, **response}, {})
+
+    def get_details(self, index: int, **kwargs):
+        return {
+            'name': self.node.properties.get('stepName'),
+            "index": index,
+            'run_time': self.context.get('run_time'),
+            'type': self.node.type,
+            'result': self.context.get('result'),
+            'status': self.status,
+            'err_message': self.err_message
+        }

pyproject.toml

@@ -57,6 +57,7 @@ dependencies = [
     "websockets==15.0.1",
     "pylint==3.3.7",
     "cohere==5.17.0",
+    "jsonpath-ng==1.7.0"
 ]
 
 [tool.uv]

ui/src/assets/workflow/icon_variable_splitting.svg

@@ -32,6 +32,7 @@ export enum WorkflowType {
   LoopStartNode = 'loop-start-node',
   LoopContinueNode = 'loop-continue-node',
   LoopBreakNode = 'loop-break-node',
+  VariableSplittingNode = 'variable-splitting-node',
 }
 export enum WorkflowMode {
   // 应用工作流

ui/src/enums/application.ts

@@ -396,6 +396,16 @@ You are a master of problem optimization, adept at accurately inferring user int
       text: 'Terminate the current loop and exit the loop body',
       isBreak: 'Break',
     },
+    variableSplittingNode: {
+      label: 'Variable Splitting',
+      text: 'Used to split variables',
+      result: 'Result',
+      splitVariables: 'Split Variables',
+      expression: {
+        label: 'Expression',
+        placeholder: 'Please enter expression',
+      },
+    },
   },
   compare: {
     is_null: 'Is null',

ui/src/locales/lang/en-US/views/application-workflow.ts

@@ -28,6 +28,9 @@ export default {
     placeholder: '请输入关键字搜索',
   },
   paramForm: {
+    variable: {
+      label: '变量',
+    },
     field: {
       label: '参数',
       placeholder: '请输入参数',