mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-01 19:16:50 +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>
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
import time
|
|
|
|
from autogen import GroupChatManager
|
|
from autogencap.ActorConnector import ActorConnector
|
|
from autogencap.ag_adapter.CAP2AG import CAP2AG
|
|
from autogencap.ag_adapter.CAPGroupChat import CAPGroupChat
|
|
|
|
from ..actor_runtime import IRuntime
|
|
|
|
|
|
class CAPGroupChatManager:
|
|
def __init__(self, groupchat: CAPGroupChat, llm_config: dict, network: IRuntime):
|
|
self._ensemble: IRuntime = network
|
|
self._cap_group_chat: CAPGroupChat = groupchat
|
|
self._ag_group_chat_manager: GroupChatManager = GroupChatManager(
|
|
groupchat=self._cap_group_chat, llm_config=llm_config
|
|
)
|
|
self._cap_proxy: CAP2AG = CAP2AG(
|
|
ag_agent=self._ag_group_chat_manager,
|
|
the_other_name=self._cap_group_chat.chat_initiator,
|
|
init_chat=False,
|
|
self_recursive=True,
|
|
)
|
|
self._ensemble.register(self._cap_proxy)
|
|
|
|
def initiate_chat(self, txt_msg: str) -> None:
|
|
self._ensemble.connect()
|
|
user_proxy_conn: ActorConnector = self._ensemble.find_by_name(self._cap_group_chat.chat_initiator)
|
|
user_proxy_conn.send_txt_msg(txt_msg)
|
|
self._wait_for_user_exit()
|
|
|
|
def is_running(self) -> bool:
|
|
return self._cap_group_chat.is_running()
|
|
|
|
def _wait_for_user_exit(self) -> None:
|
|
try:
|
|
while self.is_running():
|
|
# Hang out for a while and print out
|
|
# status every now and then
|
|
time.sleep(0.5)
|
|
except KeyboardInterrupt:
|
|
print("Interrupted by user, shutting down.")
|