Add arguments for specifying the aider model and the evaluator model. #106

Merged
Jmaa merged 3 commits from issue-97-add-arguments-for-specifying-the-aider-model-and-the-evaluator-model into main 2025-04-24 09:59:19 +00:00
Showing only changes of commit 7a35029a18 - Show all commits

View File

@ -45,6 +45,16 @@ def parse_args():
default=300,
help='Interval in seconds between checks in daemon mode (default: 300)',
)
parser.add_argument(
'--aider-model',
help='Model to use for generating code (overrides default)',
default=None,
)
parser.add_argument(
'--evaluator-model',
help='Model to use for evaluating code (overrides default)',
default=None,
)
return parser.parse_args()
@ -52,6 +62,13 @@ def main():
logging.basicConfig(level='INFO')
args = parse_args()
# Override default models if provided
import aider_gitea as core
if args.aider_model:
core.CODE_MODEL = args.aider_model
if args.evaluator_model:
core.EVALUATOR_MODEL = args.evaluator_model
seen_issues_db = SeenIssuesDB()
client = GiteaClient(args.gitea_url, secrets.gitea_token())