Skip to content

Commit 631e727

Browse files
Merge branch 'webb/anthropic/sync-iterators' into webb/anthropic/async-iterators
2 parents a21406f + a3cc18f commit 631e727

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

sentry_sdk/integrations/anthropic.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def setup_once() -> None:
144144

145145
# Before https://github.com/anthropics/anthropic-sdk-python/commit/b1a1c0354a9aca450a7d512fdbdeb59c0ead688a
146146
# MessageStream inherits from Stream, so patching Stream is sufficient on these versions.
147-
if version is not None and version >= (0, 26, 2):
147+
if not issubclass(MessageStream, Stream):
148148
MessageStream.__iter__ = _wrap_message_stream_iter(MessageStream.__iter__)
149149
MessageStream.__next__ = _wrap_message_stream_next(MessageStream.__next__)
150150
MessageStream.close = _wrap_message_stream_close(MessageStream.close)
@@ -783,8 +783,7 @@ def _wrap_stream_iter(
783783

784784
def __iter__(self: "Stream") -> "Iterator[RawMessageStreamEvent]":
785785
if not hasattr(self, "_span"):
786-
for event in f(self):
787-
yield event
786+
yield from f(self)
788787
return
789788

790789
_initialize_data_accumulation_state(self)
@@ -1073,8 +1072,7 @@ def _wrap_message_stream_iter(
10731072

10741073
def __iter__(self: "MessageStream") -> "Iterator[MessageStreamEvent]":
10751074
if not hasattr(self, "_span"):
1076-
for event in f(self):
1077-
yield event
1075+
yield from f(self)
10781076
return
10791077

10801078
_initialize_data_accumulation_state(self)

0 commit comments

Comments
 (0)