mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-24 14:25:00 +00:00
refactor into Model.token_count()
This commit is contained in:
parent
90f57664a6
commit
c24a4a4392
3 changed files with 24 additions and 10 deletions
|
@ -420,15 +420,9 @@ class Coder:
|
|||
prompt = prompt.format(fence=self.fence)
|
||||
return prompt
|
||||
|
||||
def send_new_user_message(self, inp):
|
||||
def format_messages(self):
|
||||
self.choose_fence()
|
||||
|
||||
self.cur_messages += [
|
||||
dict(role="user", content=inp),
|
||||
]
|
||||
|
||||
main_sys = self.gpt_prompts.main_system
|
||||
# if self.main_model.max_context_tokens > 4 * 1024:
|
||||
main_sys += "\n" + self.fmt_system_reminder()
|
||||
|
||||
messages = [
|
||||
|
@ -440,6 +434,15 @@ class Coder:
|
|||
messages += self.get_files_messages()
|
||||
messages += self.cur_messages
|
||||
|
||||
return messages
|
||||
|
||||
def send_new_user_message(self, inp):
|
||||
self.cur_messages += [
|
||||
dict(role="user", content=inp),
|
||||
]
|
||||
|
||||
messages = self.format_messages()
|
||||
|
||||
if self.verbose:
|
||||
utils.show_messages(messages, functions=self.functions)
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue