From 5ed181dd427e9a0f2d6ce734652b91aeb06e8055 Mon Sep 17 00:00:00 2001 From: jyong Date: Mon, 18 Mar 2024 15:54:59 +0800 Subject: [PATCH] knowledge entities fix --- api/core/workflow/nodes/knowledge_retrieval/entities.py | 2 -- api/core/workflow/nodes/question_classifier/entities.py | 2 -- api/core/workflow/nodes/variable_assigner/entities.py | 2 -- 3 files changed, 6 deletions(-) diff --git a/api/core/workflow/nodes/knowledge_retrieval/entities.py b/api/core/workflow/nodes/knowledge_retrieval/entities.py index 8f5c828061..d6a5111a43 100644 --- a/api/core/workflow/nodes/knowledge_retrieval/entities.py +++ b/api/core/workflow/nodes/knowledge_retrieval/entities.py @@ -43,8 +43,6 @@ class KnowledgeRetrievalNodeData(BaseNodeData): """ Knowledge retrieval Node Data. """ - title: str - desc: str type: str = 'knowledge-retrieval' query_variable_selector: list[str] dataset_ids: list[str] diff --git a/api/core/workflow/nodes/question_classifier/entities.py b/api/core/workflow/nodes/question_classifier/entities.py index 4b8d431a76..5371862ea8 100644 --- a/api/core/workflow/nodes/question_classifier/entities.py +++ b/api/core/workflow/nodes/question_classifier/entities.py @@ -43,8 +43,6 @@ class QuestionClassifierNodeData(BaseNodeData): Knowledge retrieval Node Data. """ query_variable_selector: list[str] - title: str - desc: str type: str = 'question-classifier' model: ModelConfig classes: list[ClassConfig] diff --git a/api/core/workflow/nodes/variable_assigner/entities.py b/api/core/workflow/nodes/variable_assigner/entities.py index 177213a707..8a205810e0 100644 --- a/api/core/workflow/nodes/variable_assigner/entities.py +++ b/api/core/workflow/nodes/variable_assigner/entities.py @@ -7,8 +7,6 @@ class VariableAssignerNodeData(BaseNodeData): """ Knowledge retrieval Node Data. """ - title: str - desc: str type: str = 'variable-assigner' output_type: str variables: list[str]