feat(tracer): add debug logging for step creation and trace completion

Author: viniciusdsmello

src/openlayer/lib/integrations/litellm_tracer.py

@@ -204,6 +204,11 @@ def stream_chunks(
     except Exception as e:
         logger.error("Failed to yield chunk. %s", e)
     finally:
+        # #region agent log - Debug: trace finally block execution
+        _parent = tracer.get_current_step()
+        _trace = tracer.get_current_trace()
+        print(f"[OPENLAYER_DEBUG] litellm_tracer.py:finally | has_parent_step={_parent is not None} | parent_step_name={_parent.name if _parent else None} | has_trace={_trace is not None} | trace_steps_count={len(_trace.steps) if _trace else 0}", flush=True)
+        # #endregion
         # Try to add step to the trace
         try:
             collected_output_data = [message for message in collected_output_data if message is not None]

src/openlayer/lib/tracing/tracer.py

@@ -1322,6 +1322,10 @@ def _create_and_initialize_step(
     parent_step = get_current_step()
     is_root_step = parent_step is None
 
+    # #region agent log - Debug: step creation
+    print(f"[OPENLAYER_DEBUG] tracer.py:_create_and_initialize_step | step_name={step_name} | step_type={step_type} | is_root_step={is_root_step} | parent_step_name={parent_step.name if parent_step else None}", flush=True)
+    # #endregion
+
     if parent_step is None:
         logger.debug("Starting a new trace...")
         current_trace = traces.Trace()
@@ -1345,6 +1349,16 @@ def _handle_trace_completion(
     on_flush_failure: Optional[OnFlushFailureCallback] = None,
 ) -> None:
     """Handle trace completion and data streaming."""
+    # #region agent log - Debug: trace completion
+    _trace = get_current_trace()
+    _steps = [s.name for s in _trace.steps] if _trace and _trace.steps else []
+    _nested = []
+    if _trace and _trace.steps:
+        for s in _trace.steps:
+            if hasattr(s, 'steps') and s.steps:
+                _nested.extend([ns.name for ns in s.steps])
+    print(f"[OPENLAYER_DEBUG] tracer.py:_handle_trace_completion | step_name={step_name} | is_root_step={is_root_step} | has_trace={_trace is not None} | root_steps={_steps} | nested_steps={_nested}", flush=True)
+    # #endregion
     if is_root_step:
         logger.debug("Ending the trace...")
         current_trace = get_current_trace()