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 manual control of --edit-format

Browse source
This commit is contained in:
Paul Gauthier 2023-06-21 12:12:12 -07:00
parent e1d987725e
commit 75e84017b3
2 changed files with 19 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
aider/coders/base_coder.py
View file

@ -34,7 +34,13 @@ class Coder:
@classmethod
def create(
self, main_model, io, openai_api_key, openai_api_base="https://api.openai.com/v1", **kwargs
self,
main_model,
edit_format,
io,
openai_api_key,
openai_api_base="https://api.openai.com/v1",
**kwargs,
):
from . import EditBlockCoder, WholeFileCoder
@ -50,9 +56,12 @@ class Coder:
)
main_model = models.GPT35_16k
if main_model.edit_format == "diff":
if edit_format is None:
edit_format = main_model.edit_format
if edit_format == "diff":
return EditBlockCoder(main_model, io, **kwargs)
elif main_model.edit_format == "whole":
elif edit_format == "whole":
return WholeFileCoder(main_model, io, **kwargs)
else:
raise ValueError(f"{main_model} has unknown edit format {main_model.edit_format}")