compatible query is None

2024-04-03 18:39:29 +08:00 · 2024-04-03 18:39:29 +08:00 · 85e568e114
parent 0026cb404f
commit 85e568e114
1 changed files with 6 additions and 0 deletions
--- 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(