diff --git a/examples/streaming-v2/streaming_comparison.py b/examples/streaming-v2/streaming_comparison.py index cc7016fe9..1a1f635d0 100644 --- a/examples/streaming-v2/streaming_comparison.py +++ b/examples/streaming-v2/streaming_comparison.py @@ -446,7 +446,7 @@ async def run_v2_async_with_transformer(question: str) -> None: _ = await run.output - snapshots = list(run.extensions["token_metrics"]) + snapshots = [s async for s in run.extensions["token_metrics"]] final = next((s for s in snapshots if s.is_final), None) periodic = [s for s in snapshots if not s.is_final]