From 59afc5301f55037e7b379497767f4af62fd65b31 Mon Sep 17 00:00:00 2001
From: geekan <geekan@foxmail.com>
Date: Fri, 26 Jan 2024 15:08:08 +0800
Subject: [PATCH] update token counter

---
 metagpt/utils/token_counter.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/metagpt/utils/token_counter.py b/metagpt/utils/token_counter.py
index feec20928..94506e373 100644
--- a/metagpt/utils/token_counter.py
+++ b/metagpt/utils/token_counter.py
@@ -79,7 +79,10 @@ def count_message_tokens(messages, model="gpt-3.5-turbo-0613"):
         "gpt-4-32k-0314",
         "gpt-4-0613",
         "gpt-4-32k-0613",
+        "gpt-4-turbo-preview",
+        "gpt-4-0125-preview",
         "gpt-4-1106-preview",
+        "gpt-4-1106-vision-preview",
     }:
         tokens_per_message = 3  # # every reply is primed with <|start|>assistant<|message|>
         tokens_per_name = 1