fix tests

2026-07-11 22:47:12 +03:00
parent 7cf8d9bcf3
commit f615b6befb
3 changed files with 30 additions and 8 deletions
--- a/tests/test_gossip_and_relay.py
+++ b/tests/test_gossip_and_relay.py
@@ -906,7 +906,7 @@ def test_stream_line_tokens_accounting():
    )
    assert observed == 2 and reported is None
    observed, reported = _stream_line_tokens(b'data: {"usage": {"total_tokens": 42}}')
-    assert reported == 42
+    assert reported == {"prompt": None, "completion": None, "total": 42}
    assert _stream_line_tokens(b"data: not-json") == (1, None)