docs

Author: alexander-alderman-webb

sentry_sdk/integrations/anthropic.py

@@ -1008,7 +1008,8 @@ def _wrap_message_stream_iter(
 ) -> "Callable[..., Iterator[MessageStreamEvent]]":
     """
     Accumulates output data while iterating. When the returned iterator ends, set
-    output attributes on the AI Client Span and end the span.
+    output attributes on the AI Client Span and ends the span (unless the `close()`
+    or `__next__()` patches have already closed it).
     """
 
     def __iter__(self: "MessageStream") -> "Iterator[MessageStreamEvent]":
@@ -1030,6 +1031,7 @@ def _wrap_message_stream_next(
 ) -> "Callable[..., MessageStreamEvent]":
     """
     Accumulates output data from the returned event.
+    Closes the AI Client Span if `StopIteration` is raised.
     """
 
     def __next__(self: "MessageStream") -> "MessageStreamEvent":
@@ -1064,7 +1066,8 @@ def _wrap_message_stream_close(
     f: "Callable[..., None]",
 ) -> "Callable[..., None]":
     """
-    Closes the AI Client Span, unless the finally block in `_wrap_synchronous_message_iterator()` runs first.
+    Closes the AI Client Span, unless the finally block in `_wrap_synchronous_message_iterator()` or
+    the except block in the `__next__()` patch runs first.
     """
 
     def close(self: "MessageStream") -> None: