94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
"""Stale extraction PR cleanup — closes extraction PRs that produce no claims.
|
|
|
|
When an extraction PR sits open >30 min with claims_count=0, it indicates:
|
|
- Extraction failed (model couldn't extract anything useful)
|
|
- Batch job stalled (no claims written)
|
|
- Source material is empty/junk
|
|
|
|
Auto-closing prevents zombie PRs from blocking the pipeline.
|
|
Logs each close for root cause analysis (model failures, bad sources, etc.).
|
|
|
|
Epimetheus owns this module.
|
|
"""
|
|
|
|
import json
|
|
import logging
|
|
from datetime import datetime, timezone
|
|
|
|
from . import config, db
|
|
from .forgejo import api, repo_path
|
|
|
|
logger = logging.getLogger("pipeline.stale_pr")
|
|
|
|
STALE_THRESHOLD_MINUTES = 45
|
|
|
|
|
|
async def check_stale_prs(conn) -> tuple[int, int]:
|
|
"""Auto-close extraction PRs open >30 min with zero claims.
|
|
|
|
Returns (stale_closed, stale_errors) — count of closed PRs and close failures.
|
|
"""
|
|
stale_closed = 0
|
|
stale_errors = 0
|
|
|
|
# Find extraction PRs: open >30 min, source has 0 claims
|
|
stale_prs = conn.execute(
|
|
"""SELECT p.number, p.branch, p.source_path, p.created_at
|
|
FROM prs p
|
|
LEFT JOIN sources s ON p.source_path = s.path
|
|
WHERE p.status = 'open'
|
|
AND p.commit_type = 'extract'
|
|
AND datetime(p.created_at) < datetime('now', '-' || ? || ' minutes')
|
|
AND COALESCE(s.claims_count, 0) = 0""",
|
|
(STALE_THRESHOLD_MINUTES,),
|
|
).fetchall()
|
|
|
|
for pr in stale_prs:
|
|
pr_num = pr["number"]
|
|
source_path = pr["source_path"] or "unknown"
|
|
|
|
try:
|
|
# Close the PR via Forgejo
|
|
result = await api(
|
|
"PATCH",
|
|
repo_path(f"pulls/{pr_num}"),
|
|
body={"state": "closed"},
|
|
)
|
|
if result is None:
|
|
stale_errors += 1
|
|
logger.warning(
|
|
"Failed to close stale extraction PR #%d (%s, %s)",
|
|
pr_num, source_path, pr["branch"],
|
|
)
|
|
continue
|
|
|
|
# Update local DB status
|
|
conn.execute(
|
|
"UPDATE prs SET status = 'closed' WHERE number = ?",
|
|
(pr_num,),
|
|
)
|
|
db.audit(
|
|
conn,
|
|
"watchdog",
|
|
"stale_pr_closed",
|
|
json.dumps({
|
|
"pr": pr_num,
|
|
"branch": pr["branch"],
|
|
"source": source_path,
|
|
"open_minutes": STALE_THRESHOLD_MINUTES,
|
|
}),
|
|
)
|
|
stale_closed += 1
|
|
logger.info(
|
|
"WATCHDOG: closed stale extraction PR #%d (no claims after %d min): %s",
|
|
pr_num, STALE_THRESHOLD_MINUTES, source_path,
|
|
)
|
|
|
|
except Exception as e:
|
|
stale_errors += 1
|
|
logger.warning(
|
|
"Stale PR close exception for #%d: %s",
|
|
pr_num, e,
|
|
)
|
|
|
|
return stale_closed, stale_errors
|