token count correction according to AntonOsika#645 (AntonOsika#681)

70ziko · Oct 25, 2023 · 754f922 · 754f922
1 parent d0f9caf
commit 754f922
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/gpt_engineer/ai.py b/gpt_engineer/ai.py
@@ -171,13 +171,12 @@ def next(
 
         callsbacks = [StreamingStdOutCallbackHandler()]
         response = self.llm(messages, callbacks=callsbacks)  # type: ignore
-        messages.append(response)
-
-        logger.debug(f"Chat completion finished: {messages}")
 
         self.update_token_usage_log(
             messages=messages, answer=response.content, step_name=step_name
         )
+        messages.append(response)
+        logger.debug(f"Chat completion finished: {messages}")
 
         return messages