Update tracing.py

Author: kritinv

deepeval/tracing/tracing.py

@@ -969,13 +969,14 @@ def decorator(func):
 
         # Async generator function
         if inspect.isasyncgenfunction(func):
+
             @functools.wraps(func)
             def asyncgen_wrapper(*args, **func_kwargs):
 
                 sig = inspect.signature(func)
                 bound = sig.bind(*args, **func_kwargs)
                 bound.apply_defaults()
-                
+
                 complete_kwargs = dict(bound.arguments)
                 if "self" in complete_kwargs:
                     complete_kwargs["self"] = replace_self_with_class_name(
@@ -995,6 +996,7 @@ def asyncgen_wrapper(*args, **func_kwargs):
                 )
                 observer.__enter__()
                 agen = func(*args, **func_kwargs)
+
                 async def gen():
                     try:
                         async for chunk in agen:
@@ -1003,13 +1005,15 @@ async def gen():
                     except Exception as e:
                         observer.__exit__(type(e), e, e.__traceback__)
                         raise
+
                 return gen()
-                
+
             setattr(asyncgen_wrapper, "_is_deepeval_observed", True)
             return asyncgen_wrapper
 
         # Sync generator function
         if inspect.isgeneratorfunction(func):
+
             @functools.wraps(func)
             def gen_wrapper(*args, **func_kwargs):
 
@@ -1026,7 +1030,7 @@ def gen_wrapper(*args, **func_kwargs):
                     "observe_kwargs": observe_kwargs,
                     "function_kwargs": make_json_serializable(complete_kwargs),
                 }
-                
+
                 observer = Observer(
                     type,
                     metrics=metrics,
@@ -1036,13 +1040,15 @@ def gen_wrapper(*args, **func_kwargs):
                 )
                 observer.__enter__()
                 original_gen = func(*args, **func_kwargs)
+
                 def gen():
                     try:
                         yield from original_gen
                         observer.__exit__(None, None, None)
                     except Exception as e:
                         observer.__exit__(type(e), e, e.__traceback__)
                         raise
+
                 return gen()
 
             setattr(gen_wrapper, "_is_deepeval_observed", True)