Compare commits

..

3 Commits

Author SHA1 Message Date
da8f6cb04c Ruff after aider
All checks were successful
Run Python tests (through Pytest) / Test (push) Successful in 24s
Verify Python project can be installed, loaded and have version checked / Test (push) Successful in 23s
2025-04-23 23:40:35 +02:00
d546d1b294 feat: add CLI options to specify aider and evaluator models 2025-04-23 23:40:31 +02:00
5f0815732b Initial ruff pass 2025-04-23 23:39:59 +02:00

View File

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