durabletask 1.2.0.dev8__tar.gz → 1.2.0.dev10__tar.gz

{durabletask-1.2.0.dev8 → durabletask-1.2.0.dev10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durabletask
-Version: 1.2.0.dev8
+Version: 1.2.0.dev10
 Summary: A Durable Task Client SDK for Python
 License:     MIT License
         

{durabletask-1.2.0.dev8 → durabletask-1.2.0.dev10}/durabletask/internal/helpers.py

@@ -116,11 +116,18 @@ def new_sub_orchestration_failed_event(event_id: int, ex: Exception) -> pb.Histo
     )
 
 
-def new_failure_details(ex: Exception) -> pb.TaskFailureDetails:
+def new_failure_details(ex: Exception, _visited: Optional[set[int]] = None) -> pb.TaskFailureDetails:
+    if _visited is None:
+        _visited = set()
+    _visited.add(id(ex))
+    inner: Optional[BaseException] = ex.__cause__ or ex.__context__
+    if len(_visited) > 10 or (inner and id(inner) in _visited) or not isinstance(inner, Exception):
+        inner = None
     return pb.TaskFailureDetails(
         errorType=type(ex).__name__,
         errorMessage=str(ex),
-        stackTrace=wrappers_pb2.StringValue(value=''.join(traceback.format_tb(ex.__traceback__)))
+        stackTrace=wrappers_pb2.StringValue(value=''.join(traceback.format_tb(ex.__traceback__))),
+        innerFailure=new_failure_details(inner, _visited) if inner else None
     )
 
 

{durabletask-1.2.0.dev8 → durabletask-1.2.0.dev10}/durabletask/task.py

@@ -117,7 +117,7 @@ class OrchestrationContext(ABC):
     def call_activity(self, activity: Union[Activity[TInput, TOutput], str], *,
                       input: Optional[TInput] = None,
                       retry_policy: Optional[RetryPolicy] = None,
-                      tags: Optional[dict[str, str]] = None) -> Task[TOutput]:
+                      tags: Optional[dict[str, str]] = None) -> CompletableTask[TOutput]:
         """Schedule an activity for execution.
 
         Parameters
@@ -142,7 +142,7 @@ class OrchestrationContext(ABC):
     def call_entity(self,
                     entity: EntityInstanceId,
                     operation: str,
-                    input: Optional[TInput] = None) -> Task:
+                    input: Optional[TInput] = None) -> CompletableTask:
         """Schedule entity function for execution.
 
         Parameters
@@ -182,7 +182,7 @@ class OrchestrationContext(ABC):
         pass
 
     @abstractmethod
-    def lock_entities(self, entities: list[EntityInstanceId]) -> Task[EntityLock]:
+    def lock_entities(self, entities: list[EntityInstanceId]) -> CompletableTask[EntityLock]:
         """Creates a Task object that locks the specified entity instances.
 
         The locks will be acquired the next time the orchestrator yields.
@@ -206,7 +206,7 @@ class OrchestrationContext(ABC):
                               input: Optional[TInput] = None,
                               instance_id: Optional[str] = None,
                               retry_policy: Optional[RetryPolicy] = None,
-                              version: Optional[str] = None) -> Task[TOutput]:
+                              version: Optional[str] = None) -> CompletableTask[TOutput]:
         """Schedule sub-orchestrator function for execution.
 
         Parameters
@@ -231,7 +231,7 @@ class OrchestrationContext(ABC):
     # TOOD: Add a timeout parameter, which allows the task to be canceled if the event is
     # not received within the specified timeout. This requires support for task cancellation.
     @abstractmethod
-    def wait_for_external_event(self, name: str) -> Task:
+    def wait_for_external_event(self, name: str) -> CompletableTask:
         """Wait asynchronously for an event to be raised with the name `name`.
 
         Parameters
@@ -302,8 +302,10 @@ class FailureDetails:
 class TaskFailedError(Exception):
     """Exception type for all orchestration task failures."""
 
-    def __init__(self, message: str, details: pb.TaskFailureDetails):
+    def __init__(self, message: str, details: Union[pb.TaskFailureDetails, Exception]):
         super().__init__(message)
+        if isinstance(details, Exception):
+            details = pbh.new_failure_details(details)
         self._details = FailureDetails(
             details.errorMessage,
             details.errorType,
@@ -424,7 +426,7 @@ class CompletableTask(Task[T]):
         if self._parent is not None:
             self._parent.on_child_completed(self)
 
-    def fail(self, message: str, details: pb.TaskFailureDetails):
+    def fail(self, message: str, details: Union[Exception, pb.TaskFailureDetails]):
         if self._is_complete:
             raise ValueError('The task has already completed.')
         self._exception = TaskFailedError(message, details)

{durabletask-1.2.0.dev8 → durabletask-1.2.0.dev10}/durabletask/worker.py

@@ -1014,7 +1014,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
             self,
             fire_at: Union[datetime, timedelta],
             retryable_task: Optional[task.RetryableTask] = None,
-    ) -> task.Task:
+    ) -> task.TimerTask:
         id = self.next_sequence_number()
         if isinstance(fire_at, timedelta):
             fire_at = self.current_utc_datetime + fire_at
@@ -1034,7 +1034,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
             input: Optional[TInput] = None,
             retry_policy: Optional[task.RetryPolicy] = None,
             tags: Optional[dict[str, str]] = None,
-    ) -> task.Task[TOutput]:
+    ) -> task.CompletableTask[TOutput]:
         id = self.next_sequence_number()
 
         self.call_activity_function_helper(
@@ -1047,7 +1047,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
             entity: EntityInstanceId,
             operation: str,
             input: Optional[TInput] = None,
-    ) -> task.Task:
+    ) -> task.CompletableTask:
         id = self.next_sequence_number()
 
         self.call_entity_function_helper(
@@ -1068,7 +1068,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
             id, entity_id, operation_name, input
         )
 
-    def lock_entities(self, entities: list[EntityInstanceId]) -> task.Task[EntityLock]:
+    def lock_entities(self, entities: list[EntityInstanceId]) -> task.CompletableTask[EntityLock]:
         id = self.next_sequence_number()
 
         self.lock_entities_function_helper(
@@ -1084,7 +1084,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
             instance_id: Optional[str] = None,
             retry_policy: Optional[task.RetryPolicy] = None,
             version: Optional[str] = None,
-    ) -> task.Task[TOutput]:
+    ) -> task.CompletableTask[TOutput]:
         id = self.next_sequence_number()
         if isinstance(orchestrator, str):
             orchestrator_name = orchestrator
@@ -1229,7 +1229,7 @@ class _RuntimeOrchestrationContext(task.OrchestrationContext):
             action = pb.OrchestratorAction(id=task_id, sendEntityMessage=entity_unlock_message)
             self._pending_actions[task_id] = action
 
-    def wait_for_external_event(self, name: str) -> task.Task:
+    def wait_for_external_event(self, name: str) -> task.CompletableTask:
         # Check to see if this event has already been received, in which case we
         # can return it immediately. Otherwise, record out intent to receive an
         # event with the given name so that we can resume the generator when it

{durabletask-1.2.0.dev8 → durabletask-1.2.0.dev10}/durabletask.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durabletask
-Version: 1.2.0.dev8
+Version: 1.2.0.dev10
 Summary: A Durable Task Client SDK for Python
 License:     MIT License
         

{durabletask-1.2.0.dev8 → durabletask-1.2.0.dev10}/pyproject.toml

@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "durabletask"
-version = "1.2.0.dev8"
+version = "1.2.0.dev10"
 description = "A Durable Task Client SDK for Python"
 keywords = [
     "durable",