From 85e568e114cdc847af55129c4757608520c814eb Mon Sep 17 00:00:00 2001 From: jyong Date: Wed, 3 Apr 2024 18:39:29 +0800 Subject: [PATCH] compatible query is None --- .../nodes/knowledge_retrieval/knowledge_retrieval_node.py | 6 ++++++ 1 file changed, 6 insertions(+) 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 ea370c908b..e9aa655401 100644 --- a/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py +++ b/api/core/workflow/nodes/knowledge_retrieval/knowledge_retrieval_node.py @@ -49,6 +49,12 @@ class KnowledgeRetrievalNode(BaseNode): variables = { 'query': query } + if not query: + return NodeRunResult( + status=WorkflowNodeExecutionStatus.FAILED, + inputs=variables, + error="Query is required." + ) # retrieve knowledge try: results = self._fetch_dataset_retriever(