Compare commits

..

3 Commits

Author SHA1 Message Date
652ae7d6c0 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-24 11:09:20 +02:00
85319d8bf4 feat: add CLI flags to specify code and evaluator models in aider_gitea 2025-04-24 11:09:16 +02:00
6f47924cc5 Initial ruff pass 2025-04-24 11:08:27 +02:00

View File

@ -8,13 +8,7 @@ import argparse
import logging import logging
import time import time
from . import ( from . import RepositoryConfig, secrets, solve_issues_in_repository
CODE_MODEL,
EVALUATOR_MODEL,
RepositoryConfig,
secrets,
solve_issues_in_repository,
)
from .gitea_client import GiteaClient from .gitea_client import GiteaClient
from .seen_issues_db import SeenIssuesDB from .seen_issues_db import SeenIssuesDB
@ -52,14 +46,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(
'--aider-model', '--code-model',
default=CODE_MODEL, help='Model to use for code generation (default: o4-mini)',
help='Aider model name to use (default: o4-mini)', default=None,
) )
parser.add_argument( parser.add_argument(
'--evaluator-model', '--evaluator-model',
default=EVALUATOR_MODEL, help='Model to use for evaluation (default: ollama/gemma3:27b)',
help='Evaluator model to use (default: ollama/gemma3:27b)', default=None,
) )
return parser.parse_args() return parser.parse_args()
@ -69,8 +63,10 @@ def main():
args = parse_args() args = parse_args()
import aider_gitea import aider_gitea
aider_gitea.CODE_MODEL = args.aider_model if args.code_model:
aider_gitea.EVALUATOR_MODEL = args.evaluator_model aider_gitea.CODE_MODEL = args.code_model
if args.evaluator_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())