diff --git a/py/src/braintrust/wrappers/claude_agent_sdk/_wrapper.py b/py/src/braintrust/wrappers/claude_agent_sdk/_wrapper.py index 7ad1f1b48..c4db45ea9 100644 --- a/py/src/braintrust/wrappers/claude_agent_sdk/_wrapper.py +++ b/py/src/braintrust/wrappers/claude_agent_sdk/_wrapper.py @@ -255,6 +255,7 @@ async def receive_response(self) -> AsyncGenerator[Any, None]: span.log(output=final_results[-1] if final_results else None) except Exception as e: log.warning("Error in tracing code", exc_info=e) + raise finally: llm_tracker.cleanup() if hasattr(_thread_local, 'parent_span_export'):