opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix question classifier node type

This commit is contained in:
jyong 2024-03-27 17:03:49 +08:00
parent 078b10a9f0
commit 9e4b39e19f
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py
View File

@ -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)

2
api/core/workflow/nodes/question_classifier/question_classifier_node.py
View File

@ -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)