feat: add CLI options to specify aider and evaluator models

This commit is contained in:
Jon Michael Aanes (aider) 2025-04-24 10:21:30 +02:00
parent 529738857a
commit bbc5d4e6f7

View File

@ -11,6 +11,7 @@ import time
from . import RepositoryConfig, secrets, solve_issues_in_repository from . import RepositoryConfig, secrets, solve_issues_in_repository
from .gitea_client import GiteaClient from .gitea_client import GiteaClient
from .seen_issues_db import SeenIssuesDB from .seen_issues_db import SeenIssuesDB
import aider_gitea
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -45,6 +46,16 @@ def parse_args():
default=300, default=300,
help='Interval in seconds between checks in daemon mode (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() return parser.parse_args()
@ -52,6 +63,11 @@ def main():
logging.basicConfig(level='INFO') logging.basicConfig(level='INFO')
args = parse_args() 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() seen_issues_db = SeenIssuesDB()
client = GiteaClient(args.gitea_url, secrets.gitea_token()) client = GiteaClient(args.gitea_url, secrets.gitea_token())