vrr/free-claude-code
2
0
Fork 0
mirror of https://github.com/Alishahryar1/free-claude-code.git synced 2026-05-01 21:00:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Fixing sse stream

Browse source
This commit is contained in:
Shantanu Suryawanshi 2026-02-18 21:31:28 -05:00
parent e7ac85264f
commit 24a5e4d968
4 changed files with 9 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/providers/test_streaming_errors.py
View file

@ -110,12 +110,11 @@ class TestStreamingExceptionHandling:
):
events = await _collect_stream(provider, request)
# Should have message_start, error text block, close blocks, message_delta, message_stop, done
# Should have message_start, error text block, close blocks, message_delta, message_stop
event_text = "".join(events)
assert "message_start" in event_text
assert "API failed" in event_text
assert "message_stop" in event_text
assert "[DONE]" in event_text
@pytest.mark.asyncio
async def test_error_after_partial_content(self):
@ -434,7 +433,6 @@ class TestStreamChunkEdgeCases:
event_text = "".join(events)
assert "message_start" in event_text
assert "message_stop" in event_text
assert "[DONE]" in event_text
@pytest.mark.asyncio
async def test_stream_chunk_with_none_delta_handled(self):
@ -471,11 +469,10 @@ class TestStreamChunkEdgeCases:
event_text = "".join(events)
assert "message_start" in event_text
assert "message_stop" in event_text
assert "[DONE]" in event_text
@pytest.mark.asyncio
async def test_stream_generator_cleanup_on_exception(self):
"""When stream raises mid-iteration, message_stop and [DONE] still emitted."""
"""When stream raises mid-iteration, message_stop still emitted."""
provider = _make_provider()
request = _make_request()
@ -504,7 +501,6 @@ class TestStreamChunkEdgeCases:
assert "Partial" in event_text
assert "Connection reset" in event_text
assert "message_stop" in event_text
assert "[DONE]" in event_text
def test_stream_malformed_tool_args_chunked(self):
"""Chunked tool args that never form valid JSON are flushed with {}."""