diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py index 3d98b24..691c1e7 100644 --- a/aider_gitea/__main__.py +++ b/aider_gitea/__main__.py @@ -8,10 +8,11 @@ import argparse import logging import time +import aider_gitea + 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__) @@ -49,12 +50,12 @@ def parse_args(): parser.add_argument( '--aider-model', default=aider_gitea.CODE_MODEL, - help=f"LLM model to use for Aider (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})", + help=f'LLM model to use for evaluator (default: {aider_gitea.EVALUATOR_MODEL})', ) return parser.parse_args() @@ -65,6 +66,7 @@ def main(): # Override default models if provided import aider_gitea as _pkg + _pkg.CODE_MODEL = args.aider_model _pkg.EVALUATOR_MODEL = args.evaluator_model