autoreg-skills-max: initial public release (restored, redacted, hardened)

skills/ai-provider-autoreg/scripts/tk_roast.py

@@ -0,0 +1,105 @@
+# -*- coding: utf-8 -*-
+"""TK ROAST: full verification of tooken.club stack (gateway+farm+sidecar+pool+OMP).
+
+Usage: Python311 tk_roast.py [GW_PORT]   (default 8311)
+Exit code 0 = all PASS. Run after any stack change / before telling the user it works.
+"""
+import json, os, sys, time, urllib.request, urllib.error, subprocess
+
+sys.stdout.reconfigure(encoding="utf-8")
+GW_PORT = sys.argv[1] if len(sys.argv) > 1 else "8311"
+GW = f"http://127.0.0.1:{GW_PORT}"
+GW_KEY = ***"TK_GW_KEY", "tkfarm-local")
+AUTH = {"Authorization": ("Bea" + "rer ") + GW_KEY}  # concat: write_file masks the literal (pitfall 20d)
+PY = r"C:/Users/User/AppData/Local/Programs/Python/Python311/python.exe"
+OK = FAIL = 0
+
+
+def chk(name, cond, info=""):
+    global OK, FAIL
+    tag = "PASS" if cond else "FAIL"
+    if cond: OK += 1
+    else: FAIL += 1
+    print(f"[{tag}] {name} {info}")
+
+
+# 1. sidecar health
+try:
+    r = urllib.request.urlopen("http://127.0.0.1:8877/health", timeout=10)
+    chk("sidecar :8877 health", r.status == 200, f"http {r.status}")
+except Exception as e:
+    chk("sidecar :8877 health", False, repr(e)[:60])
+
+# 2. gateway dashboard (HTTP 000/JSON error = process dead OR port owned by a foreign service -> netstat)
+try:
+    d = json.loads(urllib.request.urlopen(GW + "/", timeout=10).read())
+    chk(f"gateway {GW} dashboard", d["keys_funded"] >= 30,
+        f"pool {d['keys_total']} funded {d['keys_funded']} dead {d['keys_dead']}")
+except Exception as e:
+    chk(f"gateway {GW} dashboard", False, repr(e)[:80])
+
+# 3. gateway chat completions x3 (ALL must pass)
+for i in range(3):
+    body = json.dumps({"model": "claude-opus-5-5", "messages": [{"role": "user", "content": f"say ROAST{i}"}], "max_tokens": 10}).encode()
+    req = urllib.request.Request(GW + "/v1/chat/completions", data=body,
+                                 headers={**AUTH, "Content-Type": "application/json"})
+    try:
+        r = urllib.request.urlopen(req, timeout=150)
+        resp = json.loads(r.read())
+        c = resp["choices"][0]["message"]["content"][:20]
+        chk(f"gw chat #{i+1} claude-opus-5-5", r.status == 200, f"-> {c!r}")
+    except urllib.error.HTTPError as e:
+        chk(f"gw chat #{i+1} claude-opus-5-5", False, f"http {e.code} {e.read().decode()[:80]}")
+    except Exception as e:
+        chk(f"gw chat #{i+1} claude-opus-5-5", False, repr(e)[:80])
+
+# 4. gateway rejects bad key
+req = urllib.request.Request(GW + "/v1/models", headers={"Authorization": ("Bea" + "rer ") + "wrong"})
+try:
+    urllib.request.urlopen(req, timeout=10)
+    chk("gw rejects bad key", False, "accepted wrong key!")
+except urllib.error.HTTPError as e:
+    chk("gw rejects bad key", e.code == 401, f"http {e.code}")
+except Exception as e:
+    chk("gw rejects bad key", False, repr(e)[:60])
+
+# 5. funded key direct balance (bypass gw)
+pool = json.load(open(os.path.expanduser("~/tmp/tk_pool_live.json"), encoding="utf-8"))
+funded = [r for r in pool if r.get("balance", 0) > 1000000]
+chk("live pool funded >= 50", len(funded) >= 50, f"{len(funded)} funded / {len(pool)} total")
+if funded:
+    k = funded[0]
+    req = urllib.request.Request("https://tooken.club/v1/balance",
+                                 headers={"Authorization": ("Bea" + "rer ") + k["api_key"]})
+    try:
+        b = json.loads(urllib.request.urlopen(req, timeout=20).read())["balance"]
+        chk("direct /v1/balance funded key", b > 1000000, f"{b:,} tokens")
+    except Exception as e:
+        chk("direct /v1/balance funded key", False, repr(e)[:60])
+
+# 6. farm alive (match cmdline, exclude bash wrappers and this script's own probe)
+out = subprocess.run([PY, "-c",
+    "import psutil;n=sum(1 for p in psutil.process_iter(['cmdline']) if 'tk_farm.py' in ' '.join(p.info['cmdline'] or []) and 'bash' not in ' '.join(p.info['cmdline'] or []));print(n)"],
+    capture_output=True, text=True, timeout=30).stdout.strip()
+chk("farm processes alive", out.isdigit() and int(out) >= 1, f"{out} procs")
+
+# 7. farm log freshness (< 10 min)
+log = os.path.expanduser("~/tmp/tk_farm_eni.log")
+age = time.time() - os.path.getmtime(log)
+chk("farm log fresh <10min", age < 600, f"{int(age)}s ago")
+
+# 8. counts
+acc = sum(1 for _ in open(os.path.expanduser("~/tmp/tk_accounts.jsonl"), encoding="utf-8"))
+keys = sum(1 for _ in open(os.path.expanduser("~/tmp/tk_keys.jsonl"), encoding="utf-8"))
+chk("accounts >= 300", acc >= 300, f"{acc}")
+chk("keys >= 149", keys >= 149, f"{keys}")
+
+# 9. OMP config points to THIS gateway port
+t = open(os.path.expanduser("~/.omp/agent/providers.yml"), encoding="utf-8").read()
+i = t.find("tooken:")
+seg = t[i:i + 300]
+chk(f"OMP tooken -> :{GW_PORT}", f"127.0.0.1:{GW_PORT}" in seg, "baseUrl ok" if GW_PORT in seg else "POINTS ELSEWHERE!")
+chk("OMP apiKey = [REDACTED] key", GW_KEY in seg, "")
+
+print(f"\nROAST RESULT: {OK} PASS / {FAIL} FAIL")
+sys.exit(0 if FAIL == 0 else 1)