A2-R1 Scoped GenerationRunner Cancellation Contract Fix

#103
Files changed (1) hide show
  1. core/generation_runner.py +2 -1
core/generation_runner.py CHANGED
@@ -1247,7 +1247,8 @@ class GenerationRunner:
1247
  passes = max(1, int((APP_CONFIG.get("adaptive_completion") or {}).get("section_repair_passes", 2)))
1248
  scopes = ("paragraph", "section", "section")
1249
  for idx in range(passes):
1250
- self._check_cancelled()
 
1251
  repaired = agent.adaptive_repair_academic_unit(
1252
  current, violations=remaining, section_id=section_id, section_name=section_name,
1253
  mode=mode, study_evidence=study_evidence, evidence_contract=evidence_contract,
 
1247
  passes = max(1, int((APP_CONFIG.get("adaptive_completion") or {}).get("section_repair_passes", 2)))
1248
  scopes = ("paragraph", "section", "section")
1249
  for idx in range(passes):
1250
+ if self.cancel_checker():
1251
+ raise RuntimeError("task_cancel_requested")
1252
  repaired = agent.adaptive_repair_academic_unit(
1253
  current, violations=remaining, section_id=section_id, section_name=section_name,
1254
  mode=mode, study_evidence=study_evidence, evidence_contract=evidence_contract,