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