IL-306 Exception record output

src/intelligence_layer/core/tracer/tracer.py

@@ -228,6 +228,21 @@ def record_output(self, output: PydanticSerializable) -> None:
         """
         ...
 
+    def __exit__(
+        self,
+        exc_type: Optional[type[BaseException]],
+        exc_value: Optional[BaseException],
+        _traceback: Optional[TracebackType],
+    ) -> None:
+        if exc_type is not None and exc_value is not None and _traceback is not None:
+            error_value = ErrorValue(
+                error_type=str(exc_type.__qualname__),
+                message=str(exc_value),
+                stack_trace=str(traceback.format_exc()),
+            )
+            self.record_output(error_value)
+        self.end()
+
 
 TracerVar = TypeVar("TracerVar", bound=Tracer)
 

tests/core/test_tracer.py

@@ -118,10 +118,10 @@ def test_task_logs_error_value() -> None:
     tracer = InMemoryTracer()
 
     with pytest.raises(ValueError):
-        with tracer.task_span("failing task", None):
+        with tracer.span("failing task"):
             raise ValueError("my bad, sorry")
 
-    assert isinstance(tracer.entries[0], InMemoryTaskSpan)
+    assert isinstance(tracer.entries[0], InMemorySpan)
     assert isinstance(tracer.entries[0].entries[0], LogEntry)
     error = tracer.entries[0].entries[0].value
     assert isinstance(error, ErrorValue)
@@ -130,6 +130,21 @@ def test_task_logs_error_value() -> None:
     assert error.stack_trace.startswith("Traceback")
 
 
+def test_task_span_records_error_value() -> None:
+    tracer = InMemoryTracer()
+
+    with pytest.raises(ValueError):
+        with tracer.task_span("failing task", None):
+            raise ValueError("my bad, sorry")
+
+    assert isinstance(tracer.entries[0], InMemoryTaskSpan)
+    error = tracer.entries[0].output
+    assert isinstance(error, ErrorValue)
+    assert error.message == "my bad, sorry"
+    assert error.error_type == "ValueError"
+    assert error.stack_trace.startswith("Traceback")
+
+
 def test_task_automatically_logs_input_and_output(
     complete: Task[CompleteInput, CompleteOutput],
 ) -> None: