Skip to content

Commit 5b5177e

Browse files
committed
fixing the yield_ thigns
1 parent a2496e4 commit 5b5177e

File tree

1 file changed

+3
-1
lines changed
  • examples/tutorials/10_async/00_base/020_streaming/tests

1 file changed

+3
-1
lines changed

examples/tutorials/10_async/00_base/020_streaming/tests/test_agent.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,9 @@ async def test_send_event_and_poll(self, client: AsyncAgentex, agent_id: str):
9696
user_message=user_message,
9797
timeout=30,
9898
sleep_interval=1.0,
99+
yield_updates=False,
99100
):
101+
100102
messages.append(message)
101103

102104
assert len(messages) > 0
@@ -199,7 +201,7 @@ async def stream_messages() -> None:
199201
await asyncio.sleep(1) # wait for state to be updated
200202
states = await client.states.list(agent_id=agent_id, task_id=task.id)
201203
assert len(states) == 1
202-
state = states[0].state
204+
state: dict[str, object] = states[0].state
203205
messages = state.get("messages", [])
204206

205207
assert isinstance(messages, list)

0 commit comments

Comments
 (0)