mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-02 03:27:34 +00:00
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
import time
|
|
|
|
from AppAgents import FidelityAgent, FinancialPlannerAgent, PersonalAssistant, QuantAgent, RiskManager
|
|
from autogencap.ComponentEnsemble import ComponentEnsemble
|
|
from termcolor import colored
|
|
|
|
|
|
def complex_actor_demo():
|
|
"""
|
|
This function demonstrates the usage of a complex actor system.
|
|
It creates a local actor graph, registers various agents,
|
|
connects them, and interacts with a personal assistant agent.
|
|
The function continuously prompts the user for input messages,
|
|
sends them to the personal assistant agent, and terminates
|
|
when the user enters "quit".
|
|
"""
|
|
ensemble = ComponentEnsemble()
|
|
# Register agents
|
|
ensemble.register(PersonalAssistant())
|
|
ensemble.register(FidelityAgent())
|
|
ensemble.register(FinancialPlannerAgent())
|
|
ensemble.register(RiskManager())
|
|
ensemble.register(QuantAgent())
|
|
# Tell agents to connect to other agents
|
|
ensemble.connect()
|
|
# Get a channel to the personal assistant agent
|
|
pa = ensemble.find_by_name(PersonalAssistant.cls_agent_name)
|
|
info_msg = """
|
|
This is an imaginary personal assistant agent scenario.
|
|
Five actors are connected in a self-determined graph. The user
|
|
can interact with the personal assistant agent by entering
|
|
their name. The personal assistant agent will then enlist
|
|
the other four agents to create a financial plan.
|
|
|
|
Start by entering your name.
|
|
"""
|
|
print(colored(info_msg, "blue"))
|
|
|
|
while True:
|
|
# For aesthetic reasons, better to let network messages complete
|
|
time.sleep(0.1)
|
|
# Get a message from the user
|
|
msg = input(colored("Enter your name (or quit): ", "light_red"))
|
|
# Send the message to the personal assistant agent
|
|
pa.send_txt_msg(msg)
|
|
if msg.lower() == "quit":
|
|
break
|
|
# Cleanup
|
|
|
|
pa.close()
|
|
ensemble.disconnect()
|