From b1a4ae91d2f95942660df8d53900e5ce4eeee77f Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 6 Jan 2026 19:54:00 +0000 Subject: [PATCH] Updated streaming tests --- tests/integration/test_rag_streaming_protocol.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/integration/test_rag_streaming_protocol.py b/tests/integration/test_rag_streaming_protocol.py index 8f9feb09..d2ceea95 100644 --- a/tests/integration/test_rag_streaming_protocol.py +++ b/tests/integration/test_rag_streaming_protocol.py @@ -347,4 +347,5 @@ class TestStreamingProtocolEdgeCases: # Assert assert len(chunks_with_flags) == 4 assert chunks_with_flags[-1] == ("", True) # Final empty chunk - assert end_of_stream_flags.count(True) for end_of_stream_flags in [[f for c, f in chunks_with_flags]] == 1 + end_of_stream_flags = [f for c, f in chunks_with_flags] + assert end_of_stream_flags.count(True) == 1