File tree 1 file changed +12
-2
lines changed
1 file changed +12
-2
lines changed Original file line number Diff line number Diff line change @@ -45,10 +45,15 @@ def __stream__(self) -> Iterator[ResponseT]:
45
45
cast_to = self ._cast_to
46
46
response = self .response
47
47
process_data = self ._client ._process_response_data
48
+ iterator = self ._iter_events ()
48
49
49
- for sse in self . _iter_events () :
50
+ for sse in iterator :
50
51
yield process_data (data = sse .json (), cast_to = cast_to , response = response )
51
52
53
+ # Ensure the entire stream is consumed
54
+ for sse in iterator :
55
+ ...
56
+
52
57
53
58
class AsyncStream (Generic [ResponseT ]):
54
59
"""Provides the core interface to iterate over an asynchronous stream response."""
@@ -83,10 +88,15 @@ async def __stream__(self) -> AsyncIterator[ResponseT]:
83
88
cast_to = self ._cast_to
84
89
response = self .response
85
90
process_data = self ._client ._process_response_data
91
+ iterator = self ._iter_events ()
86
92
87
- async for sse in self . _iter_events () :
93
+ async for sse in iterator :
88
94
yield process_data (data = sse .json (), cast_to = cast_to , response = response )
89
95
96
+ # Ensure the entire stream is consumed
97
+ async for sse in iterator :
98
+ ...
99
+
90
100
91
101
class ServerSentEvent :
92
102
def __init__ (
You can’t perform that action at this time.
0 commit comments