Compare commits
3 Commits
a25fa697d3
...
a4fcad9ad8
Author | SHA1 | Date | |
---|---|---|---|
a4fcad9ad8 | |||
5d7ad85d2b | |||
8715f5187d |
|
@ -8,9 +8,15 @@ import argparse
|
||||||
import logging
|
import logging
|
||||||
import time
|
import time
|
||||||
|
|
||||||
import aider_gitea
|
import aider_gitea as ag
|
||||||
|
|
||||||
from . import RepositoryConfig, secrets, solve_issues_in_repository
|
from . import (
|
||||||
|
CODE_MODEL,
|
||||||
|
EVALUATOR_MODEL,
|
||||||
|
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
|
||||||
|
|
||||||
|
@ -49,13 +55,13 @@ def parse_args():
|
||||||
)
|
)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
'--aider-model',
|
'--aider-model',
|
||||||
default=aider_gitea.CODE_MODEL,
|
default=CODE_MODEL,
|
||||||
help=f'LLM model to use for Aider (default: {aider_gitea.CODE_MODEL})',
|
help='Aider model to use (default: %(default)s)',
|
||||||
)
|
)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
'--evaluator-model',
|
'--evaluator-model',
|
||||||
default=aider_gitea.EVALUATOR_MODEL,
|
default=EVALUATOR_MODEL,
|
||||||
help=f'LLM model to use for evaluator (default: {aider_gitea.EVALUATOR_MODEL})',
|
help='Evaluator model to use (default: %(default)s)',
|
||||||
)
|
)
|
||||||
return parser.parse_args()
|
return parser.parse_args()
|
||||||
|
|
||||||
|
@ -63,12 +69,8 @@ def parse_args():
|
||||||
def main():
|
def main():
|
||||||
logging.basicConfig(level='INFO')
|
logging.basicConfig(level='INFO')
|
||||||
args = parse_args()
|
args = parse_args()
|
||||||
|
ag.CODE_MODEL = args.aider_model
|
||||||
# Override default models if provided
|
ag.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())
|
||||||
|
|
Loading…
Reference in New Issue
Block a user