feat: adapt review budget to diff size

This commit is contained in:
Claude
2026-09-01 11:50:46 +00:00
parent f9eef969e6
commit 4d51e14a1a
5 changed files with 60 additions and 2 deletions
+2 -2
View File
@@ -440,7 +440,7 @@ def run_lenses_review(
always has: prose + a final ```json fence with the legacy schema).
"""
os.makedirs(WORK_ROOT, exist_ok=True)
budget = budget or Budget.from_config(config)
budget = budget or Budget.for_review(config, diff)
budget_state = budget_state or BudgetState(budget)
workdir = tempfile.mkdtemp(prefix=f"{repo.replace('/', '_')}-{sha[:8]}-", dir=WORK_ROOT)
keep = bool(os.environ.get("PRAGENT_KEEP_WORK"))
@@ -687,7 +687,7 @@ def run(
additional_context=additional_context,
)
budget = Budget.from_config(config)
budget = Budget.for_review(config, diff)
budget_state = BudgetState(budget)
os.makedirs(WORK_ROOT, exist_ok=True)
workdir = tempfile.mkdtemp(prefix=f"{repo.replace('/', '_')}-{sha[:8]}-", dir=WORK_ROOT)