File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed
Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments