From 9b07b4a7c898d73e4b3478368602fd1a7c0110f0 Mon Sep 17 00:00:00 2001 From: 9qeklajc Date: Tue, 29 Sep 2026 20:33:08 +0200 Subject: [PATCH] fix: normalize streamed test chunks for mypy --- tests/unit/test_upstream_stream_timeout.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/unit/test_upstream_stream_timeout.py b/tests/unit/test_upstream_stream_timeout.py index 7c42df41..867677b8 100644 --- a/tests/unit/test_upstream_stream_timeout.py +++ b/tests/unit/test_upstream_stream_timeout.py @@ -213,7 +213,12 @@ async def test_responses_idle_timeout_does_not_emit_completed( result = await provider.handle_streaming_responses_completion( response, key, 100, reservation_snapshot=MagicMock() ) - emitted = b"".join([chunk async for chunk in result.body_iterator]) + emitted = b"".join( + [ + chunk.encode() if isinstance(chunk, str) else bytes(chunk) + async for chunk in result.body_iterator + ] + ) assert b'"type": "response.failed"' in emitted assert b'"code": "UPSTREAM_TIMEOUT"' in emitted