diff options
| author | Anonymous Authors <anonymous@invalid.example> | 2026-07-24 14:03:51 -0500 |
|---|---|---|
| committer | Anonymous Authors <anonymous@invalid.example> | 2026-07-24 14:03:51 -0500 |
| commit | 708f2af9c6985e9cb5cd53e434a7d3b8dfa2b4ac (patch) | |
| tree | f4aa3d2240a621e9057ddad2a4d8ec7820db5cde /src/gap_pipeline/prompts.py | |
| parent | d4eb26780a8a8c70ca75812af0c3c6a295c0797c (diff) | |
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") |