diff options
Diffstat (limited to 'src/gap_pipeline/prompts.py')
| -rw-r--r-- | src/gap_pipeline/prompts.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/gap_pipeline/prompts.py b/src/gap_pipeline/prompts.py index 2ba87cf..30d65b2 100644 --- a/src/gap_pipeline/prompts.py +++ b/src/gap_pipeline/prompts.py @@ -8,7 +8,7 @@ from __future__ import annotations import json -from .models import CanonicalItem, KernelCandidate, KernelPlan +from .models import CanonicalItem, KernelCandidate, KernelPlan, ProofPlanDAG # Source: PutnamVariants@c3bed737370df2dbf73afd66bf6e86d4ece82d68 @@ -220,12 +220,13 @@ def kernel_generate_user(item: CanonicalItem, plan: KernelPlan) -> str: def judge_user( item: CanonicalItem, plan: KernelPlan, + dag: ProofPlanDAG, candidate: KernelCandidate, ) -> str: return JUDGE_USER_TEMPLATE.format( original_problem=item.problem, original_solution=item.solution, - method_labels=json.dumps(plan.core_steps, ensure_ascii=False), + method_labels=json.dumps(dag.method_labels_payload(), ensure_ascii=False), slot_replacement=json.dumps( { key: value.model_dump(mode="json") |