wip: composite_tracer

src/intelligence_layer/core/tracer/composite_tracer.py

@@ -2,6 +2,7 @@
 from typing import Generic, Optional, Sequence, TypeVar
 
 from intelligence_layer.core.tracer.tracer import (
+    ExportedSpan,
     PydanticSerializable,
     Span,
     TaskSpan,
@@ -41,30 +42,31 @@ def span(
         self,
         name: str,
         timestamp: Optional[datetime] = None,
-        trace_id: Optional[str] = None,
     ) -> "CompositeSpan[Span]":
         timestamp = timestamp or utc_now()
-        trace_id = self.ensure_id(trace_id)
         return CompositeSpan(
-            [tracer.span(name, timestamp, trace_id) for tracer in self.tracers]
+            [tracer.span(name, timestamp) for tracer in self.tracers]
         )
 
     def task_span(
         self,
         task_name: str,
         input: PydanticSerializable,
         timestamp: Optional[datetime] = None,
-        trace_id: Optional[str] = None,
     ) -> "CompositeTaskSpan":
         timestamp = timestamp or utc_now()
-        trace_id = self.ensure_id(trace_id)
         return CompositeTaskSpan(
             [
-                tracer.task_span(task_name, input, timestamp, trace_id)
+                tracer.task_span(task_name, input, timestamp)
                 for tracer in self.tracers
             ]
         )
 
+    def export_for_viewing(self) -> Sequence[ExportedSpan]:
+        if len(self.tracers) > 0:
+            return self.tracers[0].export_for_viewing()
+        return []
+
 
 class CompositeSpan(Generic[SpanVar], CompositeTracer[SpanVar], Span):
     """A :class:`Span` that allows for recording to multiple spans simultaneously.

tests/core/tracer/test_composite_tracer.py

@@ -8,11 +8,14 @@
 
 
 def test_composite_tracer(test_task: Task[str, str]) -> None:
-    tracer1 = InMemoryTracer()
-    tracer2 = InMemoryTracer()
-    test_task.run(input=input, tracer=CompositeTracer([tracer1, tracer2]))
+    tracer_1 = InMemoryTracer()
+    tracer_2 = InMemoryTracer()
+    test_task.run(input="input", tracer=CompositeTracer([tracer_1, tracer_2]))
 
-    assert tracer1 == tracer2
+    trace_1 = tracer_1.export_for_viewing()[0]
+    trace_2 = tracer_2.export_for_viewing()[0]
+    assert trace_1.name == trace_2.name
+    assert trace_1.attributes = trace_2.attributes
 
 
 def test_composite_tracer_id_consistent_across_children(