feat: 支持表单收集功能 30%

Author: shaohuzhang1

apps/application/flow/i_step_node.py

@@ -7,6 +7,7 @@
     @desc:
 """
 import time
+import uuid
 from abc import abstractmethod
 from typing import Type, Dict, List
 
@@ -118,7 +119,12 @@ class FlowParamsSerializer(serializers.Serializer):
 
 
 class INode:
-    def __init__(self, node, workflow_params, workflow_manage):
+
+    @abstractmethod
+    def save_context(self, details, workflow_manage):
+        pass
+
+    def __init__(self, node, workflow_params, workflow_manage, runtime_node_id=None):
         # 当前步骤上下文,用于存储当前步骤信息
         self.status = 200
         self.err_message = ''
@@ -130,6 +136,10 @@ def __init__(self, node, workflow_params, workflow_manage):
         self.flow_params_serializer = None
         self.context = {}
         self.id = node.id
+        if runtime_node_id is None:
+            self.runtime_node_id = str(uuid.uuid1())
+        else:
+            self.runtime_node_id = runtime_node_id
 
     def valid_args(self, node_params, flow_params):
         flow_params_serializer_class = self.get_flow_params_serializer_class()

apps/application/flow/step_node/__init__.py

@@ -8,16 +8,17 @@
 """
 from .ai_chat_step_node import *
 from .condition_node import *
-from .question_node import *
-from .search_dataset_node import *
-from .start_node import *
 from .direct_reply_node import *
+from .form_node import *
 from .function_lib_node import *
 from .function_node import *
+from .question_node import *
 from .reranker_node import *
+from .search_dataset_node import *
+from .start_node import *
 
 node_list = [BaseStartStepNode, BaseChatNode, BaseSearchDatasetNode, BaseQuestionNode, BaseConditionNode, BaseReplyNode,
-             BaseFunctionNodeNode, BaseFunctionLibNodeNode, BaseRerankerNode]
+             BaseFunctionNodeNode, BaseFunctionLibNodeNode, BaseRerankerNode, BaseFormNode]
 
 
 def get_node(node_type):

apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py

@@ -73,6 +73,10 @@ def get_default_model_params_setting(model_id):
 
 
 class BaseChatNode(IChatNode):
+    def save_context(self, details, workflow_manage):
+        self.context['answer'] = details.get('answer')
+        self.context['question'] = details.get('question')
+
     def execute(self, model_id, system, prompt, dialogue_number, history_chat_record, stream, chat_id, chat_record_id,
                 model_params_setting=None,
                 **kwargs) -> NodeResult:

apps/application/flow/step_node/condition_node/impl/base_condition_node.py

@@ -14,6 +14,10 @@
 
 
 class BaseConditionNode(IConditionNode):
+    def save_context(self, details, workflow_manage):
+        self.context['branch_id'] = details.get('branch_id')
+        self.context['branch_name'] = details.get('branch_name')
+
     def execute(self, **kwargs) -> NodeResult:
         branch_list = self.node_params_serializer.data['branch']
         branch = self._execute(branch_list)

apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py

@@ -13,6 +13,9 @@
 
 
 class BaseReplyNode(IReplyNode):
+    def save_context(self, details, workflow_manage):
+        self.context['answer'] = details.get('answer')
+
     def execute(self, reply_type, stream, fields=None, content=None, **kwargs) -> NodeResult:
         if reply_type == 'referencing':
             result = self.get_reference_content(fields)

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

@@ -0,0 +1,9 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎
+    @file： __init__.py.py
+    @date：2024/11/4 14:48
+    @desc:
+"""
+from .impl import *

apps/application/flow/step_node/form_node/i_form_node.py

@@ -0,0 +1,32 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎
+    @file： i_form_node.py
+    @date：2024/11/4 14:48
+    @desc:
+"""
+from typing import Type
+
+from rest_framework import serializers
+
+from application.flow.i_step_node import INode, NodeResult
+from common.util.field_message import ErrMessage
+
+
+class FormNodeParamsSerializer(serializers.Serializer):
+    form_field_list = serializers.ListField(required=True, error_messages=ErrMessage.list("表单配置"))
+    form_content_format = serializers.CharField(required=True, error_messages=ErrMessage.char('表单输出内容'))
+
+
+class IFormNode(INode):
+    type = 'form-node'
+
+    def get_node_params_serializer_class(self) -> Type[serializers.Serializer]:
+        return FormNodeParamsSerializer
+
+    def _run(self):
+        return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
+
+    def execute(self, form_field_list, form_content_format, **kwargs) -> NodeResult:
+        pass

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

@@ -0,0 +1,9 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎
+    @file： __init__.py.py
+    @date：2024/11/4 14:49
+    @desc:
+"""
+from .base_form_node import BaseFormNode

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

@@ -0,0 +1,65 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎
+    @file： base_form_node.py
+    @date：2024/11/4 14:52
+    @desc:
+"""
+import json
+import time
+from typing import Dict
+
+from langchain_core.prompts import PromptTemplate
+
+from application.flow.i_step_node import NodeResult
+from application.flow.step_node.form_node.i_form_node import IFormNode
+
+
+def write_context(step_variable: Dict, global_variable: Dict, node, workflow):
+    if step_variable is not None:
+        for key in step_variable:
+            node.context[key] = step_variable[key]
+        if workflow.is_result(node, NodeResult(step_variable, global_variable)) and 'result' in step_variable:
+            result = step_variable['result']
+            yield result
+            workflow.answer += result
+    node.context['run_time'] = time.time() - node.context['start_time']
+
+
+class BaseFormNode(IFormNode):
+    def save_context(self, details, workflow_manage):
+        self.context['result'] = details.get('result')
+        self.context['form_content_format'] = details.get('form_content_format')
+        self.context['form_field_list'] = details.get('form_field_list')
+        self.context['run_time'] = details.get('run_time')
+        self.context['start_time'] = details.get('start_time')
+
+    def execute(self, form_field_list, form_content_format, **kwargs) -> NodeResult:
+        form = f'<form_rander>{json.dumps({"form_field_list": form_field_list, "runtime_node_id": self.runtime_node_id, "chat_record_id": self.flow_params_serializer.data.get("chat_record_id")})}</form_rander>'
+        prompt_template = PromptTemplate.from_template(form_content_format, template_format='jinja2')
+        value = prompt_template.format(form=form)
+        return NodeResult(
+            {'result': value, 'form_field_list': form_field_list, 'form_content_format': 'form_content_format'}, {},
+            _write_context=write_context)
+
+    def get_details(self, index: int, **kwargs):
+        form_content_format = self.context.get('form_content_format')
+        form_field_list = self.context.get('form_field_list')
+        form_data = self.context.get('form_data')
+        form = f'<form_rander>{json.dumps({"form_field_list": form_field_list, "node_id": self.node.id, "form_data": form_data})}</form_rander>'
+        prompt_template = PromptTemplate.from_template(form_content_format, template_format='jinja2')
+        value = prompt_template.format(form=form)
+        return {
+            'name': self.node.properties.get('stepName'),
+            "index": index,
+            "result": value,
+            "form_content_format": self.context.get('form_content_format'),
+            "form_field_list": self.context.get('form_field_list'),
+            'form_data': self.context.get('form_data'),
+            'start_time': self.context.get('start_time'),
+            'run_time': self.context.get('run_time'),
+            'type': self.node.type,
+            'status': self.status,
+            'err_message': self.err_message
+        }

apps/application/flow/step_node/function_lib_node/impl/base_function_lib_node.py

@@ -89,6 +89,9 @@ def convert_value(name: str, value, _type, is_required, source, node):
 
 
 class BaseFunctionLibNodeNode(IFunctionLibNode):
+    def save_context(self, details, workflow_manage):
+        self.context['result'] = details.get('result')
+
     def execute(self, function_lib_id, input_field_list, **kwargs) -> NodeResult:
         function_lib = QuerySet(FunctionLib).filter(id=function_lib_id).first()
         if not function_lib.is_active: