We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a4c5b9b commit 2eb07f3Copy full SHA for 2eb07f3
1 file changed
tests/contrib/langgraph/test_streaming.py
@@ -150,14 +150,13 @@ async def test_workflow_publishes_astream_chunks(client: Client):
150
151
ws_client = WorkflowStreamClient.create(client, handle.id)
152
chunks: list[dict[str, Any]] = []
153
- async with asyncio.timeout(15.0):
154
- async for item in ws_client.topic("astream", type=dict).subscribe(
155
- from_offset=0,
156
- poll_cooldown=timedelta(0),
157
- ):
158
- chunks.append(item.data)
159
- if chunks[-1].get("done"):
160
- break
+ async for item in ws_client.topic("astream", type=dict).subscribe(
+ from_offset=0,
+ poll_cooldown=timedelta(0),
+ ):
+ chunks.append(item.data)
+ if chunks[-1].get("done"):
+ break
161
162
await handle.result()
163
0 commit comments