review

Author: alexander-alderman-webb

sentry_sdk/integrations/anthropic.py

@@ -835,19 +835,17 @@ def __next__(self: "Stream") -> "RawMessageStreamEvent":
         except StopIteration:
             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)
@@ -870,6 +868,7 @@ def close(self: "Stream") -> None:
 
         if not hasattr(self, "_model"):
             self._span.__exit__(None, None, None)
+            del self._span
             return f(self)
 
         _finish_streaming_span(
@@ -1041,19 +1040,17 @@ def __next__(self: "MessageStream") -> "MessageStreamEvent":
         except StopIteration:
             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)
@@ -1076,6 +1073,7 @@ def close(self: "MessageStream") -> None:
 
         if not hasattr(self, "_model"):
             self._span.__exit__(None, None, None)
+            del self._span
             return f(self)
 
         _finish_streaming_span(