review

Author: alexander-alderman-webb

sentry_sdk/integrations/anthropic.py

@@ -963,19 +963,17 @@ async def __anext__(self: "AsyncStream") -> "RawMessageStreamEvent":
         except StopAsyncIteration:
             exc_info = sys.exc_info()
             with capture_internal_exceptions():
-                if not hasattr(self, "_span"):
-                    raise
-
-                _finish_streaming_span(
-                    self._span,
-                    self._integration,
-                    self._model,
-                    self._usage,
-                    self._content_blocks,
-                    self._response_id,
-                    self._finish_reason,
-                )
-                del self._span
+                if hasattr(self, "_span"):
+                    _finish_streaming_span(
+                        self._span,
+                        self._integration,
+                        self._model,
+                        self._usage,
+                        self._content_blocks,
+                        self._response_id,
+                        self._finish_reason,
+                    )
+                    del self._span
             reraise(*exc_info)
 
         _accumulate_event_data(self, event)
@@ -998,6 +996,7 @@ async def close(self: "AsyncStream") -> None:
 
         if not hasattr(self, "_model"):
             self._span.__exit__(None, None, None)
+            del self._span
             return await f(self)
 
         _finish_streaming_span(
@@ -1306,19 +1305,17 @@ async def __anext__(self: "AsyncMessageStream") -> "MessageStreamEvent":
         except StopAsyncIteration:
             exc_info = sys.exc_info()
             with capture_internal_exceptions():
-                if not hasattr(self, "_span"):
-                    raise
-
-                _finish_streaming_span(
-                    self._span,
-                    self._integration,
-                    self._model,
-                    self._usage,
-                    self._content_blocks,
-                    self._response_id,
-                    self._finish_reason,
-                )
-                del self._span
+                if hasattr(self, "_span"):
+                    _finish_streaming_span(
+                        self._span,
+                        self._integration,
+                        self._model,
+                        self._usage,
+                        self._content_blocks,
+                        self._response_id,
+                        self._finish_reason,
+                    )
+                    del self._span
             reraise(*exc_info)
 
         _accumulate_event_data(self, event)
@@ -1341,6 +1338,7 @@ async def close(self: "AsyncMessageStream") -> None:
 
         if not hasattr(self, "_model"):
             self._span.__exit__(None, None, None)
+            del self._span
             return await f(self)
 
         _finish_streaming_span(