Something went wrong. Try again.
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180#!/usr/bin/env python3"""Multi-bot launcher for running all 5 Void agents simultaneously with aggregated logs.Usage: python run_bots.py [bsky.py arguments...]Example: python run_bots.py --synthesis-interval 0 --no-git
Bots: void, herald, archivist, grunk, ezra"""
import subprocessimport threadingimport signalimport sysimport osfrom typing import List
# ANSI color codes for each botCOLORS = { 'void': '\033[91m', # Red 'herald': '\033[94m', # Blue 'archivist': '\033[92m', # Green 'grunk': '\033[93m', # Yellow 'ezra': '\033[95m', # Magenta}RESET = '\033[0m'BOLD = '\033[1m'
# Bot names and their config filesBOTS = { 'void': 'configs/config.yaml', 'herald': 'configs/herald.yaml', 'archivist': 'configs/archivist.yaml', 'grunk': 'configs/grunk.yaml', 'ezra': 'configs/ezra.yaml',}
# Track all running processesprocesses: List[subprocess.Popen] = []shutdown_flag = threading.Event()shutdown_in_progress = threading.Lock()
def stream_output(proc: subprocess.Popen, bot_name: str, stream_name: str): """Read from a process stream and print with colored prefix.""" stream = proc.stdout if stream_name == 'stdout' else proc.stderr color = COLORS.get(bot_name, '') prefix = f"{color}{BOLD}[{bot_name}]{RESET} "
try: for line in iter(stream.readline, b''): if shutdown_flag.is_set(): break try: decoded = line.decode('utf-8').rstrip() if decoded: # Only print non-empty lines print(f"{prefix}{decoded}", flush=True) except UnicodeDecodeError: # Handle binary output gracefully print(f"{prefix}[binary output]", flush=True) except Exception as e: if not shutdown_flag.is_set(): print(f"{prefix}Error reading {stream_name}: {e}", flush=True) finally: stream.close()
def shutdown_handler(signum, frame): """Handle shutdown signals gracefully.""" # Prevent re-entry if already shutting down if not shutdown_in_progress.acquire(blocking=False): # Already shutting down, force kill on second Ctrl+C print(f"\n{BOLD}⚡ Force killing all processes...{RESET}", flush=True) for proc in processes: if proc.poll() is None: try: proc.kill() except: pass sys.exit(1)
try: print(f"\n{BOLD}🛑 Shutting down all bots...{RESET}", flush=True) shutdown_flag.set()
# Send SIGTERM to all processes for proc in processes: if proc.poll() is None: # Still running try: proc.terminate() except Exception as e: print(f"Error terminating process: {e}", flush=True)
# Wait for graceful shutdown (up to 3 seconds each) for i, proc in enumerate(processes): try: proc.wait(timeout=3) except subprocess.TimeoutExpired: print(f"Process {i} didn't exit gracefully, force killing...", flush=True) proc.kill() try: proc.wait(timeout=1) except: pass
print(f"{BOLD}✅ All bots stopped{RESET}", flush=True) finally: sys.exit(0)
def main(): """Launch all bots with aggregated output.""" # Set up signal handlers signal.signal(signal.SIGINT, shutdown_handler) signal.signal(signal.SIGTERM, shutdown_handler)
# Get command-line arguments to pass to each bot bot_args = sys.argv[1:]
print(f"{BOLD}🚀 Starting {len(BOTS)} bots...{RESET}", flush=True) if bot_args: print(f"{BOLD}Arguments: {' '.join(bot_args)}{RESET}", flush=True) print()
# Start each bot threads = [] for bot_name, config_file in BOTS.items(): cmd = ['python', 'bsky.py', '--config', config_file] + bot_args
try: proc = subprocess.Popen( cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, ) processes.append(proc)
# Start threads to read stdout and stderr stdout_thread = threading.Thread( target=stream_output, args=(proc, bot_name, 'stdout'), daemon=True ) stderr_thread = threading.Thread( target=stream_output, args=(proc, bot_name, 'stderr'), daemon=True )
stdout_thread.start() stderr_thread.start() threads.append(stdout_thread) threads.append(stderr_thread)
color = COLORS.get(bot_name, '') print(f"{color}{BOLD}[{bot_name}]{RESET} Started (PID: {proc.pid})", flush=True)
except Exception as e: print(f"Failed to start {bot_name}: {e}", flush=True)
print() print(f"{BOLD}📡 Monitoring {len(processes)} bots. Press Ctrl+C to stop all.{RESET}", flush=True) print()
# Wait for all processes to complete (or until interrupted) try: for proc in processes: proc.wait() except KeyboardInterrupt: # Signal handler will take care of cleanup pass
# If we get here naturally (all processes exited), clean up if not shutdown_flag.is_set(): print(f"\n{BOLD}All bots have exited.{RESET}", flush=True) shutdown_flag.set()
if __name__ == '__main__': main()