mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-27 02:41:12 +00:00
46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
|
import os
|
||
|
|
|
||
|
|
from autogen import UserProxyAgent, config_list_from_json
|
||
|
|
from autogen.agentchat.contrib.web_surfer import WebSurferAgent
|
||
|
|
from autogen.browser_utils import (
|
||
|
|
BingMarkdownSearch,
|
||
|
|
PlaywrightMarkdownBrowser,
|
||
|
|
RequestsMarkdownBrowser,
|
||
|
|
SeleniumMarkdownBrowser,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
# Load LLM inference endpoints from an env variable or a file
|
||
|
|
# See https://microsoft.github.io/autogen/docs/FAQ#set-your-api-endpoints
|
||
|
|
# and OAI_CONFIG_LIST_sample.
|
||
|
|
# For example, if you have created a OAI_CONFIG_LIST file in the current working directory, that file will be used.
|
||
|
|
config_list = config_list_from_json(env_or_file="OAI_CONFIG_LIST")
|
||
|
|
|
||
|
|
browser = RequestsMarkdownBrowser(
|
||
|
|
# PlaywrightMarkdownBrowser(
|
||
|
|
viewport_size=1024 * 3,
|
||
|
|
downloads_folder=os.getcwd(),
|
||
|
|
search_engine=BingMarkdownSearch(bing_api_key=os.environ["BING_API_KEY"]),
|
||
|
|
# launch_args={"channel": "msedge", "headless": False},
|
||
|
|
)
|
||
|
|
|
||
|
|
web_surfer = WebSurferAgent(
|
||
|
|
"web_surfer",
|
||
|
|
llm_config={"config_list": config_list},
|
||
|
|
summarizer_llm_config={"config_list": config_list},
|
||
|
|
is_termination_msg=lambda x: x.get("content", "").rstrip().find("TERMINATE") >= 0,
|
||
|
|
code_execution_config=False,
|
||
|
|
browser=browser,
|
||
|
|
)
|
||
|
|
|
||
|
|
# Create the agent that represents the user in the conversation.
|
||
|
|
user_proxy = UserProxyAgent("user", code_execution_config=False)
|
||
|
|
|
||
|
|
# Let the assistant start the conversation. It will end when the user types exit.
|
||
|
|
web_surfer.initiate_chat(user_proxy, message="How can I help you today?")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|