From 2bafdb2c65edfe9913448b3eead28a0a82141ce4 Mon Sep 17 00:00:00 2001 From: Eric Zhu Date: Thu, 29 Feb 2024 23:20:26 -0800 Subject: [PATCH] update notebook and test --- notebook/agentchat_nestedchat.ipynb | 1 + test/agentchat/test_nested.py | 1 + 2 files changed, 2 insertions(+) diff --git a/notebook/agentchat_nestedchat.ipynb b/notebook/agentchat_nestedchat.ipynb index c4a25871e25..f9a7f78efd8 100644 --- a/notebook/agentchat_nestedchat.ipynb +++ b/notebook/agentchat_nestedchat.ipynb @@ -800,6 +800,7 @@ "]\n", "assistant_1.register_nested_chats(\n", " nested_chat_queue,\n", + " trigger=user,\n", ")\n", "# user.initiate_chat(assistant, message=tasks[0], max_turns=1)\n", "\n", diff --git a/test/agentchat/test_nested.py b/test/agentchat/test_nested.py index 7addbe2c12a..a04fd009348 100755 --- a/test/agentchat/test_nested.py +++ b/test/agentchat/test_nested.py @@ -112,6 +112,7 @@ def writing_message(recipient, messages, sender, config): ] assistant.register_nested_chats( nested_chat_queue, + trigger=user, ) user.initiate_chats([{"recipient": assistant, "message": tasks[0]}, {"recipient": assistant, "message": tasks[1]}])