File size: 1,443 Bytes
6e38ce1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | import asyncio
import argparse
from autogen_agentchat.ui import Console
from autogen_ext.models.openai import OpenAIChatCompletionClient
from autogen_agentchat.conditions import TextMentionTermination
from magentic_ui.teams import RoundRobinGroupChat
from magentic_ui.agents import FileSurfer
from autogen_agentchat.agents import UserProxyAgent
# Configure logging to print to console
async def main(work_dir: str) -> None:
model_client = OpenAIChatCompletionClient(model="gpt-4o")
termination = TextMentionTermination("EXITT")
user_proxy = UserProxyAgent(name="user_proxy")
file_surfer = FileSurfer(
name="file_surfer",
model_client=model_client,
work_dir=work_dir,
bind_dir=work_dir,
)
team = RoundRobinGroupChat(
participants=[file_surfer, user_proxy],
max_turns=30,
termination_condition=termination,
)
await team.lazy_init()
user_message = await asyncio.get_event_loop().run_in_executor(None, input, ">: ")
stream = team.run_stream(task=user_message)
await Console(stream)
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description="Run FileSurfer with a specified work directory."
)
parser.add_argument(
"--work-dir",
type=str,
default="debug",
help="Directory to use for file surfer.",
)
args = parser.parse_args()
asyncio.run(main(args.work_dir))
|