fix: harden partial 429 success, fallbacks, zh errors
Gatekeep evals / eval-floor (push) Successful in 3s
Gatekeep evals / eval-floor (pull_request) Successful in 15s
Gatekeep / gatekeep (pull_request) Failing after 1m46s

Retry HTTP 429 with backoff; keep models beyond first 3 as
fallbacks; degrade agreement when some models fail; force-high
risk only if zero usable responses. Clearer zh-CN errors/docs for
dogfood uses: ./ rebase + workflow_dispatch re-run.
This commit is contained in:
layue13 committed 2026-10-27 08:35:52 +00:00
1 parent ef13f8806a
commit c6c2d071fb
10 files changed
+193 -61

No files matched your search

Regular → Executable
+31 -21
View File
@@ -2,6 +2,7 @@
# Call one model via provider protocol.
# Protocols: openai | openai-completions | openai-responses | anthropic
# (openai / openai-compatible → chat completions)
# Retries HTTP 429 with exponential backoff (GATEKEEP_HTTP_RETRIES, default 3).
# Args: <provider> <model> <protocol> <base_url> <prompt.json> <out.json> <secrets.json>
set -euo pipefail
@@ -21,6 +22,7 @@ if [[ -z "$KEY" ]]; then
fi
MAX_TOKENS="${GATEKEEP_MAX_TOKENS:-4096}"
RETRIES="${GATEKEEP_HTTP_RETRIES:-3}"
SYSTEM=$(python3 -c 'import json,sys; print(json.load(open(sys.argv[1]))["system"])' "$PROMPT_FILE")
USER=$(python3 -c 'import json,sys; print(json.load(open(sys.argv[1]))["user"])' "$PROMPT_FILE")
@@ -54,13 +56,9 @@ print(json.dumps({
}))
PY
)
HTTP_CODE=$(curl -sS -o "$OUT_FILE.raw" -w "%{http_code}" \
"${BASE_URL%/}/chat/completions" \
-H "Authorization: Bearer $KEY" \
-H "Content-Type: application/json" \
-H "HTTP-Referer: https://github.com/${GITHUB_REPOSITORY:-gatekeep}" \
-H "X-Title: Gatekeep" \
-d "$BODY")
URL="${BASE_URL%/}/chat/completions"
EXTRA_HDR=(-H "HTTP-Referer: https://github.com/${GITHUB_REPOSITORY:-gatekeep}" -H "X-Title: Gatekeep")
AUTH=(-H "Authorization: Bearer $KEY")
EXTRACT=openai
elif [[ "$WIRE" == "openai-responses" ]]; then
BODY=$(python3 - "$MODEL" "$SYSTEM" "$USER" "$MAX_TOKENS" <<'PY'
@@ -77,11 +75,9 @@ print(json.dumps({
}))
PY
)
HTTP_CODE=$(curl -sS -o "$OUT_FILE.raw" -w "%{http_code}" \
"${BASE_URL%/}/responses" \
-H "Authorization: Bearer $KEY" \
-H "Content-Type: application/json" \
-d "$BODY")
URL="${BASE_URL%/}/responses"
EXTRA_HDR=()
AUTH=(-H "Authorization: Bearer $KEY")
EXTRACT=openai-responses
elif [[ "$WIRE" == "anthropic" ]]; then
BODY=$(python3 - "$MODEL" "$SYSTEM" "$USER" "$MAX_TOKENS" <<'PY'
@@ -96,21 +92,36 @@ print(json.dumps({
}))
PY
)
HTTP_CODE=$(curl -sS -o "$OUT_FILE.raw" -w "%{http_code}" \
"${BASE_URL%/}/v1/messages" \
-H "x-api-key: $KEY" \
-H "anthropic-version: 2023-06-01" \
-H "Content-Type: application/json" \
-d "$BODY")
URL="${BASE_URL%/}/v1/messages"
EXTRA_HDR=(-H "anthropic-version: 2023-06-01")
AUTH=(-H "x-api-key: $KEY")
EXTRACT=anthropic
fi
if [[ "$HTTP_CODE" != "200" ]]; then
attempt=0
HTTP_CODE="000"
while true; do
HTTP_CODE=$(curl -sS -o "$OUT_FILE.raw" -w "%{http_code}" \
"$URL" \
"${AUTH[@]}" \
-H "Content-Type: application/json" \
"${EXTRA_HDR[@]}" \
-d "$BODY" || echo "000")
if [[ "$HTTP_CODE" == "200" ]]; then
break
fi
if [[ "$HTTP_CODE" == "429" && "$attempt" -lt "$RETRIES" ]]; then
sleep_s=$((2 ** attempt))
echo "$LABEL HTTP 429 — retry $((attempt + 1))/${RETRIES} after ${sleep_s}s" >&2
sleep "$sleep_s"
attempt=$((attempt + 1))
continue
fi
echo "$LABEL failed HTTP $HTTP_CODE" >&2
head -c 800 "$OUT_FILE.raw" >&2 || true
echo "{\"error\":\"http\",\"model\":\"$MODEL\",\"provider\":\"$PROVIDER\",\"http\":$HTTP_CODE}" > "$OUT_FILE"
exit 0
fi
done
python3 - "$OUT_FILE.raw" "$OUT_FILE" "$MODEL" "$PROVIDER" "$EXTRACT" <<'PY'
import json,sys
@@ -123,7 +134,6 @@ if kind == "anthropic":
if c.get("type") == "text"
)
elif kind == "openai-responses":
# Best-effort extract from Responses API
text = raw.get("output_text") or ""
if not text:
chunks = []
+15 -5
View File
@@ -167,10 +167,22 @@ for (const m of models) {
}
}
// Prefer successful model tiers; only if every call failed, fail-closed high.
const okModels = models.filter((m) => !m.error);
const modelErrors = models.filter((m) => m.error).length;
const degraded = modelErrors > 0 || okModels.length < agreementMin;
// When some models 429/fail, lower the bar to surface usable findings from survivors
// (still require at least 1 successful model). Full fail-closed high only if zero ok.
let effectiveAgreement = agreementMin;
if (okModels.length > 0 && okModels.length < agreementMin) {
effectiveAgreement = Math.max(1, okModels.length);
}
const agreed = [];
const disputed = [];
for (const f of bucket.values()) {
if (f.models.length >= agreementMin) agreed.push(f);
if (f.models.length >= effectiveAgreement) agreed.push(f);
else disputed.push(f);
}
@@ -178,8 +190,6 @@ const bySeverity = (a, b) => (sevRank[a.severity] ?? 9) - (sevRank[b.severity] ?
agreed.sort(bySeverity);
disputed.sort(bySeverity);
// Prefer successful model tiers; only if every call failed, fail-closed high.
const okModels = models.filter((m) => !m.error);
const tierSources = okModels.length ? okModels : models;
let risk_tier = floor;
for (const m of tierSources) risk_tier = maxTier(risk_tier, m.risk_tier);
@@ -193,13 +203,13 @@ if (scanHardFail) risk_tier = maxTier(risk_tier, "high");
const openSecurityBlocking =
scanHardFail || agreed.some((f) => (f.kind === "security" || f.severity === "Critical") && f.blocking);
const modelErrors = models.filter((m) => m.error).length;
const result = {
risk_tier,
policy_floor: floor,
agreement_min: agreementMin,
effective_agreement_min: effectiveAgreement,
language,
degraded,
model_errors: modelErrors,
models_ok: okModels.length,
models: models.map((m) => ({
+22 -7
View File
@@ -81,8 +81,13 @@ const T = zh
"看起来风险低 — Gatekeep + 安全绿灯后可不需人工(仅当 kill switch 打开才可自动合并)。",
approveMed: "需要 **1 名人工**过目。Bot 点赞不算人工。",
approveHigh: "需要 **2 名人工**批准。Bot 不能顶替。",
errNote: (n, ok) =>
`> **说明:** ${n} 个模型调用失败(常见于免费档限流 429)。风险档按**失败关闭**规则取高,**不等于**代码一定有问题。成功响应:${ok} 个。`,
errNote: (n, ok, degAgree) =>
`> **降级说明:** ${n} 个模型调用失败(常见免费档限流 429;已重试/尝试备用模型)。` +
(ok === 0
? `全部失败 → 风险档**失败关闭为 high**,**不等于**代码一定有问题。`
: `已用 ${ok} 个成功响应继续审查` +
(degAgree ? `(一致门槛临时降为 ${degAgree})` : "") +
`;失败关闭的 high **仅在零成功响应时**强制抬升。`),
statusBlock: "存在未关闭的安全 / Critical 结论 — 合并阻断",
statusOk: (tier) => `PR 风险 ${tier} — 无未关闭安全阻断`,
modelsAgree: "一致门槛 ≥",
@@ -135,8 +140,13 @@ const T = zh
"Looks low-risk — no human needed once Gatekeep + security are green (auto-merge only if the kill switch is on).",
approveMed: "Needs **1 human** look. Bot thumbs-up don't count as that look.",
approveHigh: "Needs **2 human** approvals. Bots never fill those seats.",
errNote: (n, ok) =>
`> **Note:** ${n} model call(s) failed (often free-tier 429). Risk tier is **fail-closed** high when calls fail — that does **not** mean the code is necessarily bad. Successful responses: ${ok}.`,
errNote: (n, ok, degAgree) =>
`> **Degraded:** ${n} model call(s) failed (often free-tier 429; retried / tried fallbacks). ` +
(ok === 0
? `Zero usable responses → risk tier **fail-closed high** (not proof the code is bad).`
: `Reviewing from ${ok} successful response(s)` +
(degAgree ? ` (agreement bar temporarily ${degAgree})` : "") +
`; fail-closed high is forced only when **zero** models succeed.`),
statusBlock: "Open security / Critical findings — merge blocked",
statusOk: (tier) => `PR risk ${tier} — no open security blockers`,
modelsAgree: "≥",
@@ -236,11 +246,16 @@ const okCount = Number(
? merged.models_ok
: (merged.models || []).filter((m) => !m.error).length,
);
const errBanner = errCount > 0 ? `\n${T.errNote(errCount, okCount)}\n` : "";
const effAgree = merged.effective_agreement_min;
const degAgree =
merged.degraded && effAgree != null && effAgree < merged.agreement_min ? effAgree : null;
const errBanner =
errCount > 0 || merged.degraded ? `\n${T.errNote(errCount, okCount, degAgree)}\n` : "";
const agreementBar = zh
? `${T.modelsAgree}${merged.agreement_min} 个模型`
: `≥${merged.agreement_min} models`;
? `${T.modelsAgree}${merged.agreement_min} 个模型` +
(degAgree ? `(本轮降为 ${degAgree})` : "")
: `≥${merged.agreement_min} models` + (degAgree ? ` (this run: ${degAgree})` : "");
const body = `${marker}
${T.title}
+52 -21
View File
@@ -46,8 +46,14 @@ KNOWN_DEFAULTS = {
}
def die(msg: str, code: int = 1) -> None:
print(f"gatekeep: {msg}", file=sys.stderr)
def _zh() -> bool:
lang = (os.environ.get("GATEKEEP_LANGUAGE") or "").strip().lower()
return lang.startswith("zh")
def die(msg: str, code: int = 1, msg_zh: str | None = None) -> None:
text = msg_zh if (_zh() and msg_zh) else msg
print(f"gatekeep: {text}", file=sys.stderr)
sys.exit(code)
@@ -57,7 +63,10 @@ def load_raw(path: Path) -> dict:
try:
import yaml # type: ignore
except ImportError:
die("PyYAML required to load providers.yaml (pip install pyyaml)")
die(
"PyYAML required to load providers.yaml (pip install pyyaml)",
msg_zh="加载 providers.yaml 需要 PyYAML(请 pip install pyyaml 或使用 Action 的 venv 步骤)",
)
raw = yaml.safe_load(text) or {}
else:
raw = json.loads(text)
@@ -240,24 +249,40 @@ def main() -> None:
requested = uniq
if len(requested) < 2:
die(f"need 2–3 models for parallel review; got {len(requested)}")
die(
f"need 2–3 models for parallel review; got {len(requested)}",
msg_zh=f"并行审查至少需要 2–3 个模型,当前只有 {len(requested)} 个",
)
fallback_requested: list[dict] = []
if len(requested) > 3:
print(f"gatekeep: truncating to first 3 of {len(requested)} models", file=sys.stderr)
print(
f"gatekeep: primary parallel set = first 3 of {len(requested)}; "
f"rest kept as fallbacks on 429/failure",
file=sys.stderr,
)
fallback_requested = requested[3:]
requested = requested[:3]
provider_meta: dict[str, dict] = {}
for pid in {r["provider"] for r in requested}:
for pid in {r["provider"] for r in requested + fallback_requested}:
pconf = providers[pid]
key_name = str(pconf["key"])
api_key = secret_value(key_name)
if not api_key:
die(
f"missing secret/env for provider {pid!r}: expected env var named {key_name!r} "
f"(from providers.*.key — free-form; set it on the workflow step env from secrets)."
f"(from providers.*.key — free-form; set it on the workflow step env from secrets).",
msg_zh=(
f"缺少供应商 {pid!r} 的密钥:需要环境变量 {key_name!r}"
f"(来自 providers.*.key;请在 workflow step 的 env 里从 secrets 注入)。"
),
)
available = discover_model_ids(pconf["base_url"], pconf["protocol"], api_key)
if not available:
die(f"provider {pid}: model list empty — check base_url/protocol/key")
die(
f"provider {pid}: model list empty — check base_url/protocol/key",
msg_zh=f"供应商 {pid} 模型列表为空 — 请检查 base_url / protocol / key",
)
provider_meta[pid] = {
"base_url": pconf["base_url"].rstrip("/"),
"protocol": pconf["protocol"],
@@ -268,26 +293,30 @@ def main() -> None:
"_available": available,
}
plan_models = []
for r in requested:
def to_plan_entry(r: dict) -> dict:
pid, model = r["provider"], r["model"]
avail = provider_meta[pid]["_available"]
if model not in avail:
sample = sorted(avail)[:8]
die(
f"model {model!r} not available on provider {pid} "
f"({provider_meta[pid]['available_count']} listed). Examples: {sample}"
f"({provider_meta[pid]['available_count']} listed). Examples: {sample}",
msg_zh=(
f"模型 {model!r} 在供应商 {pid} 上不可用"
f"(列表共 {provider_meta[pid]['available_count']} 个)。示例:{sample}"
),
)
plan_models.append(
{
"provider": pid,
"model": model,
"base_url": provider_meta[pid]["base_url"],
"protocol": provider_meta[pid]["protocol"],
"wire_protocol": provider_meta[pid]["wire_protocol"],
"key_name": provider_meta[pid]["key_name"],
}
)
return {
"provider": pid,
"model": model,
"base_url": provider_meta[pid]["base_url"],
"protocol": provider_meta[pid]["protocol"],
"wire_protocol": provider_meta[pid]["wire_protocol"],
"key_name": provider_meta[pid]["key_name"],
}
plan_models = [to_plan_entry(r) for r in requested]
fallback_models = [to_plan_entry(r) for r in fallback_requested]
secrets_path = out_path.with_suffix(".secrets.json")
secrets_path.write_text(
@@ -306,6 +335,7 @@ def main() -> None:
"max_tokens": int(cfg.get("max_tokens", 4096)),
"language": language,
"models": plan_models,
"fallback_models": fallback_models,
"secrets_file": str(secrets_path),
"providers_used": sorted(provider_meta.keys()),
}
@@ -315,6 +345,7 @@ def main() -> None:
{
"providers": plan["providers_used"],
"models": [f"{m['provider']}:{m['model']}" for m in plan_models],
"fallbacks": [f"{m['provider']}:{m['model']}" for m in fallback_models],
"keys": {m["provider"]: m["key_name"] for m in plan_models},
"agreement_min": plan["agreement_min"],
"language": plan["language"],
Regular → Executable
+47 -1
View File
@@ -151,7 +151,7 @@ json.dump({"system": system, "user": json.dumps(user, indent=2), "language": lan
print(f"Prompt ready; floor={floor}; language={language}; plan={len(plan.get('models',[]))} models", flush=True)
PY
# Fire models in parallel (may be mixed providers)
# Fire primary models in parallel (may be mixed providers)
pids=()
i=0
while IFS= read -r line; do
@@ -173,6 +173,52 @@ for pid in "${pids[@]}"; do
done
echo "Parallel model calls finished (wait_ec=$ec)" >&2
# On partial failure, try fallback_models from providers.yaml (beyond the first 3)
python3 - "$MODELS_DIR" "$PLAN" <<'PY' >"$OUT/fallback-needed.txt"
import json, pathlib, sys
models_dir = pathlib.Path(sys.argv[1])
plan = json.load(open(sys.argv[2]))
failed = ok = 0
for p in models_dir.glob("model-*.json"):
try:
j = json.loads(p.read_text())
except Exception:
failed += 1
continue
if j.get("error") or j.get("parse_error"):
failed += 1
else:
ok += 1
fallbacks = plan.get("fallback_models") or []
# Aim for ≥2 usable responses; if zero ok, try up to 3 reserves
if ok == 0:
need = min(len(fallbacks), 3)
elif ok < 2:
need = min(len(fallbacks), 2 - ok)
else:
need = 0
print(need)
print(f"gatekeep: primary ok={ok} failed={failed} fallbacks={len(fallbacks)} need={need}", file=sys.stderr)
PY
NEED_FB=$(head -1 "$OUT/fallback-needed.txt" | tr -d '[:space:]')
NEED_FB=${NEED_FB:-0}
if [[ "$NEED_FB" -gt 0 ]]; then
echo "gatekeep: trying $NEED_FB fallback model(s) after partial failure" >&2
fb_i=0
while IFS= read -r line; do
fb_i=$((fb_i + 1))
[[ "$fb_i" -gt "$NEED_FB" ]] && break
provider=$(printf '%s' "$line" | python3 -c 'import json,sys; print(json.load(sys.stdin)["provider"])')
model=$(printf '%s' "$line" | python3 -c 'import json,sys; print(json.load(sys.stdin)["model"])')
protocol=$(printf '%s' "$line" | python3 -c 'import json,sys; print(json.load(sys.stdin)["protocol"])')
base_url=$(printf '%s' "$line" | python3 -c 'import json,sys; print(json.load(sys.stdin)["base_url"])')
safe=$(echo "fb_${provider}_${model}" | tr '/:' '__')
bash "$ACTION/scripts/call-model.sh" \
"$provider" "$model" "$protocol" "$base_url" \
"$OUT/prompt.json" "$MODELS_DIR/model-fb$fb_i-$safe.json" "$SECRETS_FILE" || true
done < <(python3 -c 'import json,sys; [print(json.dumps(m)) for m in json.load(open(sys.argv[1])).get("fallback_models") or []]' "$PLAN")
fi
# Scrub secrets file from workspace artifact path if present
rm -f "$SECRETS_FILE"