| 5 | |
| 6 | |
| 7 | def parse_args(): |
| 8 | parser = argparse.ArgumentParser(description="SPO PromptOptimizer CLI") |
| 9 | |
| 10 | # LLM parameter |
| 11 | parser.add_argument("--opt-model", type=str, default="claude-3-5-sonnet-20240620", help="Model for optimization") |
| 12 | parser.add_argument("--opt-temp", type=float, default=0.7, help="Temperature for optimization") |
| 13 | parser.add_argument("--eval-model", type=str, default="gpt-4o-mini", help="Model for evaluation") |
| 14 | parser.add_argument("--eval-temp", type=float, default=0.3, help="Temperature for evaluation") |
| 15 | parser.add_argument("--exec-model", type=str, default="gpt-4o-mini", help="Model for execution") |
| 16 | parser.add_argument("--exec-temp", type=float, default=0, help="Temperature for execution") |
| 17 | |
| 18 | # PromptOptimizer parameter |
| 19 | parser.add_argument("--workspace", type=str, default="workspace", help="Path for optimized output") |
| 20 | parser.add_argument("--initial-round", type=int, default=1, help="Initial round number") |
| 21 | parser.add_argument("--max-rounds", type=int, default=10, help="Maximum number of rounds") |
| 22 | parser.add_argument("--template", type=str, default="Poem.yaml", help="Template file name") |
| 23 | parser.add_argument("--name", type=str, default="Poem", help="Project name") |
| 24 | parser.add_argument("--mode", type = str, default= "base_model", help="Execution model mode") |
| 25 | return parser.parse_args() |
| 26 | |
| 27 | |
| 28 | def main(): |