Compare commits

..

3 Commits

Author SHA1 Message Date
df9d207913 Ruff after aider
All checks were successful
Run Python tests (through Pytest) / Test (push) Successful in 25s
Verify Python project can be installed, loaded and have version checked / Test (push) Successful in 23s
2025-04-23 23:27:18 +02:00
ec179b1f0b feat: add CLI flags to specify aider and evaluator models in __main__.py 2025-04-23 23:27:14 +02:00
3ba18a78b5 Initial ruff pass 2025-04-23 23:26:22 +02:00

View File

@ -6,6 +6,7 @@ It assumes that the default branch (default "main") exists and that you have a v
import argparse
import logging
import sys
import time
from . import (
@ -52,14 +53,14 @@ def parse_args():
help='Interval in seconds between checks in daemon mode (default: 300)',
)
parser.add_argument(
'--aider-model',
'--code-model',
default=CODE_MODEL,
help='Aider model to use (default: %(default)s)',
help='Aider code model (default: %(default)s)',
)
parser.add_argument(
'--evaluator-model',
default=EVALUATOR_MODEL,
help='Evaluator model to use (default: %(default)s)',
help='Ollama evaluator model (default: %(default)s)',
)
return parser.parse_args()
@ -67,11 +68,9 @@ def parse_args():
def main():
logging.basicConfig(level='INFO')
args = parse_args()
# Override default models based on CLI flags
import importlib
pkg = importlib.import_module(__package__)
pkg.CODE_MODEL = args.aider_model
# Override models based on CLI args
pkg = sys.modules[__package__]
pkg.CODE_MODEL = args.code_model
pkg.EVALUATOR_MODEL = args.evaluator_model
seen_issues_db = SeenIssuesDB()