bus-agent 2.3.0

package/clients/coco_client.py

@@ -0,0 +1,216 @@
+"""
+CoCo Bus Client — Python SDK for Agent Bus
+
+Direct file-based access to the CoCo Agent Bus. No MCP needed.
+Any agent with filesystem access to the bus directory can use this.
+
+Usage:
+    from coco_client import CoCoClient
+
+    bus = CoCoClient("/path/to/mcp-coco/.bus", agent_name="my-agent")
+    bus.register(description="My Python agent", capabilities=["data-analysis", "api-calls"])
+    bus.send("hermes", "Hello from Python!")
+    msgs = bus.fetch_messages()
+    for m in msgs:
+        print(f"[{m['from']}] {m['message']}")
+        bus.reply(m, "Got it!")
+"""
+
+import json
+import os
+import time
+import uuid
+
+
+class CoCoClient:
+    def __init__(self, bus_dir, agent_name=None):
+        self.bus_dir = bus_dir
+        self.msgs_dir = os.path.join(bus_dir, "messages")
+        self.agents_file = os.path.join(bus_dir, "agents.json")
+        self.channels_dir = os.path.join(bus_dir, "channels")
+        self.events_dir = os.path.join(bus_dir, "events")
+        self.agent_name = agent_name or os.environ.get("COCO_AGENT") or os.environ.get("USER", "python-agent")
+        self._ensure_dirs()
+
+    def _ensure_dirs(self):
+        for d in [self.bus_dir, self.msgs_dir, self.channels_dir, self.events_dir]:
+            os.makedirs(d, exist_ok=True)
+
+    def _gen_id(self):
+        return f"{int(time.time() * 1000)}_{uuid.uuid4().hex[:8]}"
+
+    def _load_agents(self):
+        try:
+            if os.path.exists(self.agents_file):
+                with open(self.agents_file) as f:
+                    return json.load(f)
+        except:
+            pass
+        return {}
+
+    def _save_agents(self, agents):
+        with open(self.agents_file, "w") as f:
+            json.dump(agents, f, indent=2)
+
+    # ── Registration ──
+
+    def register(self, description="", capabilities=None, model=None, tags=None, version="1.0.0", status="idle"):
+        """Register yourself on the CoCo Agent Bus."""
+        agents = self._load_agents()
+        existing = agents.get(self.agent_name, {})
+        agents[self.agent_name] = {
+            "name": self.agent_name,
+            "description": description,
+            "capabilities": capabilities or [],
+            "model": model or None,
+            "tags": tags or [],
+            "status": status,
+            "version": version,
+            "endpoints": {},
+            "tools": [],
+            "metadata": {},
+            "last_seen": datetime.utcnow().isoformat() + "Z",
+            "registered_at": existing.get("registered_at", datetime.utcnow().isoformat() + "Z"),
+        }
+        self._save_agents(agents)
+        is_new = not existing
+        return {"registered": True, "is_new": is_new}
+
+    def heartbeat(self):
+        """Mark yourself as online."""
+        agents = self._load_agents()
+        if self.agent_name in agents:
+            agents[self.agent_name]["last_seen"] = datetime.utcnow().isoformat() + "Z"
+            agents[self.agent_name]["status"] = "idle"
+            self._save_agents(agents)
+            return True
+        return False
+
+    def set_status(self, status):
+        """Set status: idle, busy, offline."""
+        agents = self._load_agents()
+        if self.agent_name in agents:
+            agents[self.agent_name]["status"] = status
+            agents[self.agent_name]["last_seen"] = datetime.utcnow().isoformat() + "Z"
+            self._save_agents(agents)
+            return True
+        return False
+
+    def list_agents(self, online_only=False):
+        """List agents on the bus."""
+        agents = self._load_agents()
+        if online_only:
+            cutoff = time.time() * 1000 - 300000
+            agents = {k: v for k, v in agents.items()
+                      if datetime.fromisoformat(v["last_seen"].replace("Z", "+00:00")).timestamp() * 1000 > cutoff}
+        return agents
+
+    # ── Messaging ──
+
+    def send(self, to, message, metadata=None):
+        """Send a DM to another agent."""
+        msg = {
+            "id": self._gen_id(),
+            "from": self.agent_name,
+            "to": to,
+            "message": message,
+            "metadata": metadata or {},
+            "timestamp": datetime.utcnow().isoformat() + "Z",
+        }
+        inbox_dir = os.path.join(self.msgs_dir, to)
+        os.makedirs(inbox_dir, exist_ok=True)
+        with open(os.path.join(inbox_dir, f"{msg['id']}.json"), "w") as f:
+            json.dump(msg, f, indent=2)
+
+        # Outbox
+        outbox_dir = os.path.join(self.msgs_dir, f"{self.agent_name}_outbox")
+        os.makedirs(outbox_dir, exist_ok=True)
+        with open(os.path.join(outbox_dir, f"{msg['id']}.json"), "w") as f:
+            json.dump(msg, f, indent=2)
+
+        return msg["id"]
+
+    def broadcast(self, message):
+        """Send to all agents."""
+        agents = self._load_agents()
+        results = []
+        for name in agents:
+            if name != self.agent_name:
+                mid = self.send(name, message, {"broadcast": True})
+                results.append({"to": name, "message_id": mid})
+        return results
+
+    def fetch_messages(self, limit=50, from_agent=None):
+        """Read your inbox."""
+        inbox_dir = os.path.join(self.msgs_dir, self.agent_name)
+        if not os.path.exists(inbox_dir):
+            return []
+
+        files = sorted([f for f in os.listdir(inbox_dir) if f.endswith(".json")])
+        messages = []
+        for f in files[-limit:]:
+            try:
+                with open(os.path.join(inbox_dir, f)) as fp:
+                    msg = json.load(fp)
+                    if from_agent and msg.get("from") != from_agent:
+                        continue
+                    messages.append(msg)
+            except:
+                pass
+        return messages
+
+    def reply(self, original_msg, text):
+        """Reply to a message."""
+        return self.send(original_msg["from"], text, {
+            "in_reply_to": original_msg["id"],
+            "original_from": original_msg.get("from"),
+        })
+
+    # ── Channels ──
+
+    def channel_send(self, channel_id, message):
+        """Send a message to a channel."""
+        ch_path = os.path.join(self.channels_dir, f"{channel_id}.json")
+        if not os.path.exists(ch_path):
+            raise ValueError(f"Channel '{channel_id}' not found")
+
+        msg = {
+            "id": self._gen_id(),
+            "channel": channel_id,
+            "from": self.agent_name,
+            "message": message,
+            "timestamp": datetime.utcnow().isoformat() + "Z",
+        }
+
+        log_dir = os.path.join(self.channels_dir, channel_id, "log")
+        os.makedirs(log_dir, exist_ok=True)
+        with open(os.path.join(log_dir, f"{msg['id']}.json"), "w") as f:
+            json.dump(msg, f, indent=2)
+
+        # DM all members
+        with open(ch_path) as f:
+            channel = json.load(f)
+        for member in channel.get("members", []):
+            if member != self.agent_name:
+                self.send(member, f"[{channel_id}] {message}", {"channel": channel_id})
+
+        return msg["id"]
+
+    def channel_history(self, channel_id, limit=20):
+        """Read recent messages in a channel."""
+        log_dir = os.path.join(self.channels_dir, channel_id, "log")
+        if not os.path.exists(log_dir):
+            return []
+        files = sorted([f for f in os.listdir(log_dir) if f.endswith(".json")])
+        messages = []
+        for f in files[-limit:]:
+            try:
+                with open(os.path.join(log_dir, f)) as fp:
+                    messages.append(json.load(fp))
+            except:
+                pass
+        return messages
+
+
+# Make datetime available
+from datetime import datetime

package/coco-aliases.sh

@@ -0,0 +1,10 @@
+
+# ── CoCo Bus Aliases ──
+alias coco-agents='node E:\_system\.openclaw\workspace\repos\mcp-coco\coco-cli.js agents'
+alias coco-inbox='node E:\_system\.openclaw\workspace\repos\mcp-coco\coco-cli.js inbox'
+alias coco-send='node E:\_system\.openclaw\workspace\repos\mcp-coco\coco-cli.js send'
+alias coco-whoami='node E:\_system\.openclaw\workspace\repos\mcp-coco\coco-cli.js whoami'
+alias coco-status='node E:\_system\.openclaw\workspace\repos\mcp-coco\coco-cli.js status'
+alias coco-search='node E:\_system\.openclaw\workspace\repos\mcp-coco\coco-cli.js search'
+alias coco-watch='node E:\_system\.openclaw\workspace\repos\mcp-coco\coco-cli.js watch'
+alias coco-channel='node E:\_system\.openclaw\workspace\repos\mcp-coco\coco-cli.js channel'