From 7a35029a18986c0a8e7cca214a884ec4bb6c3212 Mon Sep 17 00:00:00 2001
From: "Jon Michael Aanes (aider)" <jonjmaa@gmail.com>
Date: Thu, 24 Apr 2025 11:55:21 +0200
Subject: [PATCH] feat: add CLI options to override aider and evaluator models

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

diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py
index d2e1185..8c5b35f 100644
--- a/aider_gitea/__main__.py
+++ b/aider_gitea/__main__.py
@@ -45,6 +45,16 @@ def parse_args():
         default=300,
         help='Interval in seconds between checks in daemon mode (default: 300)',
     )
+    parser.add_argument(
+        '--aider-model',
+        help='Model to use for generating code (overrides default)',
+        default=None,
+    )
+    parser.add_argument(
+        '--evaluator-model',
+        help='Model to use for evaluating code (overrides default)',
+        default=None,
+    )
     return parser.parse_args()
 
 
@@ -52,6 +62,13 @@ def main():
     logging.basicConfig(level='INFO')
     args = parse_args()
 
+    # Override default models if provided
+    import aider_gitea as core
+    if args.aider_model:
+        core.CODE_MODEL = args.aider_model
+    if args.evaluator_model:
+        core.EVALUATOR_MODEL = args.evaluator_model
+
     seen_issues_db = SeenIssuesDB()
     client = GiteaClient(args.gitea_url, secrets.gitea_token())