Ruff after aider

aider_gitea/__main__.py

@@ -8,9 +8,13 @@ import argparse
 import logging
 import time
 
-import aider_gitea
+from . import (
-
+    CODE_MODEL,
-from . import RepositoryConfig, secrets, solve_issues_in_repository
+    EVALUATOR_MODEL,
+    RepositoryConfig,
+    secrets,
+    solve_issues_in_repository,
+)
 from .gitea_client import GiteaClient
 from .seen_issues_db import SeenIssuesDB
 
@@ -49,13 +53,13 @@ def parse_args():
     )
     parser.add_argument(
         '--aider-model',
-        default=aider_gitea.CODE_MODEL,
+        default=CODE_MODEL,
-        help=f'LLM model to use for Aider (default: {aider_gitea.CODE_MODEL})',
+        help='Model to use for code generation (default: %(default)s)',
     )
     parser.add_argument(
         '--evaluator-model',
-        default=aider_gitea.EVALUATOR_MODEL,
+        default=EVALUATOR_MODEL,
-        help=f'LLM model to use for evaluator (default: {aider_gitea.EVALUATOR_MODEL})',
+        help='Model to use for evaluation (default: %(default)s)',
     )
     return parser.parse_args()
 
@@ -63,12 +67,11 @@ def parse_args():
 def main():
     logging.basicConfig(level='INFO')
     args = parse_args()
+    # override default models from CLI
+    import aider_gitea
 
-    # Override default models if provided
+    aider_gitea.CODE_MODEL = args.aider_model
-    import aider_gitea as _pkg
+    aider_gitea.EVALUATOR_MODEL = args.evaluator_model
-
-    _pkg.CODE_MODEL = args.aider_model
-    _pkg.EVALUATOR_MODEL = args.evaluator_model
 
     seen_issues_db = SeenIssuesDB()
     client = GiteaClient(args.gitea_url, secrets.gitea_token())