feat: update counting tokens for new models

2026-07-21 14:17:11 +08:00 · 2023-06-14 13:36:07 +08:00
parent a582a46ce9
commit bb737a71d5
2 changed files with 9 additions and 8 deletions
--- a/bot/chatgpt/chat_gpt_bot.py
+++ b/bot/chatgpt/chat_gpt_bot.py
@@ -121,6 +121,7 @@ class ChatGPTBot(Bot, OpenAIImage):
            if args is None:
                args = self.args
            response = openai.ChatCompletion.create(api_key=api_key, messages=session.messages, **args)
+            # logger.debug("[CHATGPT] response={}".format(response))
            # logger.info("[ChatGPT] reply={}, total_tokens={}".format(response.choices[0]['message']['content'], response["usage"]["total_tokens"]))
            return {
                "total_tokens": response["usage"]["total_tokens"],