diff --git a/src/opencode_ai/_streaming.py b/src/opencode_ai/_streaming.py index 34499b5..5dc9d33 100644 --- a/src/opencode_ai/_streaming.py +++ b/src/opencode_ai/_streaming.py @@ -57,9 +57,8 @@ class Stream(Generic[_T]): for sse in iterator: yield process_data(data=sse.json(), cast_to=cast_to, response=response) - # Ensure the entire stream is consumed - for _sse in iterator: - ... + # As we might not fully consume the response stream, we need to close it explicitly + response.close() def __enter__(self) -> Self: return self @@ -121,9 +120,8 @@ class AsyncStream(Generic[_T]): async for sse in iterator: yield process_data(data=sse.json(), cast_to=cast_to, response=response) - # Ensure the entire stream is consumed - async for _sse in iterator: - ... + # As we might not fully consume the response stream, we need to close it explicitly + await response.aclose() async def __aenter__(self) -> Self: return self