Skip to content

Commit 30902f8

Browse files
committed
feat: knowledge workflow
1 parent bc1a480 commit 30902f8

File tree

2 files changed

+3
-10
lines changed

2 files changed

+3
-10
lines changed

apps/application/flow/step_node/__init__.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -50,13 +50,9 @@
5050
BaseIntentNode, BaseLoopNode, BaseLoopStartStepNode,
5151
BaseLoopContinueNode,
5252
BaseLoopBreakNode, BaseVariableSplittingNode, BaseParameterExtractionNode, BaseVariableAggregationNode,
53-
<<<<<<< Updated upstream
54-
BaseDataSourceLocalNode,BaseDataSourceWebNode,BaseKnowledgeWriteNode]
55-
=======
56-
BaseDataSourceLocalNode, BaseDataSourceWebNode]
53+
BaseDataSourceLocalNode, BaseDataSourceWebNode, BaseKnowledgeWriteNode]
5754

5855
node_map = {n.type: {w: n for w in n.support} for n in node_list}
59-
>>>>>>> Stashed changes
6056

6157

6258
def get_node(node_type, workflow_model):

apps/application/flow/step_node/knowledge_write_node/i_knowledge_write_node.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,28 +11,25 @@
1111
from django.utils.translation import gettext_lazy as _
1212
from rest_framework import serializers
1313

14+
from application.flow.common import WorkflowMode
1415
from application.flow.i_step_node import INode, NodeResult
1516

1617

17-
1818
class KnowledgeWriteNodeParamSerializer(serializers.Serializer):
1919
paragraph_list = serializers.ListField(required=True, label=_("Paragraph list"))
2020
chunk_length = serializers.CharField(required=True, label=_("Child chunk length"))
2121

2222

23-
24-
2523
class IKnowledgeWriteNode(INode):
2624

2725
def get_node_params_serializer_class(self) -> Type[serializers.Serializer]:
2826
return KnowledgeWriteNodeParamSerializer
2927

30-
3128
def _run(self):
3229
return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
3330

3431
def execute(self, paragraph_list, chunk_length, **kwargs) -> NodeResult:
3532
pass
3633

3734
type = 'knowledge-write-node'
38-
35+
support = [WorkflowMode.KNOWLEDGE]

0 commit comments

Comments
 (0)