diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py index d2e1185..3d98b24 100644 --- a/aider_gitea/__main__.py +++ b/aider_gitea/__main__.py @@ -11,6 +11,7 @@ import time 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__) @@ -45,6 +46,16 @@ def parse_args(): default=300, help='Interval in seconds between checks in daemon mode (default: 300)', ) + parser.add_argument( + '--aider-model', + 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})", + ) return parser.parse_args() @@ -52,6 +63,11 @@ def main(): logging.basicConfig(level='INFO') args = parse_args() + # Override default models if provided + import aider_gitea as _pkg + _pkg.CODE_MODEL = args.aider_model + _pkg.EVALUATOR_MODEL = args.evaluator_model + seen_issues_db = SeenIssuesDB() client = GiteaClient(args.gitea_url, secrets.gitea_token())