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

Author: viniciusdsmello

src/openlayer/lib/integrations/litellm_tracer.py

@@ -156,12 +156,21 @@ def stream_chunks(
     provider = "unknown"
     latest_chunk_metadata = {}
 
+    # #region agent log - Debug: Check parent step before creating LiteLLM step
+    _parent = tracer.get_current_step()
+    _trace = tracer.get_current_trace()
+    print(f"[OPENLAYER_DEBUG] litellm_tracer:stream_chunks | BEFORE create_step | parent_step={_parent.name if _parent else None} | has_trace={_trace is not None}", flush=True)
+    # #endregion
+    
     # Create step immediately so it's added to parent trace before parent publishes
     with tracer.create_step(
         name="LiteLLM Chat Completion",
         step_type=tracer_enums.StepType.CHAT_COMPLETION,
         inputs={"prompt": kwargs.get("messages", [])},
     ) as step:
+        # #region agent log - Debug: Check if step is root after creation
+        print(f"[OPENLAYER_DEBUG] litellm_tracer:stream_chunks | AFTER create_step | step_id={step.id[:8] if step.id else None} | is_root={_parent is None}", flush=True)
+        # #endregion
         try:
             i = 0
             for i, chunk in enumerate(chunks):

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:_create_and_initialize_step | step_name={step_name} | is_root_step={is_root_step} | parent={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,9 @@ def _handle_trace_completion(
     on_flush_failure: Optional[OnFlushFailureCallback] = None,
 ) -> None:
     """Handle trace completion and data streaming."""
+    # #region agent log - Debug: Trace completion entry
+    print(f"[OPENLAYER_DEBUG] tracer:_handle_trace_completion | step_name={step_name} | is_root_step={is_root_step}", flush=True)
+    # #endregion
     if is_root_step:
         logger.debug("Ending the trace...")
         current_trace = get_current_trace()
@@ -1398,6 +1405,9 @@ def _handle_trace_completion(
             client = _get_client()
 
             if client:
+                # #region agent log - Debug: About to publish trace
+                print(f"[OPENLAYER_DEBUG] tracer:_handle_trace_completion | PUBLISHING | inferenceId={trace_data.get('inferenceId', 'unknown')[:8]} | step_name={step_name}", flush=True)
+                # #endregion
                 try:
                     response = client.inference_pipelines.data.stream(
                         inference_pipeline_id=inference_pipeline_id,