GuXiaopeng commited on
Commit
c96b0ba
·
1 Parent(s): c10c39c

A2-R1 Scoped GenerationRunner Cancellation Contract Fix (#103)

Browse files

- A2-R1 Scoped GenerationRunner Cancellation Contract Fix (0ca0af90d8dc67168b7910b19835c51ad8a6f23b)

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,