Skip to content

Commit cf8c3a5

Browse files
committed
test: fix test_react_agent test by ignoring stream events with interrupt
1 parent e596d05 commit cf8c3a5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

libs/prebuilt/tests/test_react_agent.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -638,8 +638,9 @@ def get_weather(location: str) -> str:
638638
for event in agent.stream(
639639
{"messages": [("user", query)]}, config, stream_mode="values"
640640
):
641-
if messages := event.get("messages"):
642-
message_types.append([m.type for m in messages])
641+
if "__interrupt__" not in event:
642+
if messages := event.get("messages"):
643+
message_types.append([m.type for m in messages])
643644

644645
if version == "v1":
645646
assert message_types == [

0 commit comments

Comments
 (0)