Set exc_info In Step Retries (#428)

kraftp · web-flow · commit 9d66c8cbb73c · 2025-08-07T15:56:41.000-07:00
Addresses #426
diff --git a/dbos/_core.py b/dbos/_core.py
@@ -1073,7 +1073,8 @@ def invoke_step(*args: Any, **kwargs: Any) -> Any:
 
             def on_exception(attempt: int, error: BaseException) -> float:
                 dbos.logger.warning(
-                    f"Step being automatically retried. (attempt {attempt + 1} of {attempts}). {traceback.format_exc()}"
+                    f"Step being automatically retried (attempt {attempt + 1} of {attempts})",
+                    exc_info=error,
                 )
                 ctx = assert_current_dbos_context()
                 span = ctx.get_current_span()

Original file line number	Diff line number	Diff line change
`@@ -1073,7 +1073,8 @@ def invoke_step(args: Any, *kwargs: Any) -> Any:`
`1073`	`1073`
`1074`	`1074`	`def on_exception(attempt: int, error: BaseException) -> float:`
`1075`	`1075`	`dbos.logger.warning(`
`1076`		`- f"Step being automatically retried. (attempt {attempt + 1} of {attempts}). {traceback.format_exc()}"`
	`1076`	`+ f"Step being automatically retried (attempt {attempt + 1} of {attempts})",`
	`1077`	`+ exc_info=error,`
`1077`	`1078`	`)`
`1078`	`1079`	`ctx = assert_current_dbos_context()`
`1079`	`1080`	`span = ctx.get_current_span()`