diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py index 976b5f5..95f5389 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 @@ -62,6 +68,7 @@ def main(): logging.basicConfig(level='INFO') args = parse_args() import aider_gitea + aider_gitea.CODE_MODEL = args.code_model aider_gitea.EVALUATOR_MODEL = args.evaluator_model