f9b84c1e2f
Major expansion of the Timmy Time Dashboard: Backend modules: - Swarm subsystem: registry, manager, bidder, coordinator, agent_runner, swarm_node, tasks, comms - L402/Lightning: payment_handler, l402_proxy with HMAC macaroons - Voice NLU: regex-based intent detection (chat, status, swarm, task, help, voice) - Notifications: push notifier for swarm events - Shortcuts: Siri Shortcuts iOS integration endpoints - WebSocket: live dashboard event manager - Inter-agent: agent-to-agent messaging layer Dashboard routes: - /swarm/* — swarm management and agent registry - /marketplace — agent catalog with sat pricing - /voice/* — voice command processing - /mobile — mobile status endpoint - /swarm/live — WebSocket live feed React web dashboard (dashboard-web/): - Sovereign Terminal design — dark theme with Bitcoin orange accents - Three-column layout: status sidebar, workspace tabs, context panel - Chat, Swarm, Tasks, Marketplace tab views - JetBrains Mono typography, terminal aesthetic - Framer Motion animations throughout Tests: 228 passing (expanded from 93) Includes Kimi's additional templates and QA work.
137 lines
3.8 KiB
Python
137 lines
3.8 KiB
Python
"""SQLite-backed agent registry for the swarm.
|
|
|
|
Each agent that joins the swarm registers here with its ID, name, and
|
|
capabilities. The registry is the source of truth for which agents are
|
|
available to bid on tasks.
|
|
"""
|
|
|
|
import sqlite3
|
|
import uuid
|
|
from dataclasses import dataclass, field
|
|
from datetime import datetime, timezone
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
DB_PATH = Path("data/swarm.db")
|
|
|
|
|
|
@dataclass
|
|
class AgentRecord:
|
|
id: str = field(default_factory=lambda: str(uuid.uuid4()))
|
|
name: str = ""
|
|
status: str = "idle" # idle | busy | offline
|
|
capabilities: str = "" # comma-separated tags
|
|
registered_at: str = field(
|
|
default_factory=lambda: datetime.now(timezone.utc).isoformat()
|
|
)
|
|
last_seen: str = field(
|
|
default_factory=lambda: datetime.now(timezone.utc).isoformat()
|
|
)
|
|
|
|
|
|
def _get_conn() -> sqlite3.Connection:
|
|
DB_PATH.parent.mkdir(parents=True, exist_ok=True)
|
|
conn = sqlite3.connect(str(DB_PATH))
|
|
conn.row_factory = sqlite3.Row
|
|
conn.execute(
|
|
"""
|
|
CREATE TABLE IF NOT EXISTS agents (
|
|
id TEXT PRIMARY KEY,
|
|
name TEXT NOT NULL,
|
|
status TEXT NOT NULL DEFAULT 'idle',
|
|
capabilities TEXT DEFAULT '',
|
|
registered_at TEXT NOT NULL,
|
|
last_seen TEXT NOT NULL
|
|
)
|
|
"""
|
|
)
|
|
conn.commit()
|
|
return conn
|
|
|
|
|
|
def _row_to_record(row: sqlite3.Row) -> AgentRecord:
|
|
return AgentRecord(
|
|
id=row["id"],
|
|
name=row["name"],
|
|
status=row["status"],
|
|
capabilities=row["capabilities"],
|
|
registered_at=row["registered_at"],
|
|
last_seen=row["last_seen"],
|
|
)
|
|
|
|
|
|
def register(name: str, capabilities: str = "", agent_id: Optional[str] = None) -> AgentRecord:
|
|
record = AgentRecord(
|
|
id=agent_id or str(uuid.uuid4()),
|
|
name=name,
|
|
capabilities=capabilities,
|
|
)
|
|
conn = _get_conn()
|
|
conn.execute(
|
|
"""
|
|
INSERT OR REPLACE INTO agents (id, name, status, capabilities, registered_at, last_seen)
|
|
VALUES (?, ?, ?, ?, ?, ?)
|
|
""",
|
|
(record.id, record.name, record.status, record.capabilities,
|
|
record.registered_at, record.last_seen),
|
|
)
|
|
conn.commit()
|
|
conn.close()
|
|
return record
|
|
|
|
|
|
def unregister(agent_id: str) -> bool:
|
|
conn = _get_conn()
|
|
cursor = conn.execute("DELETE FROM agents WHERE id = ?", (agent_id,))
|
|
conn.commit()
|
|
deleted = cursor.rowcount > 0
|
|
conn.close()
|
|
return deleted
|
|
|
|
|
|
def get_agent(agent_id: str) -> Optional[AgentRecord]:
|
|
conn = _get_conn()
|
|
row = conn.execute("SELECT * FROM agents WHERE id = ?", (agent_id,)).fetchone()
|
|
conn.close()
|
|
return _row_to_record(row) if row else None
|
|
|
|
|
|
def list_agents(status: Optional[str] = None) -> list[AgentRecord]:
|
|
conn = _get_conn()
|
|
if status:
|
|
rows = conn.execute(
|
|
"SELECT * FROM agents WHERE status = ? ORDER BY registered_at DESC",
|
|
(status,),
|
|
).fetchall()
|
|
else:
|
|
rows = conn.execute(
|
|
"SELECT * FROM agents ORDER BY registered_at DESC"
|
|
).fetchall()
|
|
conn.close()
|
|
return [_row_to_record(r) for r in rows]
|
|
|
|
|
|
def update_status(agent_id: str, status: str) -> Optional[AgentRecord]:
|
|
now = datetime.now(timezone.utc).isoformat()
|
|
conn = _get_conn()
|
|
conn.execute(
|
|
"UPDATE agents SET status = ?, last_seen = ? WHERE id = ?",
|
|
(status, now, agent_id),
|
|
)
|
|
conn.commit()
|
|
conn.close()
|
|
return get_agent(agent_id)
|
|
|
|
|
|
def heartbeat(agent_id: str) -> Optional[AgentRecord]:
|
|
"""Update last_seen timestamp for a registered agent."""
|
|
now = datetime.now(timezone.utc).isoformat()
|
|
conn = _get_conn()
|
|
conn.execute(
|
|
"UPDATE agents SET last_seen = ? WHERE id = ?",
|
|
(now, agent_id),
|
|
)
|
|
conn.commit()
|
|
conn.close()
|
|
return get_agent(agent_id)
|