15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: knowledge workflow

This commit is contained in:
shaohuzhang1 2025-11-18 10:58:11 +08:00
parent bc1a4809e0
commit 30902f85cb
2 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/application/flow/step_node/__init__.py
View File

@ -50,13 +50,9 @@ node_list = [BaseStartStepNode, BaseChatNode, BaseSearchKnowledgeNode, BaseSearc
BaseIntentNode, BaseLoopNode, BaseLoopStartStepNode,
BaseLoopContinueNode,
BaseLoopBreakNode, BaseVariableSplittingNode, BaseParameterExtractionNode, BaseVariableAggregationNode,
<<<<<<< Updated upstream
BaseDataSourceLocalNode,BaseDataSourceWebNode,BaseKnowledgeWriteNode]
=======
BaseDataSourceLocalNode, BaseDataSourceWebNode]
BaseDataSourceLocalNode, BaseDataSourceWebNode, BaseKnowledgeWriteNode]
node_map = {n.type: {w: n for w in n.support} for n in node_list}
>>>>>>> Stashed changes
def get_node(node_type, workflow_model):

7
apps/application/flow/step_node/knowledge_write_node/i_knowledge_write_node.py
View File

@ -11,23 +11,20 @@ from typing import Type
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
class KnowledgeWriteNodeParamSerializer(serializers.Serializer):
paragraph_list = serializers.ListField(required=True, label=_("Paragraph list"))
chunk_length = serializers.CharField(required=True, label=_("Child chunk length"))
class IKnowledgeWriteNode(INode):
def get_node_params_serializer_class(self) -> Type[serializers.Serializer]:
return KnowledgeWriteNodeParamSerializer
def _run(self):
return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
@ -35,4 +32,4 @@ class IKnowledgeWriteNode(INode):
pass
type = 'knowledge-write-node'
support = [WorkflowMode.KNOWLEDGE]