diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py index d2e1185..6ce3364 100644 --- a/aider_gitea/__main__.py +++ b/aider_gitea/__main__.py @@ -45,12 +45,28 @@ def parse_args(): default=300, help='Interval in seconds between checks in daemon mode (default: 300)', ) + parser.add_argument( + '--code-model', + dest='code_model', + help='Aider code model to use (overrides default)', + ) + parser.add_argument( + '--evaluator-model', + dest='evaluator_model', + help='Evaluator model to use (overrides default)', + ) return parser.parse_args() def main(): logging.basicConfig(level='INFO') args = parse_args() + # Override default models if provided + import aider_gitea + if args.code_model: + aider_gitea.CODE_MODEL = args.code_model + if args.evaluator_model: + aider_gitea.EVALUATOR_MODEL = args.evaluator_model seen_issues_db = SeenIssuesDB() client = GiteaClient(args.gitea_url, secrets.gitea_token())