Skip to content

Commit 842d0e9

Browse files
ajac-zeroDouweM
andauthored
Update tests/test_streaming.py
Co-authored-by: Douwe Maan <[email protected]>
1 parent 4f0d2ad commit 842d0e9

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/test_streaming.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -393,12 +393,12 @@ def test_streamed_text_stream_sync():
393393

394394
agent = Agent(m)
395395

396-
with agent.run_stream_sync('Hello') as result:
397-
# typehint to test (via static typing) that the stream type is correctly inferred
398-
chunks: list[str] = [c for c in result.stream_text_sync()]
399-
# one chunk with `stream_text()` due to group_by_temporal
400-
assert chunks == snapshot(['The cat sat on the mat.'])
401-
assert result.is_complete
396+
result = agent.run_stream_sync('Hello')
397+
# typehint to test (via static typing) that the stream type is correctly inferred
398+
chunks: list[str] = [c for c in result.stream_text_sync()]
399+
# one chunk with `stream_text()` due to group_by_temporal
400+
assert chunks == snapshot(['The cat sat on the mat.'])
401+
assert result.is_complete
402402

403403
with agent.run_stream_sync('Hello') as result:
404404
# typehint to test (via static typing) that the stream type is correctly inferred

0 commit comments

Comments
 (0)