mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-13 11:11:25 +00:00
42b27b9a9d
* Add isort * Apply isort on py files * Fix circular import * Fix format for notebooks * Fix format --------- Co-authored-by: Chi Wang <wang.chi@microsoft.com>
39 lines
896 B
Python
39 lines
896 B
Python
import json
|
|
import os
|
|
|
|
import testbed_utils
|
|
|
|
import autogen
|
|
|
|
testbed_utils.init()
|
|
##############################
|
|
|
|
config_list = autogen.config_list_from_json("OAI_CONFIG_LIST")
|
|
|
|
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="",
|
|
)
|
|
user_proxy.initiate_chat(
|
|
assistant,
|
|
message="""
|
|
__PROMPT__
|
|
""".strip(),
|
|
)
|
|
|
|
|
|
##############################
|
|
testbed_utils.finalize(agents=[assistant, user_proxy])
|