fix(client): close streams without requiring full consumption

Author: stainless-app[bot]

src/gradient/_streaming.py

@@ -76,9 +76,8 @@ def __stream__(self) -> Iterator[_T]:
 
             yield process_data(data=data, 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
@@ -158,9 +157,8 @@ async def __stream__(self) -> AsyncIterator[_T]:
 
             yield process_data(data=data, 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