prose-contracts/scripts/agent-health-check.py

#!/usr/bin/env python3
"""
/root/scripts/agent-health-check.py — Consolidated Agent Health Verification v2

Verifies: LiteLLM keys (agent-specific), GPU port conflicts, agent Zulip streaming,
gateway liveness, gateway log health, CT liveness, config YAML integrity,
wrapper/CLI integrity, vault secret non-emptiness. NEVER restarts anything.

Usage:
  python3 /root/scripts/agent-health-check.py          # Full check
  python3 /root/scripts/agent-health-check.py --json   # Machine-readable
  python3 /root/scripts/agent-health-check.py --quiet  # Only output on failure

Cron: */10 * * * * python3 /root/scripts/agent-health-check.py --quiet

Changelog:
  v2 (2026-07-26): Added CT liveness, config validation, wrapper integrity,
    vault secret emptiness check. Fixed Koby/Koonimo SSH hosts and agent key
    name format ({NAME}_LITELLM_API_KEY not LITELLM_API_KEY_{NAME}).
    Fleet roster: tanko (.122), mumuni (.24, inside abiba CT100), koby (.129), koonimo (.114),
    abiba (.24).
  v3 (2026-09-08): GPU unit repoint verified live (.8 llama-chat-api.service,
    .110 llama-server.service, .15 strix-server.service) — .8 was probing a stale
    llama-server unit that reads inactive, producing false UNREACHABLE legs.
    systemctl is-active no longer swallows non-zero exit as SSH failure.
    Fixed UnboundLocalError on the abiba/koonimo gateway leg (pid unbound in the
    summary f-string). Abiba's LiteLLM key now comes from /root/.pi/agent/env.sh
    (#735 agent separation; creds moved out of shared /root/.bashrc).
  v4 (2026-09-10): probe-drift round 2 (prose-contracts follow-up to #65/#66/#68).
    abiba declared pi-only runtime — Hermes-era config/wrapper/gateway checks are
    skipped (harness purge). koby declared report_only per the captain's
    2026-08-17 ruling: every koby leg is detected and reported, never counted as a
    fleet failure and never repaired. koby's PVE mapping corrected to storepve
    (CT 111 tdunna lives on .6 — the old amdpve mapping produced a false
    ct-unreachable). The wrapper infisical-path check no longer FAILs .env-based
    wrappers that legitimately never invoke infisical (koonimo/baggy). Every run
    now prints absolute execution provenance (script + cwd) in the header and in
    --json output so a stale-consumer report is distinguishable from a fault at
    read time.
"""

import subprocess, json, sys, os, time
from datetime import datetime

LITELLM = "http://192.168.68.116:80"
INFISICAL_PROJECT = "322fceab-39da-4854-a55a-568e76c0f13f"
INFISICAL_ENV = "prod"

# PVE node IPs for CT liveness checks
PVE_NODES = {
    "amdpve":  "192.168.68.15",
    "minipve": "192.168.68.12",
    "storepve": "192.168.68.6",
    "acerpve": "192.168.68.9",
    "ocupve":  "192.168.68.5",
}

# Agent definitions: ct, host, user, pve_node, vault_key_name
AGENTS = {
    "tanko":   {"ct": 112, "host": "192.168.68.122", "user": "jerome", "pve": "amdpve",  "vault_key": "TANKO_LITELLM_API_KEY", "runtime": "dsh"},
    # abiba = pi agent (.24) — no vault key; its LiteLLM key is read from its
    # local env file (key_env below), not from the shared vault or .bashrc.
    # runtime=pi: abiba has run pi-only since the harness purge. There is no
    # Hermes gateway, no ~/.hermes/config.yaml and no hermes CLI wrapper on .24
    # (the /root/.local/bin/hermes symlink is dangling), so the Hermes-era
    # config/wrapper/gateway legs are skipped rather than reported as faults.
    "abiba":   {"ct": 100, "host": "192.168.68.24", "user": "root", "pve": "minipve",
               "vault_key": None, "runtime": "pi",
               "key_env": {"file": "/root/.pi/agent/env.sh", "var": "LITELLM_API_KEY"}},
    # koby = report-only (captain's 2026-08-17 ruling, Rule 17): detect and
    # report, NEVER repair, and never count against fleet failures. CT 111
    # (tdunna) lives on storepve (.6) — verified live 2026-09-10; the previous
    # amdpve mapping made `pct status 111` fail and read as ct-unreachable.
    "koby":    {"ct": 111, "host": "192.168.68.129", "user": "root",   "pve": "storepve", "vault_key": "KOBY_LITELLM_API_KEY", "report_only": True},
    "koonimo": {"ct": 113, "host": "192.168.68.114", "user": "root",   "pve": "amdpve",  "vault_key": "KOONIMO_LITELLM_API_KEY"},
}

# Systemd units verified live 2026-09-08 (systemctl list-units on each host):
#   .8   rtx3090 (gpu-dense)  -> llama-chat-api.service  (active; the old
#        llama-server.service unit file is stale/inactive — probing it read as
#        UNREACHABLE for a healthy process)
#   .110 rtx5070 (ocu-llm VM) -> llama-server.service    (active)
#   .15  strixhalo (amdpve)   -> strix-server.service    (active)
GPU_HOSTS = {
    "gpu-rtx3090 (.8)":     {"host": "192.168.68.8",   "port": 8080, "service": "llama-chat-api.service"},
    "gpu-rtx5070 (.110)":   {"host": "192.168.68.110", "port": 8080, "service": "llama-server.service"},
    "gpu-strixhalo (.15)":  {"host": "192.168.68.15",  "port": 8080, "service": "strix-server.service"},
}

FAIL = []


def _fail(key, agent_name=None):
    """Record a failure, except for report-only agents.

    Koby is report-only per the captain's 2026-08-17 ruling (Rule 17): its legs
    are detected and reported, never repaired and never counted as fleet
    failures. A red fleet alert on a known report-only leg is a false alarm. Any
    non-report-only agent (or a leg with no agent, e.g. GPU hosts) records
    normally.
    """
    if agent_name and AGENTS.get(agent_name, {}).get("report_only"):
        print(f"  🔍 report-only ({agent_name}): {key} — reported, not counted/repaired")
        return
    FAIL.append(key)


INFISICAL_TOKEN = os.environ.get("INFISICAL_TOKEN")
INFISICAL_API_URL = os.environ.get("INFISICAL_API_URL", "https://vault.sysloggh.net")

# Fallback: if no env token, read the shared vault token file
if not INFISICAL_TOKEN:
    _token_path = os.path.expanduser("~/.infisical-token")
    if os.path.isfile(_token_path):
        try:
            with open(_token_path) as _f:
                INFISICAL_TOKEN = _f.read().strip()
        except (OSError, UnicodeDecodeError):
            pass

# ── Helpers ──────────────────────────────────────────────────────────

def ssh(host, cmd, user="root"):
    """Execute a command on a remote host, return stdout or None."""
    try:
        result = subprocess.run(
            ["ssh", "-o", "StrictHostKeyChecking=no", "-o", "ConnectTimeout=8",
             f"{user}@{host}", cmd],
            capture_output=True, text=True, timeout=15
        )
        return result.stdout.strip() if result.returncode == 0 else None
    except:
        return None

def http_get(url, headers=None, timeout=5):
    """Return HTTP status code as string."""
    try:
        cmd = ["curl", "-sfk", "--connect-timeout", str(timeout), "-o", "/dev/null", "-w", "%{http_code}"]
        if headers:
            for k, v in headers.items():
                cmd.extend(["-H", f"{k}: {v}"])
        cmd.append(url)
        result = subprocess.run(cmd, capture_output=True, text=True, timeout=timeout+3)
        return result.stdout.strip() or "000"
    except:
        return "timeout"

def http_json(url, headers=None, timeout=5):
    """Return parsed JSON from URL, or None."""
    try:
        cmd = ["curl", "-sfk", "--connect-timeout", str(timeout)]
        if headers:
            for k, v in headers.items():
                cmd.extend(["-H", f"{k}: {v}"])
        cmd.append(url)
        result = subprocess.run(cmd, capture_output=True, text=True, timeout=timeout+3)
        return json.loads(result.stdout) if result.returncode == 0 and result.stdout else None
    except:
        return None

def run_infisical(args, quiet=True):
    """Run infisical CLI with env-based auth, return stdout or None."""
    env = os.environ.copy()
    env["INFISICAL_API_URL"] = INFISICAL_API_URL
    if INFISICAL_TOKEN:
        env["INFISICAL_TOKEN"] = INFISICAL_TOKEN
    try:
        result = subprocess.run(
            ["/usr/bin/infisical"] + args,
            capture_output=True, text=True, timeout=15, env=env
        )
        return result.stdout.strip() if result.returncode == 0 else None
    except:
        return None

# ── KEY LOOKUP FIX ───────────────────────────────────────────────────

def _get_agent_key(agent_name, vault_key_name):
    """Retrieve agent-specific key from Infisical vault.
    
    Uses {NAME}_LITELLM_API_KEY format (e.g., TANKO_LITELLM_API_KEY,
    KOONIMO_LITELLM_API_KEY) which matches actual vault key names.
    """
    if not vault_key_name:
        return None

    # Primary: get the agent-specific key by name
    key = run_infisical([
        "secrets", "get", vault_key_name,
        "--projectId=" + INFISICAL_PROJECT,
        "--env=" + INFISICAL_ENV,
        "--plain",
    ])
    if key and key.startswith("sk-"):
        return key

    # Fallback: export all and search for the key name
    try:
        export = run_infisical([
            "export",
            "--projectId=" + INFISICAL_PROJECT,
            "--env=" + INFISICAL_ENV,
            "--format=dotenv",
        ])
        if export:
            for line in export.splitlines():
                if line.startswith(vault_key_name + "="):
                    value = line.split("=", 1)[1].strip().strip('"').strip("'")
                    if value.startswith("sk-"):
                        return value
    except:
        pass

    return None


def _read_env_export(path, var):
    """Parse `export VAR=value` (or `VAR=value`) out of a local env file.

    #735 agent separation (2026-09-06): agent creds moved out of the shared
    /root/.bashrc into per-agent env files under /root/.pi/agent/ (bashrc's
    source line keeps abiba shells resolving them, but the file of record is
    env.sh). Do NOT fall back to /root/.bashrc here: desktop (.200) SSH
    sessions override LITELLM_API_KEY with mumuni's key, so sourcing bashrc
    would validate the wrong identity.
    """
    try:
        with open(os.path.expanduser(path)) as _f:
            for line in _f:
                line = line.strip()
                if not (line.startswith("export " + var + "=") or line.startswith(var + "=")):
                    continue
                value = line.split("=", 1)[1].strip().strip('"').strip("'")
                if value:
                    return value
    except (OSError, UnicodeDecodeError):
        pass
    return None


def load_agent_keys():
    """Populate AGENTS[*]["key"] from the vault or the agent's local env file.

    Called from main(), not at import: keeping this out of module scope lets the
    module be imported (and unit tested) without live vault/SSH access. Vault
    format is {NAME}_LITELLM_API_KEY (project 322fceab-39da-4854-a55a-568e76c0f13f,
    env prod); abiba has no vault key and reads LITELLM_API_KEY from its local
    /root/.pi/agent/env.sh (moved there from /root/.bashrc in #735).
    """
    for agent_name in AGENTS:
        info = AGENTS[agent_name]
        key = _get_agent_key(agent_name, info.get("vault_key"))
        if not key and info.get("key_env"):
            key = _read_env_export(info["key_env"]["file"], info["key_env"]["var"])
        AGENTS[agent_name]["key"] = key


# ═══════════════════════════════════════════════════════════════════
# CHECK 1: LiteLLM Key Validation (agent-specific keys)
# ═══════════════════════════════════════════════════════════════════

def check_keys():
    for name, agent in AGENTS.items():
        key = agent.get("key")
        if not key:
            print(f"  ❌ {name}: NO KEY FOUND (vault/env empty or unreachable)")
            _fail(f"key:{name}:no-key", name)
            continue
        data = http_json(f"{LITELLM}/v1/models",
                         headers={"Authorization": f"Bearer {key}"})
        if data and data.get("data"):
            model = data["data"][0].get("id", "?")
            print(f"  ✅ {name}: key valid → {model}")
        else:
            print(f"  ❌ {name}: KEY FAILURE — auth rejected or unreachable")
            _fail(f"key:{name}", name)


# ═══════════════════════════════════════════════════════════════════
# CHECK 2: GPU Port Conflict Detection (unit names verified live 2026-09-08)
# ═══════════════════════════════════════════════════════════════════

def check_gpu_ports():
    for label, gpu in GPU_HOSTS.items():
        host = gpu["host"]
        port = gpu["port"]
        svc = gpu["service"]

        # `systemctl is-active` exits non-zero when the unit is inactive or
        # missing, which the ssh() helper would swallow as an SSH failure and
        # report as UNREACHABLE. `|| true` keeps the real state word so we can
        # tell "unit inactive" from "host unreachable".
        svc_status = ssh(host, f"systemctl is-active {svc} || true")
        port_owner = ssh(host, f"ss -tlnp 2>/dev/null | grep -Po ':{port}\\s+.*pid=\\K[0-9]+' | head -1")

        if not svc_status:
            print(f"  ❌ {label}: UNREACHABLE")
            FAIL.append(f"gpu-unreachable:{host}")
            continue

        if not port_owner:
            print(f"  ❌ {label}: PORT {port} NOT LISTENING (svc={svc_status})")
            FAIL.append(f"gpu-no-port:{label}")
        elif svc_status != "active":
            svc_pid = ssh(host, f"systemctl show {svc} -p MainPID 2>/dev/null | cut -d= -f2")
            if svc_pid and port_owner != svc_pid:
                print(f"  ❌ {label}: GHOST PROCESS — port owned by pid {port_owner}, svc pid {svc_pid} (svc={svc_status})")
                FAIL.append(f"gpu-ghost:{label}:{port_owner}")
            else:
                print(f"  ⚠️  {label}: svc={svc_status}, port owned by {port_owner}")
        else:
            health = ssh(host, f"curl -s --max-time 5 http://localhost:{port}/health")
            if health and '"status":"ok"' in health:
                print(f"  ✅ {label}: healthy (pid={port_owner})")
            elif health and '"status":"no slot available"' in health:
                print(f"  ✅ {label}: healthy (loading, pid={port_owner})")
            elif health and '"error"' in health.lower():
                print(f"  ⚠️  {label}: error response (pid={port_owner}): {health[:80]}")
            else:
                print(f"  ⚠️  {label}: unknown health (pid={port_owner}): {str(health)[:80]}")


# ═══════════════════════════════════════════════════════════════════
# CHECK 3: Agent Gateway Liveness + Streaming (now covers all agents)
# ═══════════════════════════════════════════════════════════════════

def check_agents():
    for name, agent in AGENTS.items():
        host = agent.get("host")
        user = agent.get("user")
        ct = agent["ct"]
        report_only = agent.get("report_only", False)

        # Tanko runs on DSH (DeepSeek Harness) since 2026-08-27 — it no longer runs a
        # Hermes gateway, so skip the Hermes gateway/state/streaming/journal checks.
        # Non-Hermes runtimes have no gateway to probe. dsh = Tanko since
        # 2026-08-27; pi = abiba since the harness purge (.24 is pi-only).
        if agent.get("runtime") in ("dsh", "pi"):
            is_dsh = agent.get("runtime") == "dsh"
            label = "DSH (DeepSeek Harness)" if is_dsh else "pi-only runtime"
            since = "since 2026-08-27" if is_dsh else "since the harness purge"
            live = ssh(host, "true", user=user)
            print(f"  {'✅' if live is not None else '❌'} {name}: {label} — "
                  f"no Hermes gateway {since} (CT {ct}, SSH {'OK' if live is not None else 'FAIL'})")
            if live is None:
                _fail(f"unreachable:{name}", name)
            continue

        if not host or not user:
            print(f"  ⬜ {name} (CT {ct}): cannot SSH — skip liveness check")
            continue

        # Resolve the Hermes gateway PID once, before the report-only branch:
        # the summary line below renders `pid`, and it used to be bound only in
        # the report-only path — leaving it unbound on the abiba/koonimo path
        # raised UnboundLocalError and crashed the whole check. Agents without
        # a gateway get pid=?.
        pid = ssh(host, "pgrep -f '[h]ermes_cli.main gateway run' | grep -v infisical | head -1", user=user)
        if not pid:
            pid = ssh(host, "pgrep -f '[h]ermes.*gateway' | grep -v infisical | grep -v bash | head -1", user=user)
        if not pid:
            pid = "?"

        # ⛔ KOBY IS NEVER REPAIRED — diagnostic only
        if report_only:
            print(f"  🔍 {name}: REPORT-ONLY mode (diagnostic only, no repairs on .129)")
            # Still check gateway status for reporting purposes
            if pid == "?":
                print(f"  ⚠️  {name}: GATEWAY NOT RUNNING (reported only)")
                _fail(f"gateway-down:{name}", name)
                continue
            else:
                print(f"  ✅ {name}: gateway running (pid={pid}, report-only mode)")
                continue  # Skip the rest of the check for Koby

        # Gateway state file
        state = ssh(host, "cat ~/.hermes/gateway_state.json 2>/dev/null", user=user)
        if state:
            try:
                st = json.loads(state)
                gw_state = st.get("gateway_state", "?")
                zulip = st.get("platforms", {}).get("zulip", {}).get("state", "?")
            except:
                gw_state, zulip = "corrupt", "?"
        else:
            gw_state, zulip = "no-state-file", "?"

        # Zulip streaming check
        adapter_paths = [
            "~/.hermes/plugins/zulip-platform/adapter.py",
            "~/.hermes/plugins/platforms/zulip/adapter.py",
        ]
        streaming = "no"
        for p in adapter_paths:
            has_edit = ssh(host, f"grep -c 'async def edit_message' {p} 2>/dev/null", user=user)
            if has_edit and has_edit != "0":
                streaming = "yes"
                break

        # Recent errors
        recent_errors = ssh(host,
            r"journalctl --user -u hermes-gateway --since '10 min ago' -o cat --no-pager 2>/dev/null "
            r"| grep -ci 'error\|traceback\|exception\|401\|403\|500' || echo 0",
            user=user)
        recent_errors = (recent_errors or "0").strip().split("\n")[-1]

        print(f"  {'✅' if gw_state == 'running' and zulip == 'connected' else '⚠️'} "
              f"{name}: gw={gw_state} zulip={zulip} streaming={streaming} "
              f"errors_10m={recent_errors.strip() or '0'} pid={pid}")


# ═══════════════════════════════════════════════════════════════════
# CHECK 4: CT Liveness (NEW)
# ═══════════════════════════════════════════════════════════════════

def check_ct_liveness():
    """Check that all agent CTs are running on their PVE nodes."""
    for name, agent in AGENTS.items():
        ct = agent["ct"]
        pve_node = agent.get("pve")
        if not pve_node:
            print(f"  ⬜ {name} (CT {ct}): no PVE node mapped — skip")
            continue

        pve_ip = PVE_NODES.get(pve_node)
        if not pve_ip:
            print(f"  ⬜ {name}: unknown PVE node '{pve_node}' — skip")
            continue

        status = ssh(pve_ip, f"pct status {ct} 2>/dev/null", user="root")
        if not status:
            print(f"  ❌ {name} (CT {ct} on {pve_node}): PVE UNREACHABLE")
            _fail(f"ct-unreachable:{name}:{pve_ip}", name)
        elif "running" in status:
            print(f"  ✅ {name} (CT {ct} on {pve_node}): running")
        elif "stopped" in status:
            print(f"  ❌ {name} (CT {ct} on {pve_node}): STOPPED")
            _fail(f"ct-stopped:{name}", name)
        else:
            print(f"  ⚠️  {name} (CT {ct} on {pve_node}): {status.strip()}")


# ═══════════════════════════════════════════════════════════════════
# CHECK 5: Config YAML Integrity (NEW)
# ═══════════════════════════════════════════════════════════════════

def check_config_integrity():
    """Verify agent config.yaml parses as valid YAML."""
    for name, agent in AGENTS.items():
        # Tanko runs on DSH (DeepSeek Harness) since 2026-08-27 — no Hermes config.yaml.
        if agent.get("runtime") == "dsh":
            print(f"  ⏭️ {name}: DSH — no Hermes config.yaml since 2026-08-27")
            continue
        if agent.get("runtime") == "pi":
            print(f"  ⏭️ {name}: pi-only runtime — no Hermes config.yaml since the harness purge")
            continue
        host = agent.get("host")
        user = agent.get("user")
        if not host or not user:
            print(f"  ⬜ {name}: cannot SSH — skip config check")
            continue

        # Check YAML parses
        yaml_ok = ssh(host,
            "python3 -c "
            '"import yaml; yaml.safe_load(open(\'/root/.hermes/config.yaml\')); print(\'OK\')" '
            "2>&1 || echo 'FAIL'",
            user=user)
        if not yaml_ok:
            print(f"  ❌ {name}: SSH UNREACHABLE (config check skipped)")
            _fail(f"config-unreachable:{name}", name)
        elif "OK" in yaml_ok:
            print(f"  ✅ {name}: config.yaml valid YAML")
        else:
            print(f"  ❌ {name}: config.yaml YAML ERROR — {yaml_ok[:120]}")
            _fail(f"config-yaml-error:{name}", name)


# ═══════════════════════════════════════════════════════════════════
# CHECK 6: Wrapper/CLI Integrity (NEW)
# ═══════════════════════════════════════════════════════════════════

def check_wrapper_integrity():
    """Verify the hermes CLI wrapper exists and can reach hermes-real."""
    for name, agent in AGENTS.items():
        # Tanko runs on DSH (DeepSeek Harness) since 2026-08-27 — no hermes CLI wrapper.
        if agent.get("runtime") == "dsh":
            print(f"  ⏭️ {name}: DSH — no hermes CLI wrapper since 2026-08-27")
            continue
        if agent.get("runtime") == "pi":
            print(f"  ⏭️ {name}: pi-only runtime — no hermes CLI wrapper since the harness purge")
            continue
        host = agent.get("host")
        user = agent.get("user")
        if not host or not user:
            print(f"  ⬜ {name}: cannot SSH — skip wrapper check")
            continue

        # Check wrapper exists
        wrapper = ssh(host, "ls -la /root/.local/bin/hermes 2>/dev/null", user=user)
        if not wrapper:
            # Check alternate wrapper locations
            wrapper = ssh(host, "which hermes 2>/dev/null; command -v hermes 2>/dev/null", user=user)
            if not wrapper:
                print(f"  ❌ {name}: NO HERMES CLI WRAPPER FOUND")
                _fail(f"wrapper-missing:{name}", name)
                continue
            else:
                print(f"  ⚠️  {name}: hermes at {wrapper.strip()} (not ~/.local/bin/hermes)")

        # Credential-injection mechanism. The Hermes-era wrapper injected creds
        # with `/usr/bin/infisical run`; newer wrappers source the agent key from
        # ~/.hermes/.env and never mention infisical (koonimo/baggy does exactly
        # this). A wrapper with NO infisical reference is therefore a valid
        # alternate mechanism, not a fault — only flag a wrapper that DOES call
        # infisical when the binary cannot resolve. The old check FAILed every
        # .env-based wrapper as "infisical path may be wrong": a stale
        # expectation, not a fault.
        wrapper_body = ssh(host, "cat /root/.local/bin/hermes 2>/dev/null", user=user) or ""
        if "infisical" in wrapper_body:
            inf_actual = ssh(host, "command -v infisical 2>/dev/null", user=user)
            if not inf_actual:
                print(f"  ❌ {name}: wrapper invokes infisical but the binary is MISSING")
                _fail(f"wrapper-no-infisical:{name}", name)
            elif "/usr/bin/infisical" not in wrapper_body:
                print(f"  ⚠️  {name}: wrapper infisical path differs (infisical at {inf_actual}) — informational")
            else:
                print(f"  ✅ {name}: wrapper infisical path OK")
        else:
            print(f"  ℹ️  {name}: wrapper resolves creds without infisical (e.g. ~/.hermes/.env) — OK")

        # Check hermes-real exists
        hermes_real = ssh(host,
            "ls -la /root/.local/bin/hermes-real 2>/dev/null || echo MISS",
            user=user)
        if not hermes_real or hermes_real.strip() == "MISS":
            # Check venv path
            hermes_real = ssh(host,
                "ls -la /usr/local/lib/hermes-agent/venv/bin/hermes 2>/dev/null || echo MISS",
                user=user)
            if not hermes_real or hermes_real.strip() == "MISS":
                print(f"  ❌ {name}: hermes-real NOT FOUND (wrapper broken)")
                _fail(f"wrapper-no-hermes-real:{name}", name)
            else:
                print(f"  ✅ {name}: hermes-real at alt path")

        # Check the .env file has the key
        env_has_key = ssh(host,
            "grep -c 'LITELLM_API_KEY' /root/.hermes/.env 2>/dev/null || echo 0",
            user=user)
        if env_has_key and env_has_key.strip() not in ("", "0"):
            print(f"  ✅ {name}: wrapper + .env key present")
        else:
            print(f"  ⚠️  {name}: .env may be missing LITELLM_API_KEY entry")


# ═══════════════════════════════════════════════════════════════════
# CHECK 7: Vault Secret Non-Emptiness (NEW)
# ═══════════════════════════════════════════════════════════════════

def check_vault_secrets():
    """Verify agent-specific vault secrets are non-empty and start with sk-."""
    for name, agent in AGENTS.items():
        vault_key_name = agent.get("vault_key")
        if not vault_key_name:
            continue

        key = agent.get("key")
        if not key:
            print(f"  ❌ {name}: vault secret {vault_key_name} MISSING or EMPTY")
            _fail(f"vault-empty:{name}:{vault_key_name}", name)
        elif not key.startswith("sk-"):
            print(f"  ❌ {name}: vault secret {vault_key_name} WRONG FORMAT (starts '{key[:8]}...')")
            _fail(f"vault-bad-format:{name}:{vault_key_name}", name)
        else:
            print(f"  ✅ {name}: vault {vault_key_name}=sk-...{key[-4:]}")


# ═══════════════════════════════════════════════════════════════════
# DEPLOY: copy updated script to /root/scripts/ on local host
# ═══════════════════════════════════════════════════════════════════

def deploy_self():
    """Copy this script to /root/scripts/agent-health-check.py if out of date."""
    dest = "/root/scripts/agent-health-check.py"
    try:
        with open(__file__, "r") as f:
            current = f.read()
        if os.path.isfile(dest):
            with open(dest, "r") as f:
                existing = f.read()
            if current == existing:
                return  # Already deployed
        # Write new version
        with open(dest, "w") as f:
            f.write(current)
        os.chmod(dest, 0o755)
        print(f"  📦 Deployed updated script to {dest}")
    except:
        pass  # Not fatal if deploy fails


# ═══════════════════════════════════════════════════════════════════
# MAIN
# ═══════════════════════════════════════════════════════════════════

def main():
    quiet = "--quiet" in sys.argv
    as_json = "--json" in sys.argv

    # Self-deploy to canonical location
    if not quiet and "--no-deploy" not in sys.argv:
        deploy_self()

    # Provenance: a report is only actionable if the reader can tell WHICH copy
    # of this script produced it. Emit the absolute execution path (script + cwd)
    # in both human and --json output so a stale-consumer report is
    # distinguishable from a real fault at read time.
    script_path = os.path.abspath(__file__)
    cwd = os.getcwd()

    if not quiet:
        print(f"🏥 Agent Health Check v4 — {datetime.now().strftime('%Y-%m-%d %H:%M UTC')}")
        print(f"📍 executed from: script={script_path} cwd={cwd}")
        print()

    load_agent_keys()

    print("🔑 LiteLLM Keys:")
    check_keys()
    print()

    print("🎮 GPU Port Health:")
    check_gpu_ports()
    print()

    print("🤖 Agent Gateways:")
    check_agents()
    print()

    print("🖥️  CT Liveness:")
    check_ct_liveness()
    print()

    print("📝 Config Integrity:")
    check_config_integrity()
    print()

    print("🔌 Wrapper/CLI Integrity:")
    check_wrapper_integrity()
    print()

    print("🔐 Vault Secrets:")
    check_vault_secrets()

    if FAIL:
        print(f"\n❌ {len(FAIL)} FAILURE(S): {' | '.join(FAIL)}")
        if quiet:
            print(f"ALERT agent-health:{','.join(FAIL)}")
    elif not quiet:
        print("\n✅ All checks passed")

    if as_json:
        print(json.dumps({"timestamp": datetime.now().isoformat(),
                          "execution_path": script_path, "cwd": cwd,
                          "failures": FAIL, "healthy": len(FAIL) == 0}))

    sys.exit(1 if FAIL else 0)

if __name__ == "__main__":
    main()