diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py index d2e1185..1c80dcb 100644 --- a/aider_gitea/__main__.py +++ b/aider_gitea/__main__.py @@ -7,8 +7,10 @@ It assumes that the default branch (default "main") exists and that you have a v import argparse import logging import time +import aide_gitea from . import RepositoryConfig, secrets, solve_issues_in_repository +from . import CODE_MODEL, EVALUATOR_MODEL from .gitea_client import GiteaClient from .seen_issues_db import SeenIssuesDB @@ -45,12 +47,24 @@ def parse_args(): default=300, help='Interval in seconds between checks in daemon mode (default: 300)', ) + parser.add_argument( + '--aider-model', + default=CODE_MODEL, + help='Aider model to use (default: %(default)s)', + ) + parser.add_argument( + '--evaluator-model', + default=EVALUATOR_MODEL, + help='Evaluator model to use (default: %(default)s)', + ) return parser.parse_args() def main(): logging.basicConfig(level='INFO') args = parse_args() + aide_gitea.CODE_MODEL = args.aider_model + aide_gitea.EVALUATOR_MODEL = args.evaluator_model seen_issues_db = SeenIssuesDB() client = GiteaClient(args.gitea_url, secrets.gitea_token())