mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-24 14:25:00 +00:00
pass in commit models to gitrepo
This commit is contained in:
parent
3662a4680b
commit
f1a31d3944
2 changed files with 10 additions and 6 deletions
|
@ -150,7 +150,11 @@ class Coder:
|
||||||
if use_git:
|
if use_git:
|
||||||
try:
|
try:
|
||||||
self.repo = GitRepo(
|
self.repo = GitRepo(
|
||||||
self.io, fnames, git_dname, aider_ignore_file, client=self.client
|
self.io,
|
||||||
|
fnames,
|
||||||
|
git_dname,
|
||||||
|
aider_ignore_file,
|
||||||
|
models=main_model.commit_message_models(),
|
||||||
)
|
)
|
||||||
self.root = self.repo.root
|
self.root = self.repo.root
|
||||||
except FileNotFoundError:
|
except FileNotFoundError:
|
||||||
|
|
|
@ -4,7 +4,7 @@ from pathlib import Path, PurePosixPath
|
||||||
import git
|
import git
|
||||||
import pathspec
|
import pathspec
|
||||||
|
|
||||||
from aider import models, prompts, utils
|
from aider import prompts, utils
|
||||||
from aider.sendchat import simple_send_with_retries
|
from aider.sendchat import simple_send_with_retries
|
||||||
|
|
||||||
from .dump import dump # noqa: F401
|
from .dump import dump # noqa: F401
|
||||||
|
@ -16,9 +16,9 @@ class GitRepo:
|
||||||
aider_ignore_spec = None
|
aider_ignore_spec = None
|
||||||
aider_ignore_ts = 0
|
aider_ignore_ts = 0
|
||||||
|
|
||||||
def __init__(self, io, fnames, git_dname, aider_ignore_file=None, client=None):
|
def __init__(self, io, fnames, git_dname, aider_ignore_file=None, models=None):
|
||||||
self.client = client
|
|
||||||
self.io = io
|
self.io = io
|
||||||
|
self.models = models
|
||||||
|
|
||||||
if git_dname:
|
if git_dname:
|
||||||
check_fnames = [git_dname]
|
check_fnames = [git_dname]
|
||||||
|
@ -120,8 +120,8 @@ class GitRepo:
|
||||||
dict(role="user", content=content),
|
dict(role="user", content=content),
|
||||||
]
|
]
|
||||||
|
|
||||||
for model in models.Model.commit_message_models():
|
for model in self.models:
|
||||||
commit_message = simple_send_with_retries(self.client, model.name, messages)
|
commit_message = simple_send_with_retries(None, model.name, messages)
|
||||||
if commit_message:
|
if commit_message:
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue