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