mirror of
https://github.com/microsoft/autogen.git
synced 2025-09-26 08:34:10 +00:00
db30ec8961
* 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
88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
from typing import Dict, Optional, Union
|
|
from autogen import Agent
|
|
from ..ActorConnector import ActorConnector
|
|
from ..proto.Autogen_pb2 import GenReplyReq, GenReplyResp, PrepChat, ReceiveReq, Terminate
|
|
|
|
|
|
class AutoGenConnector:
|
|
"""
|
|
A specialized ActorConnector class for sending and receiving Autogen messages
|
|
to/from the CAP system.
|
|
"""
|
|
|
|
def __init__(self, cap_sender: ActorConnector):
|
|
self._can_channel: ActorConnector = cap_sender
|
|
|
|
def close(self):
|
|
"""
|
|
Close the connector.
|
|
"""
|
|
self._can_channel.close()
|
|
|
|
def _send_msg(self, msg):
|
|
"""
|
|
Send a message to CAP.
|
|
"""
|
|
serialized_msg = msg.SerializeToString()
|
|
self._can_channel.send_bin_msg(type(msg).__name__, serialized_msg)
|
|
|
|
def send_gen_reply_req(self):
|
|
"""
|
|
Send a GenReplyReq message to CAP and receive the response.
|
|
"""
|
|
msg = GenReplyReq()
|
|
serialized_msg = msg.SerializeToString()
|
|
# Setting retry to -1 to keep trying until a response is received
|
|
# This normal AutoGen behavior but does not handle the case when an AutoGen agent
|
|
# is not running. In that case, the connector will keep trying indefinitely.
|
|
_, _, resp = self._can_channel.binary_request(type(msg).__name__, serialized_msg, retry=-1)
|
|
gen_reply_resp = GenReplyResp()
|
|
gen_reply_resp.ParseFromString(resp)
|
|
return gen_reply_resp.data
|
|
|
|
def send_receive_req(
|
|
self,
|
|
message: Union[Dict, str],
|
|
sender: Agent,
|
|
request_reply: Optional[bool] = None,
|
|
silent: Optional[bool] = False,
|
|
):
|
|
"""
|
|
Send a ReceiveReq message to CAP.
|
|
"""
|
|
msg = ReceiveReq()
|
|
if isinstance(message, dict):
|
|
for key, value in message.items():
|
|
msg.data_map.data[key] = value
|
|
elif isinstance(message, str):
|
|
msg.data = message
|
|
msg.sender = sender.name
|
|
if request_reply is not None:
|
|
msg.request_reply = request_reply
|
|
if silent is not None:
|
|
msg.silent = silent
|
|
self._send_msg(msg)
|
|
|
|
def send_terminate(self, sender: Agent):
|
|
"""
|
|
Send a Terminate message to CAP.
|
|
"""
|
|
msg = Terminate()
|
|
msg.sender = sender.name
|
|
self._send_msg(msg)
|
|
|
|
def send_prep_chat(self, recipient: "Agent", clear_history: bool, prepare_recipient: bool = True) -> None:
|
|
"""
|
|
Send a PrepChat message to CAP.
|
|
|
|
Args:
|
|
recipient (Agent): _description_
|
|
clear_history (bool): _description_
|
|
prepare_recipient (bool, optional): _description_. Defaults to True.
|
|
"""
|
|
msg = PrepChat()
|
|
msg.recipient = recipient.name
|
|
msg.clear_history = clear_history
|
|
msg.prepare_recipient = prepare_recipient
|
|
self._send_msg(msg)
|