mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-05 04:57:41 +00:00
31d2d37d88
* 1) Removed most framework sleeps 2) refactored connection code * pre-commit fixes * pre-commit * ignore protobuf files in pre-commit checks * Fix duplicate actor registration * refactor change * Nicer printing of Actors * 1) Report recv_multipart errors 4) Always send 4 parts * AutoGen generate_reply expects to wait indefinitely for an answer. CAP can wait a certain amount and give up. In order to reconcile the two, AutoGenConnector is set to wait indefinitely. * pre-commit formatting fixes * pre-commit format changes * don't check autogenerated proto py files * Iterating on CAP interface for AutoGen * User proxy must initiate chat * autogencap pypi package * added dependencies * serialize/deserialize dictionary elements to json when dealing with ReceiveReq * 1) Removed most framework sleeps 2) refactored connection code * Nicer printing of Actors * AutoGen generate_reply expects to wait indefinitely for an answer. CAP can wait a certain amount and give up. In order to reconcile the two, AutoGenConnector is set to wait indefinitely. * pre-commit formatting fixes * pre-commit format changes * Iterating on CAP interface for AutoGen * User proxy must initiate chat * autogencap pypi package * added dependencies * serialize/deserialize dictionary elements to json when dealing with ReceiveReq * pre-commit check fixes * fix pre-commit issues * Better encapsulation of logging * pre-commit fix * pip package update
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import time
|
|
|
|
import _paths
|
|
from autogencap.ag_adapter.agent import Agent
|
|
from autogencap.Config import IGNORED_LOG_CONTEXTS
|
|
from autogencap.LocalActorNetwork import LocalActorNetwork
|
|
|
|
from autogen import UserProxyAgent
|
|
|
|
# Filter out some Log message contexts
|
|
IGNORED_LOG_CONTEXTS.extend(["BROKER"])
|
|
|
|
|
|
def main():
|
|
# Standard AutoGen
|
|
user_proxy = UserProxyAgent(
|
|
"user_proxy",
|
|
code_execution_config={"work_dir": "coding"},
|
|
is_termination_msg=lambda x: "TERMINATE" in x.get("content"),
|
|
)
|
|
|
|
# Wrap AutoGen Agent in CAP
|
|
cap_user_proxy = Agent(user_proxy, counter_party_name="assistant", init_chat=True)
|
|
# Create the message bus
|
|
network = LocalActorNetwork()
|
|
# Add the user_proxy to the message bus
|
|
cap_user_proxy.register(network)
|
|
# Start message processing
|
|
network.connect()
|
|
|
|
# Wait for the user_proxy to finish
|
|
interact_with_user(network, cap_user_proxy)
|
|
# Cleanup
|
|
network.disconnect()
|
|
|
|
|
|
# Starts the Broker and the Assistant. The UserProxy is started separately.
|
|
def interact_with_user(network, cap_assistant):
|
|
user_proxy_conn = network.lookup_actor("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 cap_assistant.running():
|
|
time.sleep(0.5)
|
|
except KeyboardInterrupt:
|
|
print("Interrupted by user, shutting down.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|