mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-01 02:57:07 +00:00
8f6590e231
* Core CAP components + Autogen adapter + Demo * Cleanup Readme * C# folder * Cleanup readme * summary_method bug fix * CAN -> CAP * pre-commit fixes * pre-commit fixes * modification of sys path should ignore E402 * fix pre-commit check issues * Updated docs * Clean up docs * more refactoring * better packaging refactor * Refactoring for package changes * Run demo app without autogencap installed or in the path * Remove debug related sleep() * removed CAP in some class names * Investigate a logging framework that supports color in windows * added type hints * remove circular dependency * fixed pre-commit issues * pre-commit ruff issues * removed circular definition * pre-commit fixes * Fix pre-commit issues * pre-commit fixes * updated for _prepare_chat signature changes * Better instructions for demo and some minor refactoring * Added details that explain CAP * Reformat Readme * More ReadMe Formatting * Readme edits * Agent -> Actor * Broker can startup on it's own * Remote AutoGen Agents * Updated docs * 1) StandaloneBroker in demo 2) Removed Autogen only demo options * 1) Agent -> Actor refactor 2) init broker as early * rename user_proxy -> user_proxy_conn * Add DirectorySvc * Standalone demo refactor * Get ActorInfo from DirectorySvc when searching for Actor * Broker cleanup * Proper cleanup and remove debug sleep() * Run one directory service only. * fix paths to run demo apps from command line * Handle keyboard interrupt * Wait for Broker and Directory to start up * Move Terminate AGActor * Accept input from the user in UserProxy * Move sleeps close to operations that bind or connect * Comments * Created an encapsulated CAP Pair for AutoGen pair communication * pre-commit checks * fix pre-commit * Pair should not make assumptions about who is first and who is second * Use task passed into InitiateChat * Standalone directory svc * Fix broken LFS files * Long running DirectorySvc * DirectorySvc does not have a status * Exit DirectorySvc Loop * Debugging Remoting * Reduce frequency of status messages * Debugging remote Actor * roll back git-lfs updates * rollback git-lfs changes * Debug network connectivity * pre-commit fixes * Create a group chat interface familiar to AutoGen GroupChat users * pre-commit fixes
79 lines
3.0 KiB
Python
79 lines
3.0 KiB
Python
import zmq
|
|
import threading
|
|
import traceback
|
|
import time
|
|
from .DebugLog import Debug, Info
|
|
from .Config import xpub_url
|
|
|
|
|
|
class Actor:
|
|
def __init__(self, agent_name: str, description: str):
|
|
self.actor_name: str = agent_name
|
|
self.agent_description: str = description
|
|
self.run = False
|
|
|
|
def connect_network(self, network):
|
|
Debug(self.actor_name, f"is connecting to {network}")
|
|
Debug(self.actor_name, "connected")
|
|
|
|
def _process_txt_msg(self, msg: str, msg_type: str, topic: str, sender: str) -> bool:
|
|
Info(self.actor_name, f"InBox: {msg}")
|
|
return True
|
|
|
|
def _process_bin_msg(self, msg: bytes, msg_type: str, topic: str, sender: str) -> bool:
|
|
Info(self.actor_name, f"Msg: topic=[{topic}], msg_type=[{msg_type}]")
|
|
return True
|
|
|
|
def _recv_thread(self):
|
|
Debug(self.actor_name, "recv thread started")
|
|
self._socket: zmq.Socket = self._context.socket(zmq.SUB)
|
|
self._socket.setsockopt(zmq.RCVTIMEO, 500)
|
|
self._socket.connect(xpub_url)
|
|
str_topic = f"{self.actor_name}"
|
|
Debug(self.actor_name, f"subscribe to: {str_topic}")
|
|
self._socket.setsockopt_string(zmq.SUBSCRIBE, f"{str_topic}")
|
|
try:
|
|
while self.run:
|
|
try:
|
|
topic, msg_type, sender_topic, msg = self._socket.recv_multipart()
|
|
topic = topic.decode("utf-8") # Convert bytes to string
|
|
msg_type = msg_type.decode("utf-8") # Convert bytes to string
|
|
sender_topic = sender_topic.decode("utf-8") # Convert bytes to string
|
|
except zmq.Again:
|
|
continue # No message received, continue to next iteration
|
|
except Exception:
|
|
continue
|
|
if msg_type == "text":
|
|
msg = msg.decode("utf-8") # Convert bytes to string
|
|
if not self._process_txt_msg(msg, msg_type, topic, sender_topic):
|
|
msg = "quit"
|
|
if msg.lower() == "quit":
|
|
break
|
|
else:
|
|
if not self._process_bin_msg(msg, msg_type, topic, sender_topic):
|
|
break
|
|
except Exception as e:
|
|
Debug(self.actor_name, f"recv thread encountered an error: {e}")
|
|
traceback.print_exc()
|
|
finally:
|
|
self.run = False
|
|
Debug(self.actor_name, "recv thread ended")
|
|
|
|
def start(self, context: zmq.Context):
|
|
self._context = context
|
|
self.run: bool = True
|
|
self._thread = threading.Thread(target=self._recv_thread)
|
|
self._thread.start()
|
|
time.sleep(0.01)
|
|
|
|
def disconnect_network(self, network):
|
|
Debug(self.actor_name, f"is disconnecting from {network}")
|
|
Debug(self.actor_name, "disconnected")
|
|
self.stop()
|
|
|
|
def stop(self):
|
|
self.run = False
|
|
self._thread.join()
|
|
self._socket.setsockopt(zmq.LINGER, 0)
|
|
self._socket.close()
|