| 81 | |
| 82 | |
| 83 | def parse_args() -> argparse.Namespace: |
| 84 | parser = argparse.ArgumentParser( |
| 85 | description="Debug a generated repository given an error log and planning artifacts." |
| 86 | ) |
| 87 | parser.add_argument( |
| 88 | "--error_file_name", |
| 89 | type=str, |
| 90 | required=True, |
| 91 | help="Path to a text file containing the execution error message.", |
| 92 | ) |
| 93 | |
| 94 | # Either provide output_dir directly, or let the script construct it from the dataset style |
| 95 | parser.add_argument( |
| 96 | "--output_dir", |
| 97 | type=str, |
| 98 | required=True, |
| 99 | help=( |
| 100 | "Root output directory that contains planning_trajectories.json and the debug directory." |
| 101 | ), |
| 102 | ) |
| 103 | parser.add_argument( |
| 104 | "--paper_name", |
| 105 | type=str, |
| 106 | required=True, |
| 107 | help="Paper name for output_dir.", |
| 108 | ) |
| 109 | parser.add_argument( |
| 110 | "--model", |
| 111 | type=str, |
| 112 | default="o4-mini", |
| 113 | help="OpenAI chat model used for debugging.", |
| 114 | ) |
| 115 | parser.add_argument( |
| 116 | "--save_num", |
| 117 | type=int, |
| 118 | default=1, |
| 119 | required=True, |
| 120 | help="Backup index appended as .<save_num>.bak when saving modified files.", |
| 121 | ) |
| 122 | return parser.parse_args() |
| 123 | |
| 124 | |
| 125 | args = parse_args() |