fix(nodes): handle errors in question_classifier and parameter_extrac…

…tor (#11927) Signed-off-by: -LAN- <laipz8200@outlook.com>
langgenius · Dec 21, 2024 · c07d9e9 · c07d9e9
1 parent 810adb8
commit c07d9e9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py b/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py
@@ -179,6 +179,15 @@ def _run(self):
                 error=str(e),
                 metadata={},
             )
+        except Exception as e:
+            return NodeRunResult(
+                status=WorkflowNodeExecutionStatus.FAILED,
+                inputs=inputs,
+                process_data=process_data,
+                outputs={"__is_success": 0, "__reason": "Failed to invoke model", "__error": str(e)},
+                error=str(e),
+                metadata={},
+            )
 
         error = None
 

diff --git a/api/core/workflow/nodes/question_classifier/question_classifier_node.py b/api/core/workflow/nodes/question_classifier/question_classifier_node.py
@@ -154,8 +154,7 @@ def _run(self):
                 },
                 llm_usage=usage,
             )
-
-        except ValueError as e:
+        except Exception as e:
             return NodeRunResult(
                 status=WorkflowNodeExecutionStatus.FAILED,
                 inputs=variables,