feat(knowledge_retrieval_node): Suppress exceptions thrown by DatasetRetrieval (#11728)

Signed-off-by: -LAN- <laipz8200@outlook.com>
2024-12-17 13:12:29 +08:00 · 2024-12-17 13:12:29 +08:00 · 62b9e5a6f9
commit 62b9e5a6f9
parent a399502ecd
2 changed files with 31 additions and 14 deletions
--- a/api/core/workflow/nodes/base/node.py
+++ b/api/core/workflow/nodes/base/node.py
@ -72,7 +72,11 @@ class BaseNode(Generic[GenericNodeData]):
            result = self._run()
        except Exception as e:
            logger.exception(f"Node {self.node_id} failed to run")
-            result = NodeRunResult(status=WorkflowNodeExecutionStatus.FAILED, error=str(e), error_type="SystemError")
+            result = NodeRunResult(
+                status=WorkflowNodeExecutionStatus.FAILED,
+                error=str(e),
+                error_type="WorkflowNodeError",
+            )

        if isinstance(result, NodeRunResult):
            yield RunCompletedEvent(run_result=result)