mirror of
https://github.com/microsoft/autogen.git
synced 2025-09-29 10:09:30 +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
115 lines
3.6 KiB
Python
115 lines
3.6 KiB
Python
import time
|
|
import zmq
|
|
import threading
|
|
from autogencap.DebugLog import Debug, Info, Warn
|
|
from autogencap.Config import xsub_url, xpub_url
|
|
|
|
|
|
class Broker:
|
|
def __init__(self, context: zmq.Context = zmq.Context()):
|
|
self._context: zmq.Context = context
|
|
self._run: bool = False
|
|
self._xpub: zmq.Socket = None
|
|
self._xsub: zmq.Socket = None
|
|
|
|
def start(self) -> bool:
|
|
try:
|
|
# XPUB setup
|
|
self._xpub = self._context.socket(zmq.XPUB)
|
|
self._xpub.setsockopt(zmq.LINGER, 0)
|
|
self._xpub.bind(xpub_url)
|
|
|
|
# XSUB setup
|
|
self._xsub = self._context.socket(zmq.XSUB)
|
|
self._xsub.setsockopt(zmq.LINGER, 0)
|
|
self._xsub.bind(xsub_url)
|
|
|
|
except zmq.ZMQError as e:
|
|
Debug("BROKER", f"Unable to start. Check details: {e}")
|
|
# If binding fails, close the sockets and return False
|
|
if self._xpub:
|
|
self._xpub.close()
|
|
if self._xsub:
|
|
self._xsub.close()
|
|
return False
|
|
|
|
self._run = True
|
|
self._broker_thread: threading.Thread = threading.Thread(target=self.thread_fn)
|
|
self._broker_thread.start()
|
|
time.sleep(0.01)
|
|
return True
|
|
|
|
def stop(self):
|
|
# Error("BROKER_ERR", "fix cleanup self._context.term()")
|
|
Debug("BROKER", "stopped")
|
|
self._run = False
|
|
self._broker_thread.join()
|
|
if self._xpub:
|
|
self._xpub.close()
|
|
if self._xsub:
|
|
self._xsub.close()
|
|
# self._context.term()
|
|
|
|
def thread_fn(self):
|
|
try:
|
|
# Poll sockets for events
|
|
self._poller: zmq.Poller = zmq.Poller()
|
|
self._poller.register(self._xpub, zmq.POLLIN)
|
|
self._poller.register(self._xsub, zmq.POLLIN)
|
|
|
|
# Receive msgs, forward and process
|
|
while self._run:
|
|
events = dict(self._poller.poll(500))
|
|
if self._xpub in events:
|
|
message = self._xpub.recv_multipart()
|
|
Debug("BROKER", f"subscription message: {message[0]}")
|
|
self._xsub.send_multipart(message)
|
|
|
|
if self._xsub in events:
|
|
message = self._xsub.recv_multipart()
|
|
Debug("BROKER", f"publishing message: {message}")
|
|
self._xpub.send_multipart(message)
|
|
|
|
except Exception as e:
|
|
Debug("BROKER", f"thread encountered an error: {e}")
|
|
finally:
|
|
self._run = False
|
|
Debug("BROKER", "thread ended")
|
|
return
|
|
|
|
|
|
# Run a standalone broker that all other Actors can connect to.
|
|
# This can also run inproc with the other actors.
|
|
def main():
|
|
broker = Broker()
|
|
Info("BROKER", "Starting.")
|
|
if broker.start():
|
|
Info("BROKER", "Running.")
|
|
else:
|
|
Warn("BROKER", "Failed to start.")
|
|
return
|
|
|
|
status_interval = 300 # seconds
|
|
last_time = time.time()
|
|
|
|
# Broker is running in a separate thread. Here we are watching the
|
|
# broker's status and printing status every few seconds. This is
|
|
# a good place to print other statistics captured as the broker runs.
|
|
# -- Exits when the user presses Ctrl+C --
|
|
while broker._run:
|
|
# print a message every n seconds
|
|
current_time = time.time()
|
|
elapsed_time = current_time - last_time
|
|
if elapsed_time > status_interval:
|
|
Info("BROKER", "Running.")
|
|
last_time = current_time
|
|
try:
|
|
time.sleep(0.5)
|
|
except KeyboardInterrupt:
|
|
Info("BROKER", "KeyboardInterrupt. Stopping the broker.")
|
|
broker.stop()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|