fix: health watchdog false 504s — Caddy header timeout + per-model logic
Root cause: 9router combo models (deepseek-v4-flash-free on fallback) have TTFT up to 30-40s. Caddy 9router route inherited the default response_header_timeout 30s / read_timeout 60s → 504 'timeout awaiting response headers' even though 9router was still processing. Cloudflare/log showed repeated 504s; health watchdog (correctly) flagged the outage. Fixes: 1. Caddy: dedicated 9router route with response_header_timeout 120s + read/write 300s (was default 30/60). Removed invalid top-level flush_interval on upload block that broke caddy reload (2.11 rejects it as transport subdirective). 2. health-check: HTTP timeout 60→150s (mirror Caddy), and alert ONLY when EVERY model fails — any working model means the server's fallback chain succeeds. Early-exit on first success to bound runtime (~3s healthy). Verified: 3 runs green, ~3.6s each, silent exit 0.
This commit is contained in:
+36
-19
@@ -3,11 +3,16 @@
|
|||||||
PR-Agent Model Health Watchdog
|
PR-Agent Model Health Watchdog
|
||||||
===============================
|
===============================
|
||||||
Runs every 10 minutes via Hermes no_agent cron. Tests the exact model config
|
Runs every 10 minutes via Hermes no_agent cron. Tests the exact model config
|
||||||
the pr-agent server uses (primary + fallbacks) against 9router via litellm.
|
the pr-agent server uses (primary + fallbacks) against 9router via raw HTTP.
|
||||||
|
|
||||||
Output contract (no_agent cron):
|
Output contract (no_agent cron):
|
||||||
- OK → empty stdout (silent, $0 idle)
|
- OK → empty stdout (silent, $0 idle)
|
||||||
- FAIL → one-line alert + detail (delivered to Discord/home channel)
|
- FAIL → one-line alert + detail (delivered to Discord/home channel)
|
||||||
|
|
||||||
|
Design: alert only when EVERY configured model fails (primary AND all
|
||||||
|
fallbacks). If any model works, the server's own fallback chain will succeed,
|
||||||
|
so the system is healthy even if the primary is down/slow. This prevents
|
||||||
|
false alerts from a single slow/failed model.
|
||||||
"""
|
"""
|
||||||
import os, sys, json, hashlib, subprocess
|
import os, sys, json, hashlib, subprocess
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
@@ -16,6 +21,9 @@ BWS_SECRET_ID = "2aef2194-971d-4dae-99dd-b49a0041f97c"
|
|||||||
ROUTER_BASE = "https://9router.asepharyana.my.id/v1"
|
ROUTER_BASE = "https://9router.asepharyana.my.id/v1"
|
||||||
PRIMARY = "openai/claude-opus-4-8"
|
PRIMARY = "openai/claude-opus-4-8"
|
||||||
FALLBACKS = ["openai/ATLAS", "openai/gemini", "openai/text", "openai/deepseek-v4-flash-free"]
|
FALLBACKS = ["openai/ATLAS", "openai/gemini", "openai/text", "openai/deepseek-v4-flash-free"]
|
||||||
|
# Caddy 9router route is now response_header_timeout 120s / read 300s.
|
||||||
|
# LLM combo TTFT often 30-40s+. Give the check room to complete.
|
||||||
|
HTTP_TIMEOUT = 150
|
||||||
CONSECUTIVE_FAIL_FILE = Path("/tmp/pr-agent-health-fail-count")
|
CONSECUTIVE_FAIL_FILE = Path("/tmp/pr-agent-health-fail-count")
|
||||||
|
|
||||||
# ── key from BWS ────────────────────────────────────────────────────────────
|
# ── key from BWS ────────────────────────────────────────────────────────────
|
||||||
@@ -52,14 +60,14 @@ def get_key() -> str:
|
|||||||
if r.returncode != 0:
|
if r.returncode != 0:
|
||||||
return ""
|
return ""
|
||||||
# Value is shell-quoted KEY="value" — take first line only. BWS sometimes
|
# Value is shell-quoted KEY="value" — take first line only. BWS sometimes
|
||||||
# appends "# one or more secrets have been commented-out..." after a
|
# appends "# one or more secrets have been commented-out..."; only the
|
||||||
# problematic key rename; only the first line is the real key value.
|
# first line is the real key value.
|
||||||
line = r.stdout.split("\n")[0]
|
line = r.stdout.split("\n")[0]
|
||||||
if "=" not in line:
|
if "=" not in line:
|
||||||
return ""
|
return ""
|
||||||
val = line.split("=", 1)[1].strip()
|
val = line.split("=", 1)[1].strip().strip('"')
|
||||||
if len(val) >= 2 and val[0] == val[-1] and val[0] in "\"'":
|
if len(val) < 10:
|
||||||
val = val[1:-1]
|
return ""
|
||||||
return val
|
return val
|
||||||
except Exception:
|
except Exception:
|
||||||
return ""
|
return ""
|
||||||
@@ -88,7 +96,7 @@ def check_model(model: str, key: str) -> tuple:
|
|||||||
headers={"Authorization": f"Bearer {key}", "Content-Type": "application/json"},
|
headers={"Authorization": f"Bearer {key}", "Content-Type": "application/json"},
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
with urllib.request.urlopen(req, timeout=60) as r:
|
with urllib.request.urlopen(req, timeout=HTTP_TIMEOUT) as r:
|
||||||
return r.status == 200, f"HTTP {r.status}"
|
return r.status == 200, f"HTTP {r.status}"
|
||||||
except urllib.error.HTTPError as e:
|
except urllib.error.HTTPError as e:
|
||||||
err = e.read().decode(errors="replace")[:160].replace("\n", " ")
|
err = e.read().decode(errors="replace")[:160].replace("\n", " ")
|
||||||
@@ -103,21 +111,31 @@ def main() -> int:
|
|||||||
print("⚠️ pr-agent health: cannot fetch router key from BWS (bws unavailable)")
|
print("⚠️ pr-agent health: cannot fetch router key from BWS (bws unavailable)")
|
||||||
return 1
|
return 1
|
||||||
|
|
||||||
failures = []
|
results = {}
|
||||||
ok_primary, detail = check_model(PRIMARY, key)
|
ok_somewhere = False
|
||||||
if not ok_primary:
|
results[PRIMARY] = check_model(PRIMARY, key)
|
||||||
failures.append(f"primary {PRIMARY} → {detail}")
|
ok_somewhere = ok_somewhere or results[PRIMARY][0]
|
||||||
for fb in FALLBACKS:
|
if not ok_somewhere:
|
||||||
ok, d = check_model(fb, key)
|
for fb in FALLBACKS:
|
||||||
if not ok:
|
results[fb] = check_model(fb, key)
|
||||||
failures.append(f"fallback {fb} → {d}")
|
if results[fb][0]:
|
||||||
|
ok_somewhere = True
|
||||||
|
break # bound runtime; one working model is enough
|
||||||
|
else:
|
||||||
|
# ensure every fallback appears in results for the report
|
||||||
|
for fb in FALLBACKS:
|
||||||
|
results.setdefault(fb, (False, "not tested (prior model failed)"))
|
||||||
|
else:
|
||||||
|
for fb in FALLBACKS:
|
||||||
|
results.setdefault(fb, (True, "not checked (primary ok)"))
|
||||||
|
|
||||||
if not failures:
|
# Any model working = server's fallback chain will succeed = healthy.
|
||||||
# healthy — clear counter, stay silent
|
if ok_somewhere:
|
||||||
CONSECUTIVE_FAIL_FILE.unlink(missing_ok=True)
|
CONSECUTIVE_FAIL_FILE.unlink(missing_ok=True)
|
||||||
return 0
|
return 0
|
||||||
|
|
||||||
# At least one model failed. Count consecutive failures to avoid flapping.
|
# Every model failed. Count consecutive to avoid flapping on 1-off glitch.
|
||||||
|
failures = [f"{m} → {d}" for m, (ok, d) in results.items() if not ok]
|
||||||
n = 1
|
n = 1
|
||||||
if CONSECUTIVE_FAIL_FILE.exists():
|
if CONSECUTIVE_FAIL_FILE.exists():
|
||||||
try:
|
try:
|
||||||
@@ -127,7 +145,6 @@ def main() -> int:
|
|||||||
CONSECUTIVE_FAIL_FILE.write_text(str(n))
|
CONSECUTIVE_FAIL_FILE.write_text(str(n))
|
||||||
|
|
||||||
if n < 2:
|
if n < 2:
|
||||||
# first failure — could be transient, stay quiet
|
|
||||||
return 0
|
return 0
|
||||||
|
|
||||||
detail = " | ".join(failures)
|
detail = " | ".join(failures)
|
||||||
|
|||||||
Reference in New Issue
Block a user