From 6d75e99a210a796d0a0d5f8cf5d34c2c36d0f7cb Mon Sep 17 00:00:00 2001 From: "Jon Michael Aanes (aider)" Date: Thu, 24 Apr 2025 00:28:38 +0200 Subject: [PATCH] feat: add CLI flags to specify aider code and evaluator models --- aider_gitea/__main__.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py index d2e1185..976b5f5 100644 --- a/aider_gitea/__main__.py +++ b/aider_gitea/__main__.py @@ -8,7 +8,7 @@ import argparse import logging import time -from . import RepositoryConfig, secrets, solve_issues_in_repository +from . import RepositoryConfig, secrets, solve_issues_in_repository, CODE_MODEL, EVALUATOR_MODEL from .gitea_client import GiteaClient from .seen_issues_db import SeenIssuesDB @@ -45,12 +45,25 @@ def parse_args(): default=300, help='Interval in seconds between checks in daemon mode (default: 300)', ) + parser.add_argument( + '--code-model', + default=CODE_MODEL, + help='Aider model to use for code generation', + ) + parser.add_argument( + '--evaluator-model', + default=EVALUATOR_MODEL, + help='Model to use for evaluating solutions', + ) return parser.parse_args() 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 seen_issues_db = SeenIssuesDB() client = GiteaClient(args.gitea_url, secrets.gitea_token())