From 625e71a016bd9ecec7f9cf1ff1bd9eb5e4c708ec Mon Sep 17 00:00:00 2001 From: "Jon Michael Aanes (aider)" Date: Thu, 24 Apr 2025 09:24:30 +0200 Subject: [PATCH] feat: add CLI flags to override code and evaluator models in aider_gitea --- aider_gitea/__main__.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py index d2e1185..c4ddd66 100644 --- a/aider_gitea/__main__.py +++ b/aider_gitea/__main__.py @@ -45,6 +45,14 @@ def parse_args(): default=300, help='Interval in seconds between checks in daemon mode (default: 300)', ) + parser.add_argument( + '--code-model', + help='Override the default model used for code generation (default: o4-mini)', + ) + parser.add_argument( + '--evaluator-model', + help='Override the default evaluator model (default: ollama/gemma3:27b)', + ) return parser.parse_args() @@ -52,6 +60,12 @@ def main(): logging.basicConfig(level='INFO') args = parse_args() + import aider_gitea as ag + if args.code_model: + ag.CODE_MODEL = args.code_model + if args.evaluator_model: + ag.EVALUATOR_MODEL = args.evaluator_model + seen_issues_db = SeenIssuesDB() client = GiteaClient(args.gitea_url, secrets.gitea_token())