From 8b753a5a1f071a3972247ddcd9759b24381ecac9 Mon Sep 17 00:00:00 2001
From: 6vision <vision_wangpc@sina.com>
Date: Sun, 27 Oct 2024 21:44:06 +0800
Subject: [PATCH] Signed-off-by: 6vision <vision_wangpc@sina.com>

---
 bot/claudeapi/claude_api_bot.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bot/claudeapi/claude_api_bot.py b/bot/claudeapi/claude_api_bot.py
index f8160609..cd20800f 100644
--- a/bot/claudeapi/claude_api_bot.py
+++ b/bot/claudeapi/claude_api_bot.py
@@ -97,7 +97,7 @@ class ClaudeAPIBot(Bot, OpenAIImage):
             }
         except Exception as e:
             need_retry = retry_count < 2
-            result = {"completion_tokens": 0, "content": "我现在有点累了，等会再来吧"}
+            result = {"total_tokens": 0, "completion_tokens": 0, "content": "我现在有点累了，等会再来吧"}
             if isinstance(e, openai.error.RateLimitError):
                 logger.warn("[CLAUDE_API] RateLimitError: {}".format(e))
                 result["content"] = "提问太快啦，请休息一下再问我吧"