Pass failure details correctly

Author: andystaples

durabletask/internal/exceptions.py

@@ -1,7 +1,11 @@
+import orchestrator_service_pb2 as pb
+
+
 class VersionFailureException(Exception):
-    pass
+    def __init__(self, error_details: pb.TaskFailureDetails) -> None:
+        super().__init__()
+        self.error_details = error_details
 
 
 class AbandonOrchestrationError(Exception):
-    def __init__(self, *args: object) -> None:
-        super().__init__(*args)
+    pass

durabletask/worker.py

@@ -970,7 +970,6 @@ def execute(
             )
 
         ctx = _RuntimeOrchestrationContext(instance_id, self._registry)
-        version_failure = None
         try:
             # Rebuild local state by replaying old history into the orchestrator function
             self._logger.debug(
@@ -992,8 +991,8 @@ def execute(
 
         except pe.VersionFailureException as ex:
             if self._registry.versioning and self._registry.versioning.failure_strategy == VersionFailureStrategy.FAIL:
-                if version_failure:
-                    ctx.set_failed(version_failure)
+                if ex.error_details:
+                    ctx.set_failed(ex.error_details)
                 else:
                     ctx.set_failed(ex)
             elif self._registry.versioning and self._registry.versioning.failure_strategy == VersionFailureStrategy.REJECT:
@@ -1062,7 +1061,7 @@ def process_event(
                             f"Error action = '{self._registry.versioning.failure_strategy}'. "
                             f"Version error = '{version_failure}'"
                         )
-                        raise pe.VersionFailureException
+                        raise pe.VersionFailureException(version_failure)
 
                 # deserialize the input, if any
                 input = None