From 821e4bbdf0ac42b267d5957d249a4bebcce71e23 Mon Sep 17 00:00:00 2001
From: better629 <webdesign_xmu2012@163.com>
Date: Sun, 16 Mar 2025 16:41:54 +0800
Subject: [PATCH] update cluade token count

---
 metagpt/utils/token_counter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/metagpt/utils/token_counter.py b/metagpt/utils/token_counter.py
index e977884ab..88938ed84 100644
--- a/metagpt/utils/token_counter.py
+++ b/metagpt/utils/token_counter.py
@@ -421,7 +421,7 @@ def count_message_tokens(messages, model="gpt-3.5-turbo-0125"):
     """Return the number of tokens used by a list of messages."""
     if "claude" in model:
         num_tokens = count_claude_message_tokens(messages, model)
-        return num_tokens.input_tokens
+        return num_tokens
     try:
         encoding = tiktoken.encoding_for_model(model)
     except KeyError: