feat: add CLI options to specify aider and evaluator models

This commit is contained in:
Jon Michael Aanes (aider) 2025-04-24 08:47:25 +02:00
parent cf7f784450
commit fa6c221465

View File

@ -45,6 +45,16 @@ def parse_args():
default=300,
help='Interval in seconds between checks in daemon mode (default: 300)',
)
parser.add_argument(
'--aider-model',
default='o4-mini',
help='Model to use for Aider (default: o4-mini)',
)
parser.add_argument(
'--evaluator-model',
default='ollama/gemma3:27b',
help='Model to use for evaluator (default: ollama/gemma3:27b)',
)
return parser.parse_args()
@ -52,6 +62,11 @@ def main():
logging.basicConfig(level='INFO')
args = parse_args()
import importlib
pkg = importlib.import_module(__package__)
pkg.CODE_MODEL = args.aider_model
pkg.EVALUATOR_MODEL = args.evaluator_model
seen_issues_db = SeenIssuesDB()
client = GiteaClient(args.gitea_url, secrets.gitea_token())