Sfoglia il codice sorgente

fix(nodes): handle errors in question_classifier and parameter_extractor (#11927)

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 4 mesi fa
parent
commit
c07d9e96ce

+ 9 - 0
api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py

@@ -179,6 +179,15 @@ class ParameterExtractorNode(LLMNode):
                 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
 

+ 1 - 2
api/core/workflow/nodes/question_classifier/question_classifier_node.py

@@ -154,8 +154,7 @@ class QuestionClassifierNode(LLMNode):
                 },
                 llm_usage=usage,
             )
-
-        except ValueError as e:
+        except Exception as e:
             return NodeRunResult(
                 status=WorkflowNodeExecutionStatus.FAILED,
                 inputs=variables,