58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
[OPS] Agent Dispatch Framework
|
|
Part of the Gemini Sovereign Infrastructure Suite.
|
|
|
|
Replaces ad-hoc dispatch scripts with a unified framework for tasking agents.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import subprocess
|
|
|
|
# --- CONFIGURATION ---
|
|
FLEET = {
|
|
"allegro": "167.99.126.228",
|
|
"bezalel": "159.203.146.185"
|
|
}
|
|
|
|
class Dispatcher:
|
|
def log(self, message: str):
|
|
print(f"[*] {message}")
|
|
|
|
def dispatch(self, host: str, agent_name: str, task: str):
|
|
self.log(f"Dispatching task to {agent_name} on {host}...")
|
|
|
|
ip = FLEET[host]
|
|
# Command to run the agent on the remote machine
|
|
# Assumes hermes-agent is installed in /opt/hermes
|
|
remote_cmd = f"cd /opt/hermes && python3 run_agent.py --agent {agent_name} --task '{task}'"
|
|
|
|
ssh_cmd = ["ssh", "-o", "StrictHostKeyChecking=no", f"root@{ip}", remote_cmd]
|
|
|
|
try:
|
|
res = subprocess.run(ssh_cmd, capture_output=True, text=True)
|
|
if res.returncode == 0:
|
|
self.log(f"[SUCCESS] {agent_name} completed task.")
|
|
print(res.stdout)
|
|
else:
|
|
self.log(f"[FAILURE] {agent_name} failed task.")
|
|
print(res.stderr)
|
|
except Exception as e:
|
|
self.log(f"[ERROR] Dispatch failed: {e}")
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Gemini Agent Dispatcher")
|
|
parser.add_argument("host", choices=list(FLEET.keys()), help="Host to dispatch to")
|
|
parser.add_argument("agent", help="Agent name")
|
|
parser.add_argument("task", help="Task description")
|
|
|
|
args = parser.parse_args()
|
|
|
|
dispatcher = Dispatcher()
|
|
dispatcher.dispatch(args.host, args.agent, args.task)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|