mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-04 20:46:11 +00:00
42 lines
1.6 KiB
Python
42 lines
1.6 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 autogencap.ComponentEnsemble import ComponentEnsemble
|
|
|
|
|
|
class CAPGroupChatManager:
|
|
def __init__(self, groupchat: CAPGroupChat, llm_config: dict, network: ComponentEnsemble):
|
|
self._ensemble: ComponentEnsemble = 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.")
|