fix(telemetryapi): repair unit test

Author: iamemilio

llama_stack/providers/inline/telemetry/meta_reference/sqlite_span_processor.py

@@ -112,6 +112,12 @@ def on_end(self, span: Span):
             if parent_context:
                 parent_span_id = format_span_id(parent_context.span_id)
 
+            # Determine if this span should be marked as root:
+            # 1. If it has LOCAL_ROOT_SPAN_MARKER attribute (explicitly marked by start_trace())
+            # 2. If it has no parent (implicit root span from FastAPI instrumentation)
+            is_root_span = span.attributes.get(LOCAL_ROOT_SPAN_MARKER) or parent_span_id is None
+            root_span_id_value = span_id if is_root_span else None
+
             # Insert into traces
             cursor.execute(
                 """
@@ -126,7 +132,7 @@ def on_end(self, span: Span):
                 (
                     trace_id,
                     service_name,
-                    (span_id if span.attributes.get(LOCAL_ROOT_SPAN_MARKER) else None),
+                    root_span_id_value,
                     datetime.fromtimestamp(span.start_time / 1e9, UTC).isoformat(),
                     datetime.fromtimestamp(span.end_time / 1e9, UTC).isoformat(),
                 ),

llama_stack/providers/utils/telemetry/sqlite_trace_store.py

@@ -19,6 +19,7 @@
     QueryCondition,
     QueryMetricsResponse,
     Span,
+    SpanStatus,
     SpanWithStatus,
     Trace,
 )
@@ -343,6 +344,11 @@ async def get_span_tree(
                     raise ValueError(f"Span {span_id} not found")
 
                 for row in rows:
+                    # Map OpenTelemetry status codes to API enum values
+                    # OTEL has: UNSET, OK, ERROR -> Map to SpanStatus enum
+                    otel_status = row["status"].upper()
+                    status = SpanStatus.ERROR if otel_status == "ERROR" else SpanStatus.OK
+                    
                     span = SpanWithStatus(
                         span_id=row["span_id"],
                         trace_id=row["trace_id"],
@@ -351,7 +357,7 @@ async def get_span_tree(
                         start_time=datetime.fromisoformat(row["start_time"]),
                         end_time=datetime.fromisoformat(row["end_time"]),
                         attributes=json.loads(row["filtered_attributes"]),
-                        status=row["status"].lower(),
+                        status=status,
                     )
 
                     spans_by_id[span.span_id] = span