diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py index fb5b3db..4895c8f 100644 --- a/aider_gitea/__main__.py +++ b/aider_gitea/__main__.py @@ -8,7 +8,13 @@ import argparse import logging import time -from . import RepositoryConfig, secrets, solve_issues_in_repository, 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 +69,7 @@ def main(): args = parse_args() # Override default models based on CLI flags import importlib + pkg = importlib.import_module(__package__) pkg.CODE_MODEL = args.aider_model pkg.EVALUATOR_MODEL = args.evaluator_model