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

Refactor create method to use dynamic coder selection based on edit_format

Browse source
This commit is contained in:
Your Name (aider) 2024-07-28 18:45:21 -03:00
parent 563729ad94
commit 4d4bfba965
1 changed files with 8 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

30
aider/coders/base_coder.py
View file

@ -82,13 +82,7 @@ class Coder:
summarize_from_coder=True,
**kwargs,
):
from . import (
EditBlockCoder,
EditBlockFencedCoder,
HelpCoder,
UnifiedDiffCoder,
WholeFileCoder,
)
import aider.coders as coders
if not main_model:
if from_coder:
@ -129,22 +123,14 @@ class Coder:
kwargs = use_kwargs
if edit_format == "diff":
res = EditBlockCoder(main_model, io, **kwargs)
elif edit_format == "diff-fenced":
res = EditBlockFencedCoder(main_model, io, **kwargs)
elif edit_format == "whole":
res = WholeFileCoder(main_model, io, **kwargs)
elif edit_format == "udiff":
res = UnifiedDiffCoder(main_model, io, **kwargs)
elif edit_format == "help":
res = HelpCoder(main_model, io, **kwargs)
else:
raise ValueError(f"Unknown edit format {edit_format}")
for coder_class in coders.__all__:
coder = getattr(coders, coder_class)
if hasattr(coder, 'edit_format') and coder.edit_format == edit_format:
res = coder(main_model, io, **kwargs)
res.original_kwargs = dict(kwargs)
return res
res.original_kwargs = dict(kwargs)
return res
raise ValueError(f"Unknown edit format {edit_format}")
def clone(self, **kwargs):
return Coder.create(from_coder=self, **kwargs)