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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dex-python-sdk
-Version: 0.1.10
+Version: 0.2.0
 Requires-Dist: grpcio>=1.83.0
 Requires-Dist: grpcio-status>=1.83.0
 Requires-Dist: protobuf>=7.35.1
@@ -193,7 +193,7 @@ options = dex.StartFlowOptions(
 Register async hooks with `allow_async_handlers=True` and run them with
 `AsyncWorker`. `FAIL` produces `FlowErrorType.FLOW_TIMEOUT` and permits Flow
 retry; `CANCEL` cancels without retry. Continue-as-new preserves the deadline,
-while retry and cron runs receive a fresh budget. A zero or absent timeout
+while retry runs receive a fresh budget. A zero or absent timeout
 disables the feature.
 
 `Registry` validates every Flow, Step, RPC signature, durable name, lock, and

{dex_python_sdk-0.1.10 → dex_python_sdk-0.2.0}/README.md

@@ -177,7 +177,7 @@ options = dex.StartFlowOptions(
 Register async hooks with `allow_async_handlers=True` and run them with
 `AsyncWorker`. `FAIL` produces `FlowErrorType.FLOW_TIMEOUT` and permits Flow
 retry; `CANCEL` cancels without retry. Continue-as-new preserves the deadline,
-while retry and cron runs receive a fresh budget. A zero or absent timeout
+while retry runs receive a fresh budget. A zero or absent timeout
 disables the feature.
 
 `Registry` validates every Flow, Step, RPC signature, durable name, lock, and

{dex_python_sdk-0.1.10 → dex_python_sdk-0.2.0}/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.2.0}/dex/async_client.py

@@ -1018,7 +1018,6 @@ class AsyncClient:
                 ),
                 IdReusePolicy.DISALLOW: pb.ID_REUSE_POLICY_DISALLOW_REUSE,
             }[options.id_reuse_policy],
-            cron_schedule=options.cron_schedule or "",
             flow_already_started_options=pb.FlowAlreadyStartedOptions(
                 ignore_already_started_error=options.ignore_already_started
             ),

{dex_python_sdk-0.1.10 → dex_python_sdk-0.2.0}/dex/client.py

@@ -1019,7 +1019,6 @@ class Client:
                 ),
                 IdReusePolicy.DISALLOW: pb.ID_REUSE_POLICY_DISALLOW_REUSE,
             }[options.id_reuse_policy],
-            cron_schedule=options.cron_schedule or "",
             flow_already_started_options=pb.FlowAlreadyStartedOptions(
                 ignore_already_started_error=options.ignore_already_started
             ),