From 9e89b685963af64643415232a3ed58348688be38 Mon Sep 17 00:00:00 2001 From: Sydney Runkle Date: Tue, 22 Apr 2025 17:24:54 -0700 Subject: [PATCH] fixing prebuilt tests --- libs/prebuilt/tests/test_react_agent.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/libs/prebuilt/tests/test_react_agent.py b/libs/prebuilt/tests/test_react_agent.py index 4b469d077..eea4444e9 100644 --- a/libs/prebuilt/tests/test_react_agent.py +++ b/libs/prebuilt/tests/test_react_agent.py @@ -672,7 +672,8 @@ def test_react_agent_parallel_tool_calls( for event in agent.stream( {"messages": [("user", query)]}, config, stream_mode="values" ): - message_types.append([message.type for message in event["messages"]]) + if messages := event.get("messages"): + message_types.append([m.type for m in messages]) if version == "v1": assert message_types == [ @@ -691,7 +692,8 @@ def test_react_agent_parallel_tool_calls( for event in agent.stream( Command(resume={"data": "Hello"}), config, stream_mode="values" ): - message_types.append([message.type for message in event["messages"]]) + if messages := event.get("messages"): + message_types.append([m.type for m in messages]) assert message_types == [ ["human", "ai"],