| 10 | |
| 11 | |
| 12 | def parse_args(): |
| 13 | parser = argparse.ArgumentParser(description="Run agent") |
| 14 | |
| 15 | parser.add_argument( |
| 16 | "--mode", |
| 17 | type=str, |
| 18 | default="simple", |
| 19 | help="Valid modes include simple, plan, loop", |
| 20 | ) |
| 21 | parser.add_argument( |
| 22 | "--workflow_file", type=str, default=None, help="Path to the workflow file." |
| 23 | ) |
| 24 | parser.add_argument( |
| 25 | "--max_tokens_per_step", |
| 26 | type=int, |
| 27 | default=None, |
| 28 | help="Maximum tokens per step. If not specified, uses YAML config or default (100000).", |
| 29 | ) |
| 30 | parser.add_argument( |
| 31 | "--max_tool_calls_per_step", |
| 32 | type=int, |
| 33 | default=None, |
| 34 | help="Maximum tool calls per step. If not specified, uses YAML config or default (10).", |
| 35 | ) |
| 36 | parser.add_argument( |
| 37 | "--model", |
| 38 | type=str, |
| 39 | default=None, |
| 40 | help="Model to use. If not specified, uses YAML config or default (gpt-4.1-mini).", |
| 41 | ) |
| 42 | parser.add_argument( |
| 43 | "--temperature", |
| 44 | type=float, |
| 45 | default=None, |
| 46 | help="temperature. If not specified, uses YAML config or default (0.2).", |
| 47 | ) |
| 48 | parser.add_argument( |
| 49 | "--plan_revision_max_steps", |
| 50 | type=int, |
| 51 | default=None, |
| 52 | help="The max number of steps to revise plan. If not specified, uses YAML config or default (0).", |
| 53 | ) |
| 54 | parser.add_argument( |
| 55 | "--mcp_url", |
| 56 | type=str, |
| 57 | default="http://localhost:7002/mcp", |
| 58 | help="Remote MCP server url", |
| 59 | ) |
| 60 | parser.add_argument( |
| 61 | "--output_dir", |
| 62 | type=str, |
| 63 | default=None, |
| 64 | help="Override output directory (default: outputs/<task_name>).", |
| 65 | ) |
| 66 | parser.add_argument( |
| 67 | "--resume", |
| 68 | action="store_true", |
| 69 | help="Resume from checkpoint.json (skips completed steps).", |