feat: add TaskFailed message to TaskFailureContext

Signed-off-by: Yves Bastide <yves@botify.com>

Author: ybastide

examples/failing.py

@@ -54,7 +54,9 @@ def on_task_failure(self, failure_context: TaskFailureContext) -> TaskFailureCon
             colorize(
                 YELLOW,
                 f"FailingWorkflow.on_task_failure: {failure_context.task_name}: {failure_context.exception!r}"
-                f" (started_id: {failure_context.event.get('started_id')})",
+                f" (started_id: {failure_context.event.get('started_id')})"
+                f" message: {failure_context.task_message!r}"
+                f" {failure_context.task_error!r} ({failure_context.task_error_type})",
             )
         )
         return None  # no specific handling

simpleflow/task.py

@@ -318,6 +318,7 @@ class Decision(Enum):
     reason: str | None = attr.ib(default=None)
     details: Any = attr.ib(default=None)
     task_error: str | None = attr.ib(default=None)
+    task_message: str | None = attr.ib(default=None)
     task_error_type: type[Exception] | None = attr.ib(default=None)
 
     def __attrs_post_init__(self):
@@ -334,6 +335,8 @@ def __attrs_post_init__(self):
                 if isinstance(details, dict):
                     if "error" in details:
                         self.task_error = details["error"]
+                    if "message" in details:
+                        self.task_message = details["message"]
                     if "error_type" in details:
                         try:
                             self.task_error_type = import_from_module(details["error_type"])