compatible query is None

langgenius · Apr 3, 2024 · 85e568e · 85e568e
1 parent 0026cb4
commit 85e568e
Showing 1 changed file with 6 additions and 0 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
@@ -49,6 +49,12 @@ def _run(self, variable_pool: VariablePool) -> NodeRunResult:
         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(