mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-25 18:01:03 +00:00
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
import autogen
|
|
|
|
|
|
def test_chat_manager():
|
|
agent1 = autogen.ConversableAgent(
|
|
"alice",
|
|
max_consecutive_auto_reply=2,
|
|
human_input_mode="NEVER",
|
|
llm_config=False,
|
|
default_auto_reply="This is alice sepaking.",
|
|
)
|
|
agent2 = autogen.ConversableAgent(
|
|
"bob",
|
|
max_consecutive_auto_reply=2,
|
|
human_input_mode="NEVER",
|
|
llm_config=False,
|
|
default_auto_reply="This is bob speaking.",
|
|
)
|
|
groupchat = autogen.GroupChat(agents=[agent1, agent2], messages=[], max_round=2)
|
|
group_chat_manager = autogen.GroupChatManager(groupchat=groupchat, llm_config=False)
|
|
agent1.initiate_chat(group_chat_manager, message="hello")
|
|
|
|
assert len(agent1.chat_messages[group_chat_manager]) == 2
|
|
assert len(groupchat.messages) == 2
|
|
|
|
group_chat_manager.reset()
|
|
assert len(groupchat.messages) == 0
|
|
agent1.reset()
|
|
agent2.reset()
|
|
agent2.initiate_chat(group_chat_manager, message="hello")
|
|
assert len(groupchat.messages) == 2
|
|
|
|
|
|
def test_plugin():
|
|
# Give another Agent class ability to manage group chat
|
|
agent1 = autogen.ConversableAgent(
|
|
"alice",
|
|
max_consecutive_auto_reply=2,
|
|
human_input_mode="NEVER",
|
|
llm_config=False,
|
|
default_auto_reply="This is alice sepaking.",
|
|
)
|
|
agent2 = autogen.ConversableAgent(
|
|
"bob",
|
|
max_consecutive_auto_reply=2,
|
|
human_input_mode="NEVER",
|
|
llm_config=False,
|
|
default_auto_reply="This is bob speaking.",
|
|
)
|
|
groupchat = autogen.GroupChat(agents=[agent1, agent2], messages=[], max_round=2)
|
|
group_chat_manager = autogen.ConversableAgent(name="deputy_manager", llm_config=False)
|
|
group_chat_manager.register_reply(
|
|
autogen.Agent,
|
|
reply_func=autogen.GroupChatManager.run_chat,
|
|
config=groupchat,
|
|
reset_config=autogen.GroupChat.reset,
|
|
)
|
|
agent1.initiate_chat(group_chat_manager, message="hello")
|
|
|
|
assert len(agent1.chat_messages[group_chat_manager]) == 2
|
|
assert len(groupchat.messages) == 2
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# test_broadcast()
|
|
# test_chat_manager()
|
|
test_plugin()
|