feat: add CLI options to specify aider and evaluator models
This commit is contained in:
parent
b2dc685463
commit
5bd8bd51aa
|
@ -7,8 +7,10 @@ It assumes that the default branch (default "main") exists and that you have a v
|
|||
import argparse
|
||||
import logging
|
||||
import time
|
||||
import aide_gitea
|
||||
|
||||
from . import RepositoryConfig, secrets, solve_issues_in_repository
|
||||
from . import CODE_MODEL, EVALUATOR_MODEL
|
||||
from .gitea_client import GiteaClient
|
||||
from .seen_issues_db import SeenIssuesDB
|
||||
|
||||
|
@ -45,12 +47,24 @@ def parse_args():
|
|||
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()
|
||||
|
||||
|
||||
def main():
|
||||
logging.basicConfig(level='INFO')
|
||||
args = parse_args()
|
||||
aide_gitea.CODE_MODEL = args.aider_model
|
||||
aide_gitea.EVALUATOR_MODEL = args.evaluator_model
|
||||
|
||||
seen_issues_db = SeenIssuesDB()
|
||||
client = GiteaClient(args.gitea_url, secrets.gitea_token())
|
||||
|
|
Loading…
Reference in New Issue
Block a user