From 85319d8bf4378b93f442b404232e975ddca3407d Mon Sep 17 00:00:00 2001
From: "Jon Michael Aanes (aider)" <jonjmaa@gmail.com>
Date: Thu, 24 Apr 2025 11:09:16 +0200
Subject: [PATCH] feat: add CLI flags to specify code and evaluator models in
 aider_gitea

---
 aider_gitea/__main__.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py
index d2e1185..e9e97f7 100644
--- a/aider_gitea/__main__.py
+++ b/aider_gitea/__main__.py
@@ -45,12 +45,27 @@ def parse_args():
         default=300,
         help='Interval in seconds between checks in daemon mode (default: 300)',
     )
+    parser.add_argument(
+        '--code-model',
+        help='Model to use for code generation (default: o4-mini)',
+        default=None,
+    )
+    parser.add_argument(
+        '--evaluator-model',
+        help='Model to use for evaluation (default: ollama/gemma3:27b)',
+        default=None,
+    )
     return parser.parse_args()
 
 
 def main():
     logging.basicConfig(level='INFO')
     args = parse_args()
+    import aider_gitea
+    if args.code_model:
+        aider_gitea.CODE_MODEL = args.code_model
+    if args.evaluator_model:
+        aider_gitea.EVALUATOR_MODEL = args.evaluator_model
 
     seen_issues_db = SeenIssuesDB()
     client = GiteaClient(args.gitea_url, secrets.gitea_token())