Oct 29, 2025 · Oct 30, 2025 · Oct 30, 2025 · Oct 29, 2025
diff --git a/src/replicate/_streaming.py b/src/replicate/_streaming.py
Original file line number	Diff line number	Diff line change
Expand Up		@@ -57,9 +57,8 @@ def __stream__(self) -> Iterator[_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
Expand Down Expand Up		@@ -121,9 +120,8 @@ async def __stream__(self) -> AsyncIterator[_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
Expand Down