dex-python-sdk 0.1.10__tar.gz → 0.1.11__tar.gz

{dex_python_sdk-0.1.10 → dex_python_sdk-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dex-python-sdk
-Version: 0.1.10
+Version: 0.1.11
 Requires-Dist: grpcio>=1.83.0
 Requires-Dist: grpcio-status>=1.83.0
 Requires-Dist: protobuf>=7.35.1

{dex_python_sdk-0.1.10 → dex_python_sdk-0.1.11}/dex/_grpc_errors.py

@@ -8,6 +8,8 @@
 
 from __future__ import annotations
 
+import traceback
+from dataclasses import dataclass
 from typing import Any, Literal, cast
 
 import grpc
@@ -29,6 +31,28 @@ from dex.runtime_errors import (
 
 FlowTargetRequirement = Literal["none", "existing", "active"]
 
+MAX_WORKER_STACK_TRACE_BYTES = 16 * 1024
+_STACK_TRACE_TRUNCATION_MARKER = b"\n... stack trace truncated by Dex Python SDK ..."
+
+
+@dataclass(frozen=True)
+class RetryAfterError(Exception):
+    """Requests a delay before the next retry while preserving the current failure."""
+
+    after_seconds: int
+    cause: BaseException
+
+    def __str__(self) -> str:
+        return str(self.cause)
+
+
+def retry_after(after_seconds: int, cause: BaseException) -> RetryAfterError:
+    if cause is None:
+        raise ValueError("cause is required")
+    if after_seconds <= 0:
+        raise ValueError("after_seconds must be positive")
+    return RetryAfterError(after_seconds=after_seconds, cause=cause)
+
 
 def translate_rpc_error(
     error: grpc.RpcError,
@@ -118,11 +142,21 @@ async def async_abort_worker_error(
 
 
 def _worker_error_status(error: BaseException) -> status_pb2.Status:
-    message = str(error) or type(error).__name__
+    retry_after_error: RetryAfterError | None = None
+    reported = error
+    if isinstance(error, RetryAfterError):
+        retry_after_error = error
+        reported = error.cause
+
+    message = str(reported) or type(reported).__name__
+    stack_trace_source = error if retry_after_error is not None else reported
     worker_error = pb.WorkerErrorResponse(
         detail=message,
-        error_type=f"{type(error).__module__}.{type(error).__qualname__}",
+        error_type=f"{type(reported).__module__}.{type(reported).__qualname__}",
+        stack_trace=_worker_stack_trace(stack_trace_source),
     )
+    if retry_after_error is not None:
+        worker_error.retry_after_seconds = retry_after_error.after_seconds
     packed = any_pb2.Any()
     packed.Pack(worker_error)
     return status_pb2.Status(
@@ -132,6 +166,22 @@ def _worker_error_status(error: BaseException) -> status_pb2.Status:
     )
 
 
+def _worker_stack_trace(error: BaseException) -> str:
+    if error.__traceback__ is None:
+        return ""
+    lines = traceback.format_exception(type(error), error, error.__traceback__)
+    encoded = "".join(lines).encode()
+    if len(encoded) <= MAX_WORKER_STACK_TRACE_BYTES:
+        return encoded.decode()
+    prefix_length = MAX_WORKER_STACK_TRACE_BYTES - len(_STACK_TRACE_TRUNCATION_MARKER)
+    while prefix_length > 0 and (encoded[prefix_length] & 0xC0) == 0x80:
+        prefix_length -= 1
+    return (
+        encoded[:prefix_length].decode(errors="replace")
+        + _STACK_TRACE_TRUNCATION_MARKER.decode()
+    )
+
+
 def _map_sub_status(value: int) -> ErrorSubStatus:
     statuses: dict[int, ErrorSubStatus] = {
         int(pb.ERROR_SUB_STATUS_UNCATEGORIZED): ErrorSubStatus.UNCATEGORIZED,

{dex_python_sdk-0.1.10 → dex_python_sdk-0.1.11}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "dex-python-sdk"
 # Publish uses tag sdk-python/v*; CI stamps this for release builds.
-version = "0.1.10"
+version = "0.1.11"
 description = "Python SDK for the Dex workflow engine"
 authors = [{name = "Super Durable"}]
 readme = "README.md"