Skip to content

Commit 2eb07f3

Browse files
committed
remove timeout
1 parent a4c5b9b commit 2eb07f3

1 file changed

Lines changed: 7 additions & 8 deletions

File tree

tests/contrib/langgraph/test_streaming.py

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -150,14 +150,13 @@ async def test_workflow_publishes_astream_chunks(client: Client):
150150

151151
ws_client = WorkflowStreamClient.create(client, handle.id)
152152
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
153+
async for item in ws_client.topic("astream", type=dict).subscribe(
154+
from_offset=0,
155+
poll_cooldown=timedelta(0),
156+
):
157+
chunks.append(item.data)
158+
if chunks[-1].get("done"):
159+
break
161160

162161
await handle.result()
163162

0 commit comments

Comments
 (0)