dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-24 14:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

Added options for automatic linting and testing after changes.

Browse source
This commit is contained in:
Paul Gauthier 2024-05-19 07:34:19 -07:00
parent 398a1300dc
commit 9ee332f5d9
5 changed files with 106 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

42
aider/coders/base_coder.py
View file

@ -59,6 +59,9 @@ class Coder:
max_reflections = 5
edit_format = None
yield_stream = False
auto_lint = True
auto_test = False
test_cmd = None
@classmethod
def create(
@ -195,6 +198,10 @@ class Coder:
done_messages=None,
max_chat_history_tokens=None,
restore_chat_history=False,
auto_lint=True,
auto_test=False,
lint_cmds=None,
test_cmd=None,
):
if not fnames:
fnames = []
@ -289,8 +296,6 @@ class Coder:
self.verbose,
)
self.linter = Linter(root=self.root, encoding=io.encoding)
if max_chat_history_tokens is None:
max_chat_history_tokens = self.main_model.max_chat_history_tokens
self.summarizer = ChatSummary(
@ -307,6 +312,14 @@ class Coder:
self.done_messages = utils.split_chat_history_markdown(history_md)
self.summarize_start()
# Linting and testing
self.linter = Linter(root=self.root, encoding=io.encoding)
self.auto_lint = auto_lint
self.setup_lint_cmds(lint_cmds)
self.auto_test = auto_test
self.test_cmd = test_cmd
# validate the functions jsonschema
if self.functions:
for function in self.functions:
@ -316,6 +329,22 @@ class Coder:
self.io.tool_output("JSON Schema:")
self.io.tool_output(json.dumps(self.functions, indent=4))
def setup_lint_cmds(self, lint_cmds):
for lint_cmd in lint_cmds:
pieces = lint_cmd.split(":")
lang = pieces[0]
cmd = lint_cmd[len(lang) + 1 :]
lang = lang.strip()
cmd = cmd.strip()
if lang and cmd:
self.linter.set_linter(lang, cmd)
else:
self.io.tool_error(f'Unable to parse --lint-cmd "{lint_cmd}"')
self.io.tool_error(f'The arg should be "language: cmd --args ..."')
self.io.tool_error('For example: --lint-cmd "python: flake8 --select=E9"')
def show_announcements(self):
for line in self.get_announcements():
self.io.tool_output(line)
@ -737,13 +766,20 @@ class Coder:
self.update_cur_messages(set())
return
if edited:
if edited and self.auto_lint:
lint_errors = self.lint_edited(edited)
if lint_errors:
self.reflected_message = lint_errors
self.update_cur_messages(set())
return
if edited and self.auto_test:
test_errors = self.commands.cmd_test(self.test_cmd)
if test_errors:
self.reflected_message = test_errors
self.update_cur_messages(set())
return
self.update_cur_messages(edited)
if edited: