Ruff after aider

aider_gitea/__main__.py

@@ -6,9 +6,16 @@ It assumes that the default branch (default "main") exists and that you have a v
 
 import argparse
 import logging
+import sys
 import time
 
-from . import RepositoryConfig, secrets, solve_issues_in_repository
+from . import (
+    CODE_MODEL,
+    EVALUATOR_MODEL,
+    RepositoryConfig,
+    secrets,
+    solve_issues_in_repository,
+)
 from .gitea_client import GiteaClient
 from .seen_issues_db import SeenIssuesDB
 
@@ -45,12 +52,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())

aider_gitea/gitea_client.py

@@ -169,7 +169,9 @@ class GiteaClient:
         # If a pull request for this head/base already exists, return it instead of crashing
         if response.status_code == 409:
             logger.warning(
-                'Pull request already exists for head %s and base %s', head, base,
+                'Pull request already exists for head %s and base %s',
+                head,
+                base,
             )
             prs = self.get_pull_requests(owner, repo)
             for pr in prs: