From d2f924d6d1d7b6366e9d5bb80ced451efd9bd6f7 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Tue, 14 Nov 2023 13:03:25 -0800 Subject: [PATCH] fixed /tokens --- aider/commands.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/aider/commands.py b/aider/commands.py index ff4dc9de7..8b05e4af5 100644 --- a/aider/commands.py +++ b/aider/commands.py @@ -106,11 +106,14 @@ class Commands: self.coder.choose_fence() # system messages - main_sys = self.fmt_system_prompt(self.gpt_prompts.main_system) - main_sys += "\n" + self.fmt_system_prompt(self.gpt_prompts.system_reminder) + main_sys = self.coder.fmt_system_prompt(self.coder.gpt_prompts.main_system) + main_sys += "\n" + self.coder.fmt_system_prompt(self.coder.gpt_prompts.system_reminder) msgs = [ dict(role="system", content=main_sys), - dict(role="system", content=self.fmt_system_prompt(self.gpt_prompts.system_reminder)), + dict( + role="system", + content=self.coder.fmt_system_prompt(self.coder.gpt_prompts.system_reminder), + ), ] tokens = self.coder.main_model.token_count(msgs)