mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-13 03:01:30 +00:00
65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
|
import os
|
||
|
|
import json
|
||
|
|
import autogen
|
||
|
|
import testbed_utils
|
||
|
|
|
||
|
|
testbed_utils.init()
|
||
|
|
##############################
|
||
|
|
|
||
|
|
config_list = autogen.config_list_from_json(
|
||
|
|
"OAI_CONFIG_LIST",
|
||
|
|
filter_dict={"model": ["__MODEL__"]},
|
||
|
|
)
|
||
|
|
|
||
|
|
assistant = autogen.AssistantAgent(
|
||
|
|
"assistant",
|
||
|
|
is_termination_msg=lambda x: x.get("content", "").find("TERMINATE") >= 0,
|
||
|
|
llm_config=testbed_utils.default_llm_config(config_list, timeout=180),
|
||
|
|
)
|
||
|
|
|
||
|
|
user_proxy = autogen.UserProxyAgent(
|
||
|
|
"user_proxy",
|
||
|
|
human_input_mode="NEVER",
|
||
|
|
system_message="A human who can run code at a terminal and report back the results.",
|
||
|
|
is_termination_msg=lambda x: x.get("content", "").find("TERMINATE") >= 0,
|
||
|
|
code_execution_config={
|
||
|
|
"work_dir": "coding",
|
||
|
|
"use_docker": False,
|
||
|
|
},
|
||
|
|
max_consecutive_auto_reply=10,
|
||
|
|
default_auto_reply="",
|
||
|
|
)
|
||
|
|
|
||
|
|
third_agent = autogen.AssistantAgent(
|
||
|
|
"__3RD_AGENT_NAME__",
|
||
|
|
system_message="""
|
||
|
|
__3RD_AGENT_PROMPT__
|
||
|
|
""".strip(),
|
||
|
|
is_termination_msg=lambda x: x.get("content", "").find("TERMINATE") >= 0,
|
||
|
|
llm_config=testbed_utils.default_llm_config(config_list, timeout=180),
|
||
|
|
)
|
||
|
|
|
||
|
|
groupchat = autogen.GroupChat(
|
||
|
|
agents=[user_proxy, assistant, third_agent],
|
||
|
|
messages=[],
|
||
|
|
speaker_selection_method="__SELECTION_METHOD__",
|
||
|
|
allow_repeat_speaker=False,
|
||
|
|
max_round=12,
|
||
|
|
)
|
||
|
|
|
||
|
|
manager = autogen.GroupChatManager(
|
||
|
|
groupchat=groupchat,
|
||
|
|
is_termination_msg=lambda x: x.get("content", "").find("TERMINATE") >= 0,
|
||
|
|
llm_config=testbed_utils.default_llm_config(config_list, timeout=180),
|
||
|
|
)
|
||
|
|
|
||
|
|
user_proxy.initiate_chat(
|
||
|
|
manager,
|
||
|
|
message="""
|
||
|
|
__PROMPT__
|
||
|
|
""".strip(),
|
||
|
|
)
|
||
|
|
|
||
|
|
##############################
|
||
|
|
testbed_utils.finalize(agents=[assistant, user_proxy, third_agent, manager])
|