diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py index 018464c..119e3ec 100644 --- a/aider_gitea/__main__.py +++ b/aider_gitea/__main__.py @@ -8,8 +8,13 @@ import argparse import logging import time -from . import RepositoryConfig, secrets, solve_issues_in_repository -from . import CODE_MODEL, EVALUATOR_MODEL +from . import ( + CODE_MODEL, + EVALUATOR_MODEL, + RepositoryConfig, + secrets, + solve_issues_in_repository, +) from .gitea_client import GiteaClient from .seen_issues_db import SeenIssuesDB @@ -63,6 +68,7 @@ def main(): logging.basicConfig(level='INFO') args = parse_args() import aider_gitea as _pkg + _pkg.CODE_MODEL = args.aider_model _pkg.EVALUATOR_MODEL = args.evaluator_model