Explorar el Código

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

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/0.14.2
-LAN- hace 10 meses
padre
commit
c07d9e96ce
No account linked to committer's email address

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

@@ -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 Ver fichero

@@ -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,

Cargando…
Cancelar
Guardar