From 9e4b39e19f142e9f4a840ec79d71cd385b275c24 Mon Sep 17 00:00:00 2001 From: jyong Date: Wed, 27 Mar 2024 17:03:49 +0800 Subject: [PATCH] fix question classifier node type --- .../nodes/knowledge_retrieval/knowledge_retrieval_node.py | 2 +- .../nodes/question_classifier/question_classifier_node.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py index 5af838e057..a3f9bcd539 100644 --- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py +++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py @@ -39,7 +39,7 @@ default_retrieval_model = { class KnowledgeRetrievalNode(BaseNode): _node_data_cls = KnowledgeRetrievalNodeData - _node_type = NodeType.KNOWLEDGE_RETRIEVAL + node_type = NodeType.KNOWLEDGE_RETRIEVAL def _run(self, variable_pool: VariablePool) -> NodeRunResult: node_data: KnowledgeRetrievalNodeData = cast(self._node_data_cls, self.node_data) diff --git a/api/core/workflow/nodes/question_classifier/question_classifier_node.py b/api/core/workflow/nodes/question_classifier/question_classifier_node.py index 832964fa59..01b2908f85 100644 --- a/api/core/workflow/nodes/question_classifier/question_classifier_node.py +++ b/api/core/workflow/nodes/question_classifier/question_classifier_node.py @@ -28,7 +28,7 @@ from models.workflow import WorkflowNodeExecutionStatus class QuestionClassifierNode(LLMNode): _node_data_cls = QuestionClassifierNodeData - _node_type = NodeType.QUESTION_CLASSIFIER + node_type = NodeType.QUESTION_CLASSIFIER def _run(self, variable_pool: VariablePool) -> NodeRunResult: node_data: QuestionClassifierNodeData = cast(self._node_data_cls, self.node_data)