From 3fe3e0f700c9156bc47a91021b0ff4d27d1c5bb2 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Thu, 26 Sep 2024 11:27:14 -0700
Subject: [PATCH] feat: Add --o1-mini and --o1-preview aliases for model
 selection

---
 aider/args.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/aider/args.py b/aider/args.py
index 6709627bc..a94c97a07 100644
--- a/aider/args.py
+++ b/aider/args.py
@@ -117,6 +117,22 @@ def get_parser(default_config_files, git_root):
         const=deepseek_model,
         help=f"Use {deepseek_model} model for the main chat",
     )
+    o1_mini_model = "o1-mini"
+    group.add_argument(
+        "--o1-mini",
+        action="store_const",
+        dest="model",
+        const=o1_mini_model,
+        help=f"Use {o1_mini_model} model for the main chat",
+    )
+    o1_preview_model = "o1-preview"
+    group.add_argument(
+        "--o1-preview",
+        action="store_const",
+        dest="model",
+        const=o1_preview_model,
+        help=f"Use {o1_preview_model} model for the main chat",
+    )
 
     ##########
     group = parser.add_argument_group("Model Settings")