feat: add --aider-model and --evaluator-model CLI flags to override models
This commit is contained in:
parent
4c3d7d3797
commit
38b22d99c9
|
@ -11,6 +11,7 @@ import time
|
||||||
from . import RepositoryConfig, secrets, solve_issues_in_repository
|
from . import 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
|
||||||
|
import aider_gitea
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
@ -45,12 +46,26 @@ def parse_args():
|
||||||
default=300,
|
default=300,
|
||||||
help='Interval in seconds between checks in daemon mode (default: 300)',
|
help='Interval in seconds between checks in daemon mode (default: 300)',
|
||||||
)
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'--aider-model',
|
||||||
|
default=None,
|
||||||
|
help='Aider model to use (overrides default CODE_MODEL)',
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'--evaluator-model',
|
||||||
|
default=None,
|
||||||
|
help='Evaluator model to use (overrides default EVALUATOR_MODEL)',
|
||||||
|
)
|
||||||
return parser.parse_args()
|
return parser.parse_args()
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
logging.basicConfig(level='INFO')
|
logging.basicConfig(level='INFO')
|
||||||
args = parse_args()
|
args = parse_args()
|
||||||
|
if args.aider_model:
|
||||||
|
aider_gitea.CODE_MODEL = args.aider_model
|
||||||
|
if args.evaluator_model:
|
||||||
|
aider_gitea.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