Compare commits

..

3 Commits

Author SHA1 Message Date
a4fcad9ad8 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-24 10:09:44 +02:00
5d7ad85d2b feat: add CLI options to specify aider and evaluator models 2025-04-24 10:09:40 +02:00
8715f5187d Initial ruff pass 2025-04-24 10:09:01 +02:00

View File

@ -8,9 +8,15 @@ import argparse
import logging
import time
import aider_gitea
import aider_gitea as ag
from . import RepositoryConfig, secrets, solve_issues_in_repository
from . import (
CODE_MODEL,
EVALUATOR_MODEL,
RepositoryConfig,
secrets,
solve_issues_in_repository,
)
from .gitea_client import GiteaClient
from .seen_issues_db import SeenIssuesDB
@ -49,13 +55,13 @@ def parse_args():
)
parser.add_argument(
'--aider-model',
default=aider_gitea.CODE_MODEL,
help=f'LLM model to use for Aider (default: {aider_gitea.CODE_MODEL})',
default=CODE_MODEL,
help='Aider model to use (default: %(default)s)',
)
parser.add_argument(
'--evaluator-model',
default=aider_gitea.EVALUATOR_MODEL,
help=f'LLM model to use for evaluator (default: {aider_gitea.EVALUATOR_MODEL})',
default=EVALUATOR_MODEL,
help='Evaluator model to use (default: %(default)s)',
)
return parser.parse_args()
@ -63,12 +69,8 @@ def parse_args():
def main():
logging.basicConfig(level='INFO')
args = parse_args()
# Override default models if provided
import aider_gitea as _pkg
_pkg.CODE_MODEL = args.aider_model
_pkg.EVALUATOR_MODEL = args.evaluator_model
ag.CODE_MODEL = args.aider_model
ag.EVALUATOR_MODEL = args.evaluator_model
seen_issues_db = SeenIssuesDB()
client = GiteaClient(args.gitea_url, secrets.gitea_token())