Jmaa/aider-gitea
1
0
Fork 0
Code Issues 14 Pull Requests Actions Activity

feat: add CLI options to specify aider and evaluator models

Browse Source
This commit is contained in:
Jon Michael Aanes (aider) 2025-04-23 23:16:04 +02:00
parent 130bd7e4d4
commit 2f471d03a5
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
aider_gitea/__main__.py
View File

@ -8,7 +8,7 @@ import argparse
import logging import logging
import time import time
from . import RepositoryConfig, secrets, solve_issues_in_repository from . import RepositoryConfig, secrets, solve_issues_in_repository, CODE_MODEL, EVALUATOR_MODEL
from .gitea_client import GiteaClient from .gitea_client import GiteaClient
from .seen_issues_db import SeenIssuesDB from .seen_issues_db import SeenIssuesDB
@ -45,12 +45,27 @@ def parse_args():
default=300, default=300,
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(
'--aider-model',
default=CODE_MODEL,
help='Aider model to use (default: %(default)s)',
)
parser.add_argument(
'--evaluator-model',
default=EVALUATOR_MODEL,
help='Evaluator model to use (default: %(default)s)',
)
return parser.parse_args() return parser.parse_args()
def main(): def main():
logging.basicConfig(level='INFO') logging.basicConfig(level='INFO')
args = parse_args() args = parse_args()
# Override default models based on CLI flags
import importlib
pkg = importlib.import_module(__package__)
pkg.CODE_MODEL = args.aider_model
pkg.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())