Fix block execution status in case of error (#8267)

This commit is contained in:
vishesh10
2024-10-10 08:29:26 +05:30
committed by GitHub
parent c03e2fb949
commit 7b92bae942

View File

@@ -272,6 +272,8 @@ class Block(ABC, Generic[BlockSchemaInputType, BlockSchemaOutputType]):
for output_name, output_data in self.run(
self.input_schema(**input_data), **kwargs
):
if "error" in output_name:
raise ValueError(output_data)
if error := self.output_schema.validate_field(output_name, output_data):
raise ValueError(f"Block produced an invalid output data: {error}")
yield output_name, output_data