Compare commits

..

3 Commits

View File

@ -8,13 +8,9 @@ import argparse
import logging import logging
import time import time
from . import ( import aider_gitea
CODE_MODEL,
EVALUATOR_MODEL, from . import RepositoryConfig, secrets, solve_issues_in_repository
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
@ -53,13 +49,13 @@ def parse_args():
) )
parser.add_argument( parser.add_argument(
'--aider-model', '--aider-model',
default=CODE_MODEL, default=aider_gitea.CODE_MODEL,
help='Model to use for code generation (default: %(default)s)', help=f'LLM model to use for Aider (default: {aider_gitea.CODE_MODEL})',
) )
parser.add_argument( parser.add_argument(
'--evaluator-model', '--evaluator-model',
default=EVALUATOR_MODEL, default=aider_gitea.EVALUATOR_MODEL,
help='Model to use for evaluation (default: %(default)s)', help=f'LLM model to use for evaluator (default: {aider_gitea.EVALUATOR_MODEL})',
) )
return parser.parse_args() return parser.parse_args()
@ -67,11 +63,12 @@ def parse_args():
def main(): def main():
logging.basicConfig(level='INFO') logging.basicConfig(level='INFO')
args = parse_args() args = parse_args()
# override default models from CLI
import aider_gitea
aider_gitea.CODE_MODEL = args.aider_model # Override default models if provided
aider_gitea.EVALUATOR_MODEL = args.evaluator_model import aider_gitea as _pkg
_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())