feat: Tool workflow node execution

Author: shaohuzhang1

apps/application/flow/i_step_node.py

@@ -123,7 +123,19 @@ def __init__(self, chat_info, tool_id):
 
     def handler(self, workflow):
         state = get_workflow_state(workflow)
-        ToolRecord(tool_id=self.tool_id)
+        record = ToolRecord(id=self.chat_info.tool_record_id, tool_id=self.tool_id,
+                            workspace_id=self.chat_info.workspace_id,
+                            source_type=self.chat_info.source_type,
+                            source_id=self.chat_info.source_id,
+                            state=state,
+                            meta={
+                                'output': workflow.out_context,
+                                'details': workflow.get_runtime_details(),
+                                'answer_text_list': workflow.get_answer_text_list()
+                            })
+        self.chat_info.set_record(record)
+        self.chat_info = None
+        self.tool_id = None
 
 
 def get_loop_workflow_node(node_list):

apps/application/flow/knowledge_loop_workflow_manage.py

@@ -13,3 +13,9 @@
 class KnowledgeLoopWorkflowManage(LoopWorkflowManage):
     def get_params_serializer_class(self):
         return KnowledgeFlowParamsSerializer
+
+    def get_source_type(self):
+        return "KNOWLEDGE"
+
+    def get_source_id(self):
+        return self.params.get('knowledge_id')

apps/application/flow/knowledge_workflow_manage.py

@@ -122,3 +122,9 @@ def hand_node_result(self, current_node, node_result_future):
             current_node.node_chunk.end()
             QuerySet(KnowledgeAction).filter(id=self.params.get('knowledge_action_id')).update(
                 details=self.get_runtime_details())
+
+    def get_source_type(self):
+        return "KNOWLEDGE"
+
+    def get_source_id(self):
+        return self.params.get('knowledge_id')

apps/application/flow/loop_workflow_manage.py

@@ -191,3 +191,9 @@ def generate_prompt(self, prompt: str):
         prompt_template = PromptTemplate.from_template(prompt, template_format='jinja2')
         value = prompt_template.format(context=context)
         return value
+
+    def get_source_type(self):
+        return "APPLICATION"
+
+    def get_source_id(self):
+        return self.params.get('application_id')

apps/application/flow/step_node/__init__.py

@@ -35,6 +35,7 @@
 from .text_to_video_step_node.impl.base_text_to_video_node import BaseTextToVideoNode
 from .tool_lib_node import *
 from .tool_node import *
+from .tool_workflow_lib_node import BaseToolWorkflowLibNodeNode
 from .variable_aggregation_node.impl.base_variable_aggregation_node import BaseVariableAggregationNode
 from .variable_assign_node import BaseVariableAssignNode
 from .variable_splitting_node import BaseVariableSplittingNode
@@ -53,7 +54,7 @@
              BaseLoopContinueNode,
              BaseLoopBreakNode, BaseVariableSplittingNode, BaseParameterExtractionNode, BaseVariableAggregationNode,
              BaseDataSourceLocalNode, BaseDataSourceWebNode, BaseKnowledgeWriteNode, BaseDocumentSplitNode,
-             BaseToolStartStepNode]
+             BaseToolStartStepNode, BaseToolWorkflowLibNodeNode]
 
 node_map = {n.type: {w: n for w in n.support} for n in node_list}
 

apps/application/flow/step_node/form_node/impl/base_form_node.py

@@ -144,8 +144,9 @@ def get_answer_list(self) -> List[Answer] | None:
         value = prompt_template.format(form=form, context=context, runtime_node_id=self.runtime_node_id,
                                        chat_record_id=self.flow_params_serializer.data.get("chat_record_id"),
                                        form_field_list=form_field_list)
-        return [Answer(value, self.view_type, self.runtime_node_id, self.workflow_params['chat_record_id'], None,
-                       self.runtime_node_id, '')]
+        return [
+            Answer(value, self.view_type, self.runtime_node_id, self.workflow_params.get('chat_record_id') or '', None,
+                   self.runtime_node_id, '')]
 
     def get_details(self, index: int, **kwargs):
         form_content_format = self.context.get('form_content_format')

apps/application/flow/step_node/loop_node/impl/base_loop_node.py

@@ -268,10 +268,16 @@ def get_loop_context(self):
     def execute(self, loop_type, array, number, loop_body, **kwargs) -> NodeResult:
         from application.flow.loop_workflow_manage import LoopWorkflowManage, Workflow
         from application.flow.knowledge_loop_workflow_manage import KnowledgeLoopWorkflowManage
+        from application.flow.tool_loop_workflow_manage import ToolLoopWorkflowManage
         def workflow_manage_new_instance(loop_data, global_data, start_node_id=None,
                                          start_node_data=None, chat_record=None, child_node=None):
-            workflow_mode = WorkflowMode.KNOWLEDGE_LOOP if WorkflowMode.KNOWLEDGE == self.workflow_manage.flow.workflow_mode else WorkflowMode.APPLICATION_LOOP
-            c = KnowledgeLoopWorkflowManage if workflow_mode == WorkflowMode.KNOWLEDGE_LOOP else LoopWorkflowManage
+            workflow_mode = {WorkflowMode.APPLICATION: WorkflowMode.APPLICATION_LOOP,
+                             WorkflowMode.KNOWLEDGE: WorkflowMode.KNOWLEDGE_LOOP,
+                             WorkflowMode.TOOL: WorkflowMode.TOOL_LOOP}.get(
+                self.workflow_manage.flow.workflow_mode) or WorkflowMode.APPLICATION
+            c = {WorkflowMode.APPLICATION_LOOP: LoopWorkflowManage,
+                 WorkflowMode.KNOWLEDGE_LOOP: KnowledgeLoopWorkflowManage,
+                 WorkflowMode.TOOL_LOOP: ToolLoopWorkflowManage}.get(workflow_mode) or LoopWorkflowManage
             workflow_manage = c(Workflow.new_instance(loop_body, workflow_mode),
                                 self.workflow_manage.params,
                                 LoopWorkFlowPostHandler(

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

@@ -0,0 +1,9 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎虎
+    @file： __init__.py.py
+    @date：2026/3/16 13:53
+    @desc:
+"""
+from .impl import *

apps/application/flow/step_node/tool_workflow_lib_node/i_tool_workflow_lib_node.py

@@ -0,0 +1,57 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎
+    @file： i_function_lib_node.py
+    @date：2024/8/8 16:21
+    @desc:
+"""
+from typing import Type
+
+from django.db import connection
+from django.db.models import QuerySet
+from django.utils.translation import gettext_lazy as _
+from rest_framework import serializers
+
+from application.flow.common import WorkflowMode
+from application.flow.i_step_node import INode, NodeResult
+from common.field.common import ObjectField
+from tools.models.tool import Tool, ToolType
+
+
+class InputField(serializers.Serializer):
+    field = serializers.CharField(required=True, label=_('Variable Name'))
+    label = serializers.CharField(required=True, label=_('Variable Label'))
+    source = serializers.CharField(required=True, label=_('Variable Source'))
+    type = serializers.CharField(required=True, label=_('Variable Type'))
+    value = ObjectField(required=True, label=_("Variable Value"), model_type_list=[str, list, bool, dict, int, float])
+
+
+class FunctionLibNodeParamsSerializer(serializers.Serializer):
+    tool_lib_id = serializers.UUIDField(required=True, label=_('Library ID'))
+    input_field_list = InputField(required=True, many=True)
+    is_result = serializers.BooleanField(required=False,
+                                         label=_('Whether to return content'))
+
+    def is_valid(self, *, raise_exception=False):
+        super().is_valid(raise_exception=True)
+        f_lib = QuerySet(Tool).filter(id=self.data.get('tool_lib_id'), tool_type=ToolType.WORKFLOW).first()
+        # 归还链接到连接池
+        connection.close()
+        if f_lib is None:
+            raise Exception(_('The function has been deleted'))
+
+
+class IToolWorkflowLibNode(INode):
+    type = 'tool-workflow-lib-node'
+    support = [WorkflowMode.APPLICATION, WorkflowMode.APPLICATION_LOOP, WorkflowMode.KNOWLEDGE,
+               WorkflowMode.KNOWLEDGE_LOOP, WorkflowMode.TOOL, WorkflowMode.TOOL_LOOP]
+
+    def get_node_params_serializer_class(self) -> Type[serializers.Serializer]:
+        return FunctionLibNodeParamsSerializer
+
+    def _run(self):
+        return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
+
+    def execute(self, tool_lib_id, input_field_list, **kwargs) -> NodeResult:
+        pass

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

@@ -0,0 +1,9 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎虎
+    @file： __init__.py.py
+    @date：2026/3/16 13:53
+    @desc:
+"""
+from .base_tool_workflow_lib_node import *