autogen/samples/tools/testbed/scenarios/default_two_agents.py

35 lines
927 B
Python
Raw Normal View History

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", "").rstrip().find("TERMINATE") >= 0,
llm_config=testbed_utils.default_llm_config(config_list, timeout=180),
)
user_proxy = autogen.UserProxyAgent(
"user_proxy",
human_input_mode="NEVER",
is_termination_msg=lambda x: x.get("content", "").rstrip().find("TERMINATE") >= 0,
code_execution_config={
"work_dir": "coding",
"use_docker": False,
},
max_consecutive_auto_reply=10,
default_auto_reply="TERMINATE",
)
user_proxy.initiate_chat(assistant, message="__PROMPT__")
##############################
testbed_utils.finalize(agents=[assistant, user_proxy])