Fix e2e test for adaptive first batch size

test_large_batch_streaming expected 4 batches (200/50) but the
adaptive first batch splits the first batch into 5+45, producing
5 batches total.
This commit is contained in:
Viktor Barzin 2026-02-22 13:48:36 +00:00
parent 8ef6868881
commit c6f7b47446
No known key found for this signature in database
GPG key ID: 0EB088298288D958

View file

@ -80,7 +80,7 @@ async def test_large_batch_streaming(
batches = [l for l in lines if l["type"] == "batch"]
complete = lines[-1]
assert len(batches) == 4
assert len(batches) == 5 # first batch is smaller (FIRST_BATCH_SIZE=5), then 4x50
assert complete["total"] == 200