SwitchModel -> SwitchCoder

This commit is contained in:
Your Name 2024-07-28 17:49:02 -03:00
parent 4537e6c16a
commit f440479c48
2 changed files with 14 additions and 7 deletions

View file

@ -12,7 +12,7 @@ from prompt_toolkit.enums import EditingMode
from aider import __version__, models, utils
from aider.args import get_parser
from aider.coders import Coder
from aider.commands import SwitchModel
from aider.commands import SwitchCoder
from aider.io import InputOutput
from aider.llm import litellm # noqa: F401; properly init litellm on launch
from aider.repo import GitRepo
@ -587,8 +587,13 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
try:
coder.run()
return
except SwitchModel as switch:
coder = Coder.create(main_model=switch.model, io=io, from_coder=coder)
except SwitchCoder as switch:
coder = Coder.create(
main_model=switch.model,
edit_format=switch.edit_format,
io=io,
from_coder=coder,
)
coder.show_announcements()