Refactor tracing functions to replace update_current_span with update_current_step

Author: viniciusdsmello

examples/tracing/trace_metadata_updates.py

@@ -15,7 +15,7 @@
 os.environ["OPENLAYER_API_KEY"] = "your-api-key-here"
 os.environ["OPENLAYER_INFERENCE_PIPELINE_ID"] = "your-pipeline-id-here"
 
-from openlayer.lib import trace, trace_async, update_current_trace, update_current_span
+from openlayer.lib import trace, trace_async, update_current_trace, update_current_step
 
 
 class UserSession:
@@ -78,7 +78,7 @@ def preprocess_request(self, text: str, user_session: UserSession) -> str:
         """Preprocess user request with step-level metadata."""
 
         # Update current step with preprocessing context
-        update_current_span(
+        update_current_step(
             metadata={
                 "preprocessing_type": "standard",
                 "user_preferences_applied": True,
@@ -104,7 +104,7 @@ def generate_response(self, processed_text: str, user_session: UserSession) -> s
         # Set model-specific metadata
         model_version = "gpt-4" if user_session.preferences.get("tier") == "premium" else "gpt-3.5-turbo"
 
-        update_current_span(
+        update_current_step(
             metadata={
                 "model_used": model_version,
                 "temperature": 0.7,
@@ -131,7 +131,7 @@ def generate_response(self, processed_text: str, user_session: UserSession) -> s
     def postprocess_response(self, response: str, user_session: UserSession) -> str:
         """Postprocess response with personalization metadata."""
 
-        update_current_span(
+        update_current_step(
             metadata={
                 "personalization_applied": True,
                 "content_filtering": user_session.preferences.get("content_filter", "moderate"),
@@ -230,7 +230,7 @@ def error_handling_example():
 
     try:
         # Simulate some processing
-        update_current_span(
+        update_current_step(
             metadata={"processing_step": "initial_validation"}
         )
 
@@ -265,7 +265,7 @@ async def async_example():
     # Simulate async processing steps
     import asyncio
 
-    update_current_span(
+    update_current_step(
         metadata={"step": "async_sleep_simulation"}
     )
     await asyncio.sleep(0.1)

src/openlayer/lib/__init__.py

@@ -14,7 +14,7 @@
     "trace_oci_genai",
     "trace_oci",  # Alias for backward compatibility
     "update_current_trace",
-    "update_current_span"
+    "update_current_step"
 ]
 
 # ---------------------------------- Tracing --------------------------------- #
@@ -24,7 +24,7 @@
 trace = tracer.trace
 trace_async = tracer.trace_async
 update_current_trace = tracer.update_current_trace
-update_current_span = tracer.update_current_span
+update_current_step = tracer.update_current_step
 
 
 def trace_anthropic(client):

src/openlayer/lib/tracing/tracer.py

@@ -560,11 +560,11 @@ def update_current_trace(**kwargs) -> None:
     logger.debug("Updated current trace metadata")
 
 
-def update_current_span(
+def update_current_step(
     attributes: Optional[Dict[str, Any]] = None,
     metadata: Optional[Dict[str, Any]] = None
 ) -> None:
-    """Updates the current step (span) with the provided attributes.
+    """Updates the current step with the provided attributes.
     
     This function allows users to set step-level metadata dynamically
     during execution.
@@ -574,20 +574,20 @@ def update_current_span(
         metadata: Optional dictionary of metadata to merge with existing metadata
         
     Example:
-        >>> from openlayer.lib import trace, update_current_span
+        >>> from openlayer.lib import trace, update_current_step
         >>> 
         >>> @trace()
         >>> def my_function():
         >>>     # Update current step with additional context
-        >>>     update_current_span(
+        >>>     update_current_step(
         >>>         metadata={"model_version": "v1.2.3"}
         >>>     )
         >>>     return "result"
     """
     current_step = get_current_step()
     if current_step is None:
         logger.warning(
-            "update_current_span() called without an active step. "
+            "update_current_step() called without an active step. "
             "Make sure to call this function within a traced context "
             "(e.g., inside a function decorated with @trace)."
         )
@@ -895,9 +895,9 @@ def post_process_trace(
         **root_step.metadata,
     }
 
-    # Include trace-level metadata if set
+    # Include trace-level metadata if set - extract keys to row/record level
     if trace_obj.metadata is not None:
-        # Merge trace-level metadata (higher precedence than root step metadata)
+        # Add each trace metadata key directly to the row/record level
         trace_data.update(trace_obj.metadata)
 
     if root_step.ground_truth: