apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 20:21:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Updated streaming tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-01-06 19:54:00 +00:00
parent fed30621a2
commit b1a4ae91d2
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/integration/test_rag_streaming_protocol.py
View file

@ -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