opusmanSEEKv4/token_compactor.py

#!/usr/bin/env python3
"""
token_compactor.py
Implements token-efficient heartbeat + compaction protocol for all agents.
"""

import json
from pathlib import Path
from datetime import datetime

STATE_FILE = Path("STATE_COMPACT.json")

def compact_state(agent_id: str, new_data: dict) -> dict:
    """Maintain rolling compact state (<500 tokens)."""
    if STATE_FILE.exists():
        with open(STATE_FILE) as f:
            state = json.load(f)
    else:
        state = {"version": "1.0", "last_compacted": None, "agents": {}}

    state["agents"][agent_id] = {
        "last_update": datetime.now().isoformat(),
        "health": new_data.get("health", "unknown"),
        "tokens_saved": new_data.get("tokens_saved", 0),
        "tasks_completed": new_data.get("tasks_completed", 0)
    }
    state["last_compacted"] = datetime.now().isoformat()

    with open(STATE_FILE, "w") as f:
        json.dump(state, f, indent=2)

    return state

def get_heartbeat(agent_id: str) -> dict:
    """Return compact heartbeat packet (<300 tokens)."""
    if not STATE_FILE.exists():
        return {"agent_id": agent_id, "status": "no_state"}

    with open(STATE_FILE) as f:
        state = json.load(f)

    agent = state["agents"].get(agent_id, {})
    return {
        "agent_id": agent_id,
        "ts": agent.get("last_update"),
        "health": agent.get("health"),
        "tokens_saved": agent.get("tokens_saved", 0)
    }

if __name__ == "__main__":
    print("Token Compactor ready. Use compact_state() and get_heartbeat()")