feat: add traceback to error message

parea-ai · Jul 1, 2024 · 22a642c · 22a642c
1 parent 7076a93
commit 22a642c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/parea/utils/trace_utils.py b/parea/utils/trace_utils.py
@@ -6,6 +6,7 @@
 import logging
 import os
 import threading
+import traceback
 from collections import ChainMap
 from datetime import datetime
 from functools import wraps
@@ -263,7 +264,7 @@ async def async_wrapper(*args, **kwargs):
                 return result
             except Exception as e:
                 logger.exception(f"Error occurred in function {func.__name__}, {e}")
-                fill_trace_data(trace_id, {"error": str(e)}, UpdateTraceScenario.ERROR)
+                fill_trace_data(trace_id, {"error": traceback.format_exc()}, UpdateTraceScenario.ERROR)
                 raise e
             finally:
                 try:
@@ -283,7 +284,7 @@ def wrapper(*args, **kwargs):
                 return result
             except Exception as e:
                 logger.exception(f"Error occurred in function {func.__name__}, {e}")
-                fill_trace_data(trace_id, {"error": str(e)}, UpdateTraceScenario.ERROR)
+                fill_trace_data(trace_id, {"error": traceback.format_exc()}, UpdateTraceScenario.ERROR)
                 raise e
             finally:
                 try: