diff --git a/tests/integration/observability/test_passthrough_upstream_error_visibility.py b/tests/integration/observability/test_passthrough_upstream_error_visibility.py index 6872e81da82..02c93837d90 100644 --- a/tests/integration/observability/test_passthrough_upstream_error_visibility.py +++ b/tests/integration/observability/test_passthrough_upstream_error_visibility.py @@ -763,9 +763,7 @@ async def test_gemini_passthrough_async_streaming_429_first_frame_reaches_client first: Final = await iterator.__anext__() assert first.startswith(b'data: {"error":"rate limited"}'), first gate.set() - rest: Final = b"" - async for chunk in iterator: - rest += chunk + rest: Final = b"".join([chunk async for chunk in iterator]) assert first + rest == b"".join(frames), first + rest finally: gate.set()