From ec179b1f0b3a1b57a72fa4e63bfbbfd69fc3e63a Mon Sep 17 00:00:00 2001
From: "Jon Michael Aanes (aider)" <jonjmaa@gmail.com>
Date: Wed, 23 Apr 2025 23:27:14 +0200
Subject: [PATCH] feat: add CLI flags to specify aider and evaluator models in
 __main__.py

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

diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py
index d2e1185..6fce9fd 100644
--- a/aider_gitea/__main__.py
+++ b/aider_gitea/__main__.py
@@ -7,6 +7,8 @@ It assumes that the default branch (default "main") exists and that you have a v
 import argparse
 import logging
 import time
+import sys
+from . import CODE_MODEL, EVALUATOR_MODEL
 
 from . import RepositoryConfig, secrets, solve_issues_in_repository
 from .gitea_client import GiteaClient
@@ -45,12 +47,26 @@ def parse_args():
         default=300,
         help='Interval in seconds between checks in daemon mode (default: 300)',
     )
+    parser.add_argument(
+        '--code-model',
+        default=CODE_MODEL,
+        help='Aider code model (default: %(default)s)',
+    )
+    parser.add_argument(
+        '--evaluator-model',
+        default=EVALUATOR_MODEL,
+        help='Ollama evaluator model (default: %(default)s)',
+    )
     return parser.parse_args()
 
 
 def main():
     logging.basicConfig(level='INFO')
     args = parse_args()
+    # Override models based on CLI args
+    pkg = sys.modules[__package__]
+    pkg.CODE_MODEL = args.code_model
+    pkg.EVALUATOR_MODEL = args.evaluator_model
 
     seen_issues_db = SeenIssuesDB()
     client = GiteaClient(args.gitea_url, secrets.gitea_token())