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

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

diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py
index d2e1185..3d98b24 100644
--- a/aider_gitea/__main__.py
+++ b/aider_gitea/__main__.py
@@ -11,6 +11,7 @@ import time
 from . import RepositoryConfig, secrets, solve_issues_in_repository
 from .gitea_client import GiteaClient
 from .seen_issues_db import SeenIssuesDB
+import aider_gitea
 
 logger = logging.getLogger(__name__)
 
@@ -45,6 +46,16 @@ def parse_args():
         default=300,
         help='Interval in seconds between checks in daemon mode (default: 300)',
     )
+    parser.add_argument(
+        '--aider-model',
+        default=aider_gitea.CODE_MODEL,
+        help=f"LLM model to use for Aider (default: {aider_gitea.CODE_MODEL})",
+    )
+    parser.add_argument(
+        '--evaluator-model',
+        default=aider_gitea.EVALUATOR_MODEL,
+        help=f"LLM model to use for evaluator (default: {aider_gitea.EVALUATOR_MODEL})",
+    )
     return parser.parse_args()
 
 
@@ -52,6 +63,11 @@ def main():
     logging.basicConfig(level='INFO')
     args = parse_args()
 
+    # Override default models if provided
+    import aider_gitea as _pkg
+    _pkg.CODE_MODEL = args.aider_model
+    _pkg.EVALUATOR_MODEL = args.evaluator_model
+
     seen_issues_db = SeenIssuesDB()
     client = GiteaClient(args.gitea_url, secrets.gitea_token())