| |
| """Minimal helpers for generated code execution reports.""" |
|
|
| from __future__ import annotations |
|
|
| import json |
| import os |
| from dataclasses import dataclass, asdict |
| from typing import List, Optional |
|
|
| from .debug_memory import DebugMemoryStore |
|
|
|
|
| @dataclass |
| class DebugMetadata: |
| problem_id: int |
| notes: List[str] |
|
|
| def to_json(self) -> str: |
| return json.dumps(asdict(self), ensure_ascii=False, indent=2) |
|
|
|
|
| def sanitize_code(code: str, problem_id: int): |
| """Ensure code ends with a newline and capture any lightweight notes.""" |
| metadata = DebugMetadata(problem_id=problem_id, notes=[]) |
| cleaned = (code or "").rstrip() + "\n" if code else "" |
| return cleaned, metadata |
|
|
|
|
| def save_debug_metadata(metadata: DebugMetadata, output_dir: str) -> None: |
| """Persist metadata only when there is something noteworthy.""" |
| if not metadata.notes: |
| return |
| debug_dir = os.path.join(output_dir, "debug") |
| os.makedirs(debug_dir, exist_ok=True) |
| path = os.path.join(debug_dir, f"problem_{metadata.problem_id}.json") |
| with open(path, "w", encoding="utf-8") as fh: |
| fh.write(metadata.to_json()) |
|
|
|
|
| def write_debug_suggestions( |
| problem_id: int, |
| description: str, |
| error_message: str, |
| memory_helper, |
| memory_bank, |
| output_dir: str, |
| *, |
| status: str, |
| debug_store: Optional[DebugMemoryStore] = None, |
| top_k_cases: int = 3, |
| ) -> None: |
| """Write a straightforward debug report and optionally record the memory.""" |
| _ = memory_helper, memory_bank, top_k_cases |
| debug_dir = os.path.join(output_dir, "debug") |
| os.makedirs(debug_dir, exist_ok=True) |
| path = os.path.join(debug_dir, f"problem_{problem_id}_suggestions.md") |
|
|
| lines: List[str] = [ |
| f"# Debug Report for Problem {problem_id}", |
| "", |
| f"- **Status:** {status}", |
| ] |
| if description: |
| lines.extend(["", "## Description", description.strip(), ""]) |
| if error_message: |
| lines.extend( |
| [ |
| "## Error Traceback", |
| "```", |
| error_message.strip(), |
| "```", |
| "", |
| ] |
| ) |
| else: |
| lines.extend(["", "## Error Traceback", "_No traceback captured._", ""]) |
|
|
| lines.append("## Notes") |
| lines.append("") |
| lines.append("Automated debugging is not yet implemented. Review the trace above for hints.") |
| lines.append("") |
|
|
| with open(path, "w", encoding="utf-8") as fh: |
| fh.write("\n".join(lines)) |
|
|
| if debug_store: |
| debug_store.record_execution_feedback( |
| problem_id=problem_id, |
| description=description, |
| status=status, |
| error_text=error_message or status, |
| guidance="Automated debugging is not yet implemented.", |
| source="debug_utils.write_debug_suggestions", |
| metadata={}, |
| ) |
|
|
|
|
| __all__ = ["DebugMetadata", "sanitize_code", "save_debug_metadata", "write_debug_suggestions"] |
|
|
