carry conversation history to switched model

This commit is contained in:
Paul Gauthier 2024-04-30 16:25:47 -07:00
parent 304856fc60
commit 8e9a00006b
2 changed files with 14 additions and 2 deletions

View file

@ -153,6 +153,8 @@ class Coder:
use_git=True, use_git=True,
voice_language=None, voice_language=None,
aider_ignore_file=None, aider_ignore_file=None,
cur_messages=None,
done_messages=None,
): ):
if not fnames: if not fnames:
fnames = [] fnames = []
@ -166,8 +168,16 @@ class Coder:
self.verbose = verbose self.verbose = verbose
self.abs_fnames = set() self.abs_fnames = set()
self.cur_messages = []
self.done_messages = [] if cur_messages:
self.cur_messages = cur_messages
else:
self.cur_messages = []
if done_messages:
self.done_messages = done_messages
else:
self.done_messages = []
self.io = io self.io = io
self.stream = stream self.stream = stream

View file

@ -401,6 +401,8 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
main_model=switch.model, main_model=switch.model,
edit_format=None, edit_format=None,
fnames=coder.get_inchat_relative_files(), fnames=coder.get_inchat_relative_files(),
done_messages=coder.done_messages,
cur_messages=coder.cur_messages,
) )
coder_kwargs.update(kwargs) coder_kwargs.update(kwargs)