chore: cleanup capture exception impl

Author: abhishekg999

src/judgeval/common/tracer.py

@@ -685,7 +685,17 @@ def save(self, overwrite: bool = False) -> Tuple[str, dict]:
     def delete(self):
         return self.trace_manager_client.delete_trace(self.trace_id)
 
-
+def _capture_exception_for_trace(current_trace: Optional['TraceClient'], exc_info: Tuple[Optional[type], Optional[BaseException], Optional[types.TracebackType]]):
+    if not current_trace:
+        return
+
+    exc_type, exc_value, exc_traceback_obj = exc_info
+    formatted_exception = {
+        "type": exc_type.__name__ if exc_type else "UnknownExceptionType",
+        "message": str(exc_value) if exc_value else "No exception message",
+        "traceback": traceback.format_tb(exc_traceback_obj) if exc_traceback_obj else []
+    }
+    current_trace.record_error(formatted_exception)
 class _DeepTracer:
     _instance: Optional["_DeepTracer"] = None
     _lock: threading.Lock = threading.Lock()
@@ -877,14 +887,11 @@ def _trace(self, frame: types.FrameType, event: str, arg: Any):
                 current_span_var.reset(frame.f_locals["_judgment_span_token"])
 
         elif event == "exception":
-            exc_type, exc_value, exc_traceback = arg
-            formatted_exception = {
-                "type": exc_type.__name__,
-                "message": str(exc_value),
-                "traceback": traceback.format_tb(exc_traceback)
-            }
-            current_trace = current_trace_var.get()
-            current_trace.record_error(formatted_exception)
+            exc_type = arg[0]
+            if issubclass(exc_type, (StopIteration, StopAsyncIteration, GeneratorExit)):
+                return
+            _capture_exception_for_trace(current_trace, arg)
+            
 
         return self._trace
 
@@ -1163,14 +1170,7 @@ async def async_wrapper(*args, **kwargs):
                                 try:
                                     result = await func(*args, **kwargs)
                                 except Exception as e:
-                                    exc_type, exc_value, exc_traceback = sys.exc_info()
-                                    formatted_exception = {
-                                        "type": exc_type.__name__,
-                                        "message": str(exc_value),
-                                        "traceback": traceback.format_tb(exc_traceback)
-                                    }
-                                    current_trace = current_trace_var.get()
-                                    current_trace.record_error(formatted_exception)
+                                    _capture_exception_for_trace(current_trace, sys.exc_info())
                                     raise e
 
                             # Record output
@@ -1195,14 +1195,7 @@ async def async_wrapper(*args, **kwargs):
                             try:
                                 result = await func(*args, **kwargs)
                             except Exception as e:
-                                exc_type, exc_value, exc_traceback = sys.exc_info()
-                                formatted_exception = {
-                                    "type": exc_type.__name__,
-                                    "message": str(exc_value),
-                                    "traceback": traceback.format_tb(exc_traceback)
-                                }
-                                current_trace = current_trace_var.get()
-                                current_trace.record_error(formatted_exception)
+                                _capture_exception_for_trace(current_trace, sys.exc_info())
                                 raise e
 
                         span.record_output(result)
@@ -1252,14 +1245,7 @@ def wrapper(*args, **kwargs):
                                 try:
                                     result = func(*args, **kwargs)
                                 except Exception as e:
-                                    exc_type, exc_value, exc_traceback = sys.exc_info()
-                                    formatted_exception = {
-                                        "type": exc_type.__name__,
-                                        "message": str(exc_value),
-                                        "traceback": traceback.format_tb(exc_traceback)
-                                    }
-                                    current_trace = current_trace_var.get()
-                                    current_trace.record_error(formatted_exception)
+                                    _capture_exception_for_trace(current_trace, sys.exc_info())
                                     raise e
 
                             # Record output
@@ -1285,14 +1271,7 @@ def wrapper(*args, **kwargs):
                             try:
                                 result = func(*args, **kwargs)
                             except Exception as e:
-                                exc_type, exc_value, exc_traceback = sys.exc_info()
-                                formatted_exception = {
-                                    "type": exc_type.__name__,
-                                    "message": str(exc_value),
-                                    "traceback": traceback.format_tb(exc_traceback)
-                                }
-                                current_trace = current_trace_var.get()
-                                current_trace.record_error(formatted_exception)
+                                _capture_exception_for_trace(current_trace, sys.exc_info())
                                 raise e
 
                         span.record_output(result)