mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-03 12:08:08 +00:00
466c851743
* Added Runtime Factory to support multiple implementations * Rename to ComponentEnsemble to ZMQRuntime * rename zmq_runtime * rename zmq_runtime * pre-commit fixes * pre-commit fix * pre-commit fixes and default runtime * pre-commit fixes * Rename constants * Rename Constants --------- Co-authored-by: Li Jiang <bnujli@gmail.com>
59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
import time
|
|
|
|
import _paths
|
|
from autogencap.ag_adapter.CAP2AG import CAP2AG
|
|
from autogencap.Config import IGNORED_LOG_CONTEXTS
|
|
from autogencap.DebugLog import Info
|
|
from autogencap.runtime_factory import RuntimeFactory
|
|
|
|
from autogen import UserProxyAgent, config_list_from_json
|
|
|
|
|
|
# Starts the Broker and the Assistant. The UserProxy is started separately.
|
|
class StandaloneUserProxy:
|
|
def __init__(self):
|
|
pass
|
|
|
|
def run(self):
|
|
print("Running the StandaloneUserProxy")
|
|
|
|
user_proxy = UserProxyAgent(
|
|
"user_proxy",
|
|
code_execution_config={"work_dir": "coding"},
|
|
is_termination_msg=lambda x: "TERMINATE" in x.get("content"),
|
|
)
|
|
# Composable Agent Network adapter
|
|
ensemble = RuntimeFactory.get_runtime("ZMQ")
|
|
user_proxy_adptr = CAP2AG(ag_agent=user_proxy, the_other_name="assistant", init_chat=True, self_recursive=True)
|
|
ensemble.register(user_proxy_adptr)
|
|
ensemble.connect()
|
|
|
|
# Send a message to the user_proxy
|
|
user_proxy_conn = ensemble.find_by_name("user_proxy")
|
|
example = "Plot a chart of MSFT daily closing prices for last 1 Month."
|
|
print(f"Example: {example}")
|
|
try:
|
|
user_input = input("Please enter your command: ")
|
|
if user_input == "":
|
|
user_input = example
|
|
print(f"Sending: {user_input}")
|
|
user_proxy_conn.send_txt_msg(user_input)
|
|
|
|
# Hang around for a while
|
|
while user_proxy_adptr.run:
|
|
time.sleep(0.5)
|
|
except KeyboardInterrupt:
|
|
print("Interrupted by user, shutting down.")
|
|
ensemble.disconnect()
|
|
Info("StandaloneUserProxy", "App Exit")
|
|
|
|
|
|
def main():
|
|
IGNORED_LOG_CONTEXTS.extend(["BROKER", "DirectorySvc"])
|
|
assistant = StandaloneUserProxy()
|
|
assistant.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|