feat: add CLI options to specify aider and evaluator models

This commit is contained in:
Jon Michael Aanes (aider) 2025-04-24 00:04:58 +02:00
parent b2dc685463
commit 5bd8bd51aa

View File

@ -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())